This repository has been archived by the owner on Jul 27, 2023. It is now read-only.
/
Vagrantfile
161 lines (146 loc) · 5.81 KB
/
Vagrantfile
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
# -*- mode: ruby -*-
# vi: set ft=ruby :
require "yaml"
config_hash = {
"worker_count" => 1,
"kubeworker_count" => 1,
"control_count" => 1,
"edge_count" => 1,
"kubeworker_ip_start" => "172.20.10.15",
"worker_ip_start" => "172.20.10.20",
"control_ip_start" => "172.20.10.10",
"control_port_start" => "808",
"edge_ip_start" => "172.20.10.25",
"kubeworker_memory" => 1024,
"worker_memory" => 1024,
"control_memory" => 1024,
"edge_memory" => 512,
"worker_cpus" => 1,
"kubeworker_cpus" => 1,
"control_cpus" => 1,
"edge_cpus" => 1,
"network" => "private",
"playbooks" => ["/vagrant/sample.yml"]
}
config_path = File.join(File.dirname(__FILE__), "vagrant-config.yml")
if !File.exist? config_path
puts "No vagrant-config.yml found, using defaults"
else
config_hash = config_hash.merge(YAML.load(File.read(config_path)))
end
# This method is used to spin up the workers, kubeworkers and edge nodes.
def spin_up(config_hash:, config:, host_array:, hostvars:, hosts:, role:)
(1..config_hash["#{role}_count"]).each do |w|
hostname = "#{role}-00#{w}"
ip = config_hash["#{role}_ip_start"] + "#{w}"
config.vm.define hostname do |server|
# Tested with 2 workers w/ 512mb, and a single one w/ 1024mb memory.
server.vm.provider :virtualbox do |vb|
vb.customize ["modifyvm", :id, "--cpus", config_hash["#{role}_cpus"]]
vb.customize ["modifyvm", :id, "--memory", config_hash["#{role}_memory"]]
end
server.vm.hostname = hostname
server.vm.network "#{config_hash['network']}_network", :ip => ip
hosts += "#{ip} #{hostname}\n"
# Update Ansible variables
host_array << hostname
server_hostvars = {
hostname => {
"ansible_ssh_host" => ip,
"private_ipv4" => ip,
"public_ipv4" => ip,
"role" => role
}
}
hostvars.merge!(server_hostvars)
end
end
end
Vagrant.require_version ">= 1.8"
Vagrant.configure(2) do |config|
# Prefer VirtualBox before VMware Fusion
config.vm.provider "virtualbox"
config.vm.provider "vmware_fusion"
config.vm.box = "CiscoCloud/mantl"
# Disable shared folder(s) for non-provisioning machines
config.vm.synced_folder ".", "/vagrant", disabled: true
config.vm.synced_folder ".", "/home/vagrant/sync", disabled: true
# All hostvars will be stored in this hash, progressively as the VMs are made
# and configured. These lists will hold hostnames that belong to these groups.
hostvars, workers, controls, edges, kubeworkers = {}, [], [], [], []
# This variable will be appended to the /etc/hosts file on the provisioner
hosts = ""
# Spin up all the worker and edge nodes based on your config or default config.
spin_up(config_hash: config_hash, config: config, hostvars: hostvars, host_array: workers, hosts: hosts, role: 'worker')
spin_up(config_hash: config_hash, config: config, hostvars: hostvars, host_array: kubeworkers, hosts: hosts, role: 'kubeworker')
spin_up(config_hash: config_hash, config: config, hostvars: hostvars, host_array: edges, hosts: hosts, role: 'edge')
(1..config_hash["control_count"]).each do |c|
hostname = "control-0#{c}"
ip = config_hash["control_ip_start"] + "#{c}"
port = config_hash["control_port_start"] + "#{c}"
last = (c >= config_hash["control_count"])
config.vm.define hostname, primary: last do |control|
control.vm.provider :virtualbox do |vb|
vb.customize ["modifyvm", :id, "--cpus", config_hash["control_cpus"]]
vb.customize ["modifyvm", :id, "--memory", config_hash["control_memory"]]
end
control.vm.hostname = hostname
control.vm.network "#{config_hash['network']}_network", :ip => ip
control.vm.network "forwarded_port", guest: 80, host: port
hosts += "#{ip} #{hostname}\n"
controls << hostname
control_hostvars = {
hostname => {
"ansible_ssh_host" => ip,
"private_ipv4" => ip,
"public_ipv4" => ip,
"role" => "control"
}
}
hostvars.merge!(control_hostvars)
if last # Only run Ansible after all hosts are up
# Sync Mantl source code, get required provisioning packages
control.vm.synced_folder ".", "/vagrant", type: "rsync",
rsync__exclude: [
".terraform/", ".git/", ".vagrant/", "docs/", "builds/",
"packer_cache/"
]
control.vm.provision "shell" do |s|
s.path = "vagrant/provision.sh"
s.args = [hosts]
end
workers = controls if workers.empty?
ansible_groups = {
"role=control" => controls,
"role=control:vars" => { "consul_is_server" => true },
"role=worker" => workers,
"role=worker:vars" => { "consul_is_server" => false },
"role=edge" => edges,
"role=kubeworker" => kubeworkers,
"role=kubeworker:vars" => { "consul_is_server" => false },
"role=edge:vars" => { "consul_is_server" => false },
"dc=vagrantdc" => workers + kubeworkers + controls + edges,
"dc=vagrantdc:vars" => {
# Ansible 2.0 depreciates the 'ssh' in these vars
"ansible_ssh_user" => "vagrant",
"ansible_ssh_pass" => "vagrant",
"consul_dc" => "vagrantdc",
"provider" => "virtualbox",
"publicly_routable" => true
}
}
# Then use the built-in ansible provisioner
config_hash["playbooks"].each do |playbook|
control.vm.provision "ansible_local" do |ansible|
ansible.install = false # should be installed via provision.sh
ansible.playbook = playbook
ansible.limit = "all"
ansible.raw_arguments = ["-e", "@/vagrant/security.yml"]
ansible.groups = ansible_groups
ansible.host_vars = hostvars
end
end
end
end
end
end