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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
require 'erb'
require 'ostruct'
require 'fileutils'
require 'json'
require 'bundler'
class Benchmarker
include FileUtils
def initialize(target, size='medium')
@target = target
@size = 'large'
end
def check_submodule
submodule = File.join('benchmarks', 'full_catalog', 'puppetlabs-puppetserver_perf_control')
unless File.exist?(File.join(submodule, 'Puppetfile'))
raise RuntimeError, 'The perf control repo is not readable. Make sure to initialize submodules by running: git submodule update --init --recursive'
end
end
def setup
require 'puppet'
config = File.join(@target, 'puppet.conf')
environment = File.join(@target, 'environments', 'perf_control')
Puppet.initialize_settings(['--config', config])
FileUtils.cd(@target) do
Bundler::with_clean_env do
system("bundle install")
system("bundle exec r10k puppetfile install --puppetfile #{File.join(environment, 'Puppetfile')} --moduledir #{File.join(environment, 'modules')} --config r10k.yaml")
end
end
# Loading the base system affects the first run a lot so burn one run here
run
end
def run(args=nil)
# Probably could pare down these facts a lot
facts = Puppet::Node::Facts.new("testing", {
'pe_concat_basedir' => '/tmp/file',
'platform_symlink_writable' => true,
'pe_build' => '2016.4.4',
'identity' => {
'privileged' => true,
},
'mountpoints' => {
'/tmp' => {
'options' => ['rw'],
}
},
'puppet_files_dir_present' => true,
'puppetversion' => '4.10.4',
'aio_agent_version' => '1.10.4',
'aio_agent_build' => '1.10.4',
'memory' => {
'system' => {
'total_bytes' => 8589934592,
}
},
'memorysize' => '8 GiB',
'osfamily' => 'RedHat',
'selinux' => false,
'operatingsystem' => 'CentOS',
'operatingsystemrelease' => '6.8',
'path' => '/tmp/staging/',
'processorcount' => '2',
'fake_domain' => 'pgtomcat.mycompany.org',
'function' => 'app',
'group' => 'pgtomcat',
'stage' => 'prod',
'staging_http_get' => 'curl',
'whereami' => 'portland',
'hostname' => 'pgtomcat',
'fqdn' => 'pgtomcat.mycompany.org',
'lsbdistcodename' => 'n/a',
'processor0' => 'AMD Ryzen 7 1700 Eight-Core Processor',
})
env = Puppet.lookup(:environments).get('perf_control')
node = Puppet::Node.indirection.find("testing", :environment => env, :facts => facts)
Puppet.push_context({:current_environment => env}, 'current env for benchmark')
Puppet::Resource::Catalog.indirection.find("testing", :use_node => node)
Puppet.pop_context
Puppet.lookup(:environments).clear('perf_control')
end
def generate
check_submodule
templates = File.join('benchmarks', 'full_catalog')
source = File.join(templates, "puppetlabs-puppetserver_perf_control")
environment = File.join(@target, 'environments', 'perf_control')
mkdir_p(File.join(@target, 'environments'))
FileUtils.cp_r(source, environment)
render(File.join(templates, 'puppet.conf.erb'),
File.join(@target, 'puppet.conf'),
:codedir => File.join(@target, 'environments'),
:target => @target)
render(File.join(templates, 'site.pp.erb'),
File.join(environment, 'manifests', 'site.pp'),
:size => @size)
render(File.join(templates, 'hiera.yaml.erb'),
File.join(@target, 'hiera.yaml'),
:datadir => File.join(environment, 'hieradata'))
FileUtils.cp(File.join(templates, 'Gemfile'), @target)
FileUtils.cp(File.join(templates, 'r10k.yaml'), @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
|
