132 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			132 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
| { config, lib, pkgs, ... }:
 | |
| 
 | |
| with lib;
 | |
| 
 | |
| let
 | |
|   cfg = config.services.mesos.slave;
 | |
| 
 | |
|   mkAttributes =
 | |
|     attrs: concatStringsSep ";" (mapAttrsToList
 | |
|                                    (k: v: "${k}:${v}")
 | |
|                                    (filterAttrs (k: v: v != null) attrs));
 | |
|   attribsArg = optionalString (cfg.attributes != {})
 | |
|                               "--attributes=${mkAttributes cfg.attributes}";
 | |
| 
 | |
|   containerizers = [ "mesos" ] ++ (optional cfg.withDocker "docker");
 | |
| 
 | |
| in {
 | |
| 
 | |
|   options.services.mesos = {
 | |
|     slave = {
 | |
|       enable = mkOption {
 | |
|         description = "Whether to enable the Mesos Slave.";
 | |
|         default = false;
 | |
|         type = types.bool;
 | |
|       };
 | |
| 
 | |
|       ip = mkOption {
 | |
|         description = "IP address to listen on.";
 | |
|         default = "0.0.0.0";
 | |
|         type = types.string;
 | |
|       };
 | |
| 
 | |
|       port = mkOption {
 | |
|         description = "Port to listen on.";
 | |
|         default = 5051;
 | |
|         type = types.int;
 | |
|       };
 | |
| 
 | |
|       master = mkOption {
 | |
|         description = ''
 | |
|           May be one of:
 | |
|             zk://host1:port1,host2:port2,.../path
 | |
|             zk://username:password@host1:port1,host2:port2,.../path
 | |
|         '';
 | |
|         type = types.str;
 | |
|       };
 | |
| 
 | |
|       withHadoop = mkOption {
 | |
|         description = "Add the HADOOP_HOME to the slave.";
 | |
|         default = false;
 | |
|         type = types.bool;
 | |
|       };
 | |
| 
 | |
|       withDocker = mkOption {
 | |
|         description = "Enable the docker containerizer.";
 | |
|         default = config.virtualisation.docker.enable;
 | |
|         type = types.bool;
 | |
|       };
 | |
| 
 | |
|       workDir = mkOption {
 | |
|         description = "The Mesos work directory.";
 | |
|         default = "/var/lib/mesos/slave";
 | |
|         type = types.str;
 | |
|       };
 | |
| 
 | |
|       extraCmdLineOptions = mkOption {
 | |
|         description = ''
 | |
|           Extra command line options for Mesos Slave.
 | |
| 
 | |
|           See https://mesos.apache.org/documentation/latest/configuration/
 | |
|         '';
 | |
|         default = [ "" ];
 | |
|         type = types.listOf types.str;
 | |
|         example = [ "--gc_delay=3days" ];
 | |
|       };
 | |
| 
 | |
|       logLevel = mkOption {
 | |
|         description = ''
 | |
|           The logging level used. Possible values:
 | |
|             'INFO', 'WARNING', 'ERROR'
 | |
|         '';
 | |
|         default = "INFO";
 | |
|         type = types.str;
 | |
|       };
 | |
| 
 | |
|       attributes = mkOption {
 | |
|         description = ''
 | |
|           Machine attributes for the slave instance.
 | |
| 
 | |
|           Use caution when changing this; you may need to manually reset slave
 | |
|           metadata before the slave can re-register.
 | |
|         '';
 | |
|         default = {};
 | |
|         type = types.attrsOf types.str;
 | |
|         example = { rack = "aa";
 | |
|                     host = "aabc123";
 | |
|                     os = "nixos"; };
 | |
|       };
 | |
|     };
 | |
| 
 | |
|   };
 | |
| 
 | |
| 
 | |
|   config = mkIf cfg.enable {
 | |
|     systemd.services.mesos-slave = {
 | |
|       description = "Mesos Slave";
 | |
|       wantedBy = [ "multi-user.target" ];
 | |
|       after = [ "network-interfaces.target" ];
 | |
|       environment.MESOS_CONTAINERIZERS = concatStringsSep "," containerizers;
 | |
|       serviceConfig = {
 | |
|         ExecStart = ''
 | |
|           ${pkgs.mesos}/bin/mesos-slave \
 | |
|             --ip=${cfg.ip} \
 | |
|             --port=${toString cfg.port} \
 | |
|             --master=${cfg.master} \
 | |
|             --work_dir=${cfg.workDir} \
 | |
|             --logging_level=${cfg.logLevel} \
 | |
|             ${attribsArg} \
 | |
|             ${optionalString cfg.withHadoop "--hadoop-home=${pkgs.hadoop}"} \
 | |
|             ${optionalString cfg.withDocker "--docker=${pkgs.docker}/libexec/docker/docker"} \
 | |
|             ${toString cfg.extraCmdLineOptions}
 | |
|         '';
 | |
|         PermissionsStartOnly = true;
 | |
|       };
 | |
|       preStart = ''
 | |
|         mkdir -m 0700 -p ${cfg.workDir}
 | |
|       '';
 | |
|     };
 | |
|   };
 | |
| 
 | |
| }
 | 
