1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
require 'erb'
require 'ostruct'
require 'fileutils'
require 'json'
class Benchmarker
include FileUtils
def initialize(target, size)
@target = target
@size = size
end
def setup
require 'puppet'
config = File.join(@target, 'puppet.conf')
Puppet.initialize_settings(['--config', config])
end
def run(args=nil)
env = Puppet.lookup(:environments).get('benchmarking')
node = Puppet::Node.new("testing", :environment => env)
Puppet::Resource::Catalog.indirection.find("testing", :use_node => node)
end
def generate
environment = File.join(@target, 'environments', 'benchmarking')
templates = File.join('benchmarks', 'many_modules')
mkdir_p(File.join(environment, 'modules'))
mkdir_p(File.join(environment, 'manifests'))
render(File.join(templates, 'site.pp.erb'),
File.join(environment, 'manifests', 'site.pp'),
:size => @size)
@size.times do |i|
module_name = "module#{i}"
module_base = File.join(environment, 'modules', module_name)
manifests = File.join(module_base, 'manifests')
locales = File.join(module_base, 'locales')
mkdir_p(manifests)
mkdir_p(locales)
File.open(File.join(module_base, 'metadata.json'), 'w') do |f|
JSON.dump({
"name" => "module#{i}",
"types" => [],
"source" => "",
"author" => "ManyModules Benchmark",
"license" => "Apache 2.0",
"version" => "1.0.0",
"description" => "Many Modules benchmark module #{i}",
"summary" => "Just this benchmark module, you know?",
"dependencies" => [],
}, f)
end
File.open(File.join(locales, 'config.yaml'), 'w') do |f|
f.puts(
{"gettext"=>
{"project_name"=>"module#{i}",
"package_name"=>"module#{i}",
"default_locale"=>"en",
"bugs_address"=>"docs@puppet.com",
"copyright_holder"=>"Puppet, Inc.",
"comments_tag"=>"TRANSLATOR",
"source_files"=>["./lib/**/*.rb"]}}.to_yaml
)
end
roles = 0.upto(10).to_a
render(File.join(templates, 'module', 'init.pp.erb'),
File.join(manifests, 'init.pp'),
:name => module_name,
:roles => roles
)
render(File.join(templates, 'module', 'internal.pp.erb'),
File.join(manifests, 'internal.pp'),
:name => module_name)
roles.each do |j|
render(File.join(templates, 'module', 'role.pp.erb'),
File.join(manifests, "role#{j}.pp"),
:name => module_name,
:index => j)
end
end
render(File.join(templates, 'puppet.conf.erb'),
File.join(@target, 'puppet.conf'),
:location => @target)
end
def render(erb_file, output_file, bindings)
site = ERB.new(File.read(erb_file))
File.open(output_file, 'w') do |fh|
fh.write(site.result(OpenStruct.new(bindings).instance_eval { binding }))
end
end
end
|
