def reset_kubeadm(): """Uninstall kubernetes on a node. Runs `kubeadm reset` on the specified machine in order to remove the kubernetes services and undo all configuration set by `kubeadm init`. """ # Get script from path. script = os.path.join(os.path.dirname(__file__), 'mega-reset.sh') ctx.download_resource( os.path.join('scripts', 'mega-reset.sh'), script ) # Get worker. conn = MistConnectionClient() machine = conn.get_machine( cloud_id=ctx.instance.runtime_properties['cloud_id'], machine_id=ctx.instance.runtime_properties['machine_id'], ) ctx.logger.info('Running "kubeadm reset" on %s', machine) _add_run_remove_script( cloud_id=machine.cloud.id, machine_id=machine.id, script_path=os.path.abspath(script), script_name='kubeadm_reset_%s' % random_string(length=4) )
def configure_kubernetes_master(): """Configure the kubernetes master. Sets up the master node and stores the necessary settings inside the node instance's runtime properties, which are required by worker nodes in order to join the kubernetes cluster. """ ctx.logger.info('Setting up kubernetes master node') prepare_kubernetes_script() conn = MistConnectionClient() machine = conn.get_machine( cloud_id=ctx.instance.runtime_properties['cloud_id'], machine_id=ctx.instance.runtime_properties['machine_id'], ) # Token for secure master-worker communication. token = '%s.%s' % (random_string(length=6), random_string(length=16)) ctx.instance.runtime_properties['master_token'] = token.lower() # Store kubernetes dashboard credentials in runtime properties. ctx.instance.runtime_properties.update({ 'auth_user': ctx.node.properties['auth_user'], 'auth_pass': ctx.node.properties['auth_pass'] or random_string(10), }) ctx.logger.info('Installing kubernetes on master node') # Prepare script parameters. params = "-u '%s' " % ctx.instance.runtime_properties['auth_user'] params += "-p '%s' " % ctx.instance.runtime_properties['auth_pass'] params += "-t '%s' " % ctx.instance.runtime_properties['master_token'] params += "-r 'master'" # Run the script. script = conn.client.run_script( script_id=ctx.instance.runtime_properties['script_id'], su=True, machine_id=machine.id, cloud_id=machine.cloud.id, script_params=params, ) ctx.instance.runtime_properties['job_id'] = script['job_id']
def configure_kubernetes_master(): """Configure the kubernetes master. Sets up the master node and stores the necessary settings inside the node instance's runtime properties, which are required by worker nodes in order to join the kubernetes cluster. """ ctx.logger.info('Setting up kubernetes master node') prepare_kubernetes_script() conn = MistConnectionClient() machine = conn.get_machine( cloud_id=ctx.instance.runtime_properties['cloud_id'], machine_id=ctx.instance.runtime_properties['machine_id'], ) # Token for secure master-worker communication. token = '%s.%s' % (random_string(length=6), random_string(length=16)) ctx.instance.runtime_properties['master_token'] = token.lower() # Store kubernetes dashboard credentials in runtime properties. ctx.instance.runtime_properties.update({ 'auth_user': ctx.node.properties['auth_user'], 'auth_pass': ctx.node.properties['auth_pass'] or random_string(10), }) ctx.logger.info('Installing kubernetes on master node') # Prepare script parameters. params = "-u '%s' " % ctx.instance.runtime_properties['auth_user'] params += "-p '%s' " % ctx.instance.runtime_properties['auth_pass'] params += "-t '%s' " % ctx.instance.runtime_properties['master_token'] params += "-r 'master'" # Run the script. script = conn.client.run_script( script_id=ctx.instance.runtime_properties['script_id'], su=True, machine_id=machine.id, cloud_id=machine.cloud.id, script_params=params, ) ctx.instance.runtime_properties['job_id'] = script['job_id']
def get_master_init_args(): """Return the arguments required to install the kubernetes master.""" ctx.logger.info('Preparing cloud-init for kubernetes master') # Token for secure master-worker communication. token = '%s.%s' % (random_string(length=6), random_string(length=16)) ctx.instance.runtime_properties['master_token'] = token.lower() # Store kubernetes dashboard credentials in runtime properties. ctx.instance.runtime_properties.update({ 'auth_user': ctx.node.properties['auth_user'], 'auth_pass': ctx.node.properties['auth_pass'] or random_string(10), }) arguments = "-u '%s' " % ctx.instance.runtime_properties['auth_user'] arguments += "-p '%s' " % ctx.instance.runtime_properties['auth_pass'] arguments += "-t '%s' " % ctx.instance.runtime_properties['master_token'] arguments += "-r 'master'" return arguments
def drain_and_remove(): """Mark the node as unschedulable, evict all pods, and remove it. Runs `kubectl drain` and `kubectl delete nodes` on the kubernetes master in order to drain and afterwards remove the specified node from the cluster. """ if ctx.node.properties['master']: # FIXME Is this necessary? return # Get master instance. master = ctx.instance.relationships[0]._target.instance # Render script. script = os.path.join(os.path.dirname(__file__), 'drain-node.sh') ctx.download_resource_and_render( os.path.join('scripts', 'drain-node.sh'), script, template_variables={ 'server_ip': master.runtime_properties.get('server_ip',''), 'auth_user': master.runtime_properties['auth_user'], 'auth_pass': master.runtime_properties['auth_pass'], 'hostname': ctx.instance.runtime_properties.get('machine_name', '').lower() }, ) conn = MistConnectionClient() machine = conn.get_machine( cloud_id=master.runtime_properties['cloud_id'], machine_id=master.runtime_properties['machine_id'], ) ctx.logger.info('Running "kubectl drain && kubectl delete" on %s', machine) _add_run_remove_script( cloud_id=machine.cloud.id, machine_id=machine.id, script_path=os.path.abspath(script), script_name='kubectl_drain_%s' % random_string(length=4) )
def prepare_kubernetes_script(): """Upload kubernetes installation script, if missing. This method is executed at the very beginning, in a pre-configuration phase, to make sure that the kubernetes installation script has been uploaded to mist.io. This method is meant to be invoked early on by: configure_kubernetes_master() configure_kubernetes_worker() The script_id inside each instance's runtime properties is used later on in order to configure kubernetes on the provisioned machines. """ if ctx.instance.runtime_properties.get('script_id'): ctx.logger.info('Kubernetes installation script already exists') else: ctx.logger.info('Uploading fresh kubernetes installation script') # If a script_id does not exist in the node instance's runtime # properties, perhaps because this is the first node that is being # configured, load the script from file, upload it to mist.io, and # run it over ssh. client = MistConnectionClient().client script = os.path.join(os.path.dirname(__file__), 'mega-deploy.sh') ctx.download_resource( os.path.join('scripts', 'mega-deploy.sh'), script ) with open(os.path.abspath(script)) as fobj: script = fobj.read() script = client.add_script( name='install_kubernetes_%s' % random_string(length=4), script=script, location_type='inline', exec_type='executable' ) ctx.instance.runtime_properties['script_id'] = script['id']
def prepare_kubernetes_script(): """Upload kubernetes installation script, if missing. This method is executed at the very beginning, in a pre-configuration phase, to make sure that the kubernetes installation script has been uploaded to mist.io. This method is meant to be invoked early on by: configure_kubernetes_master() configure_kubernetes_worker() The script_id inside each instance's runtime properties is used later on in order to configure kubernetes on the provisioned machines. """ if ctx.instance.runtime_properties.get('script_id'): ctx.logger.info('Kubernetes installation script already exists') else: ctx.logger.info('Uploading fresh kubernetes installation script') # If a script_id does not exist in the node instance's runtime # properties, perhaps because this is the first node that is being # configured, load the script from file, upload it to mist.io, and # run it over ssh. client = MistConnectionClient().client script = os.path.join(os.path.dirname(__file__), 'mega-deploy.sh') ctx.download_resource(os.path.join('scripts', 'mega-deploy.sh'), script) with open(os.path.abspath(script)) as fobj: script = fobj.read() script = client.add_script(name='install_kubernetes_%s' % random_string(length=4), script=script, location_type='inline', exec_type='executable') ctx.instance.runtime_properties['script_id'] = script['id']