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
|
require 'fileutils'
class Benchmarker
include FileUtils
def initialize(target, size)
@target = target
@size = size > 100 ? size : 100
end
def setup
require 'puppet'
@config = File.join(@target, 'puppet.conf')
Puppet.initialize_settings(['--config', @config])
envs = Puppet.lookup(:environments)
@node = Puppet::Node.new('testing', :environment => envs.get('benchmarking'))
end
def run(args=nil)
@compiler = Puppet::Parser::Compiler.new(@node)
@compiler.compile do |catalog|
scope = @compiler.topscope
scope['confdir'] = 'test'
@size.times do
100.times do |index|
hiera_func = @compiler.loaders.puppet_system_loader.load(:function, 'hiera_include')
hiera_func.call(scope, 'common_entry')
end
end
catalog
end
end
def generate
env_dir = File.join(@target, 'environments', 'benchmarking')
manifests_dir = File.join(env_dir, 'manifests')
dummy_class_manifest = File.join(manifests_dir, 'foo.pp')
hiera_yaml = File.join(@target, 'hiera.yaml')
datadir = File.join(@target, 'data')
common_yaml = File.join(datadir, 'common.yaml')
groups_yaml = File.join(datadir, 'groups.yaml')
mkdir_p(env_dir)
mkdir_p(manifests_dir)
mkdir_p(datadir)
File.open(hiera_yaml, 'w') do |f|
f.puts(<<-YAML)
---
:backends: yaml
:yaml:
:datadir: #{datadir}
:hierarchy:
- common
- groups
:logger: noop
YAML
end
File.open(groups_yaml, 'w') do |f|
f.puts(<<-YAML)
---
puppet:
staff:
groups:
YAML
0.upto(50).each do |i|
f.puts(" group#{i}:")
0.upto(125).each do |j|
f.puts(" - user#{j}")
end
end
end
File.open(dummy_class_manifest, 'w') do |f|
f.puts("class dummy_class { }")
end
File.open(common_yaml, 'w') do |f|
f.puts(<<-YAML)
common_entry:
- dummy_class
YAML
end
templates = File.dirname(File.realpath(__FILE__))
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
|
