| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| # -*- mode: ruby -*- |
| # vi: set ft=ruby : |
| |
| require 'socket' |
| |
| # Vagrantfile API/syntax version. Don't touch unless you know what you're doing! |
| VAGRANTFILE_API_VERSION = "2" |
| |
| enable_dns = false |
| |
| enable_hostmanager = true |
| num_zookeepers = 1 |
| num_hosts = 3 |
| ram_megabytes = 1280 |
| base_box = "ubuntu/trusty64" |
| |
| local_config_file = File.join(File.dirname(__FILE__), "Vagrantfile.local") |
| if File.exists?(local_config_file) then |
| eval(File.read(local_config_file), binding, "Vagrantfile.local") |
| end |
| |
| Vagrant.configure(VAGRANTFILE_API_VERSION) do |config| |
| config.hostmanager.enabled = enable_hostmanager |
| config.hostmanager.manage_host = enable_dns |
| config.hostmanager.include_offline = false |
| |
| ## Provider-specific global configs |
| config.vm.provider :virtualbox do |vb,override| |
| override.vm.box = base_box |
| |
| override.hostmanager.ignore_private_ip = false |
| |
| # Brokers started with the standard script currently set Xms and Xmx to 1G, |
| # plus we need some extra head room. |
| vb.customize ["modifyvm", :id, "--memory", ram_megabytes.to_s] |
| |
| if Vagrant.has_plugin?("vagrant-cachier") |
| override.cache.scope = :box |
| # Besides the defaults, we use a custom cache to handle the Oracle JDK |
| # download, which downloads via wget during an apt install. Because of the |
| # way the installer ends up using its cache directory, we need to jump |
| # through some hoops instead of just specifying a cache directly -- we |
| # share to a temporary location and the provisioning scripts symlink data |
| # to the right location. |
| override.cache.enable :generic, { |
| "oracle-jdk8" => { cache_dir: "/tmp/oracle-jdk8-installer-cache" }, |
| } |
| end |
| end |
| |
| def name_node(node, name) |
| node.vm.hostname = name |
| node.vm.provider :aws do |aws| |
| aws.tags = { 'Name' => "-" + Socket.gethostname + "-" + name } |
| end |
| end |
| |
| def assign_local_ip(node, ip_address) |
| node.vm.provider :virtualbox do |vb,override| |
| override.vm.network :private_network, ip: ip_address |
| end |
| end |
| |
| ## Cluster definition |
| zookeepers = [] |
| (1..num_zookeepers).each { |i| |
| name = "zk" + i.to_s |
| zookeepers.push(name) |
| config.vm.define name do |zookeeper| |
| name_node(zookeeper, name) |
| ip_address = "192.168.50." + (10 + i).to_s |
| assign_local_ip(zookeeper, ip_address) |
| zookeeper.vm.provision "shell", path: "vagrant/base.sh" |
| zookeeper.vm.provision "shell", path: "vagrant/zk.sh", :args => [i.to_s, num_zookeepers] |
| end |
| } |
| |
| (1..num_hosts).each { |i| |
| name = "bk" + i.to_s |
| config.vm.define name do |broker| |
| name_node(broker, name) |
| ip_address = "192.168.50." + (50 + i).to_s |
| assign_local_ip(broker, ip_address) |
| # We need to be careful about what we list as the publicly routable |
| # address since this is registered in ZK and handed out to clients. If |
| # host DNS isn't setup, we shouldn't use hostnames -- IP addresses must be |
| # used to support clients running on the host. |
| zookeeper_connect = zookeepers.map{ |zk_addr| zk_addr + ":2181"}.join(",") |
| broker.vm.provision "shell", path: "vagrant/base.sh" |
| broker.vm.provision "shell", path: "vagrant/bk.sh", :args => [i.to_s, enable_dns ? name : ip_address, zookeeper_connect] |
| end |
| } |
| end |