commit 9ca4363191f05147d0bd23ffb61838d9c1ee890e · pyrox.dev/nixpkgs

nixos/modules/services/cluster/hadoop/conf.nix

···

       35
        
         cp ${siteXml "hdfs-site.xml" cfg.hdfsSite}/* $out/

     

       36
        
         cp ${siteXml "mapred-site.xml" cfg.mapredSite}/* $out/

     

       37
        
         cp ${siteXml "yarn-site.xml" cfg.yarnSite}/* $out/

     

       0
        
       
     

       38
        
         cp ${cfgFile "container-executor.cfg" cfg.containerExecutorCfg}/* $out/

     

       39
        
         cp ${pkgs.writeTextDir "hadoop-user-functions.sh" userFunctions}/* $out/

     

       40
        
         cp ${pkgs.writeTextDir "hadoop-env.sh" hadoopEnv}/* $out/

···

       35
        
         cp ${siteXml "hdfs-site.xml" cfg.hdfsSite}/* $out/

     

       36
        
         cp ${siteXml "mapred-site.xml" cfg.mapredSite}/* $out/

     

       37
        
         cp ${siteXml "yarn-site.xml" cfg.yarnSite}/* $out/

     

       38
       +
         cp ${siteXml "httpfs-site.xml" cfg.httpfsSite}/* $out/

     

       39
        
         cp ${cfgFile "container-executor.cfg" cfg.containerExecutorCfg}/* $out/

     

       40
        
         cp ${pkgs.writeTextDir "hadoop-user-functions.sh" userFunctions}/* $out/

     

       41
        
         cp ${pkgs.writeTextDir "hadoop-env.sh" hadoopEnv}/* $out/

+13 -1

nixos/modules/services/cluster/hadoop/default.nix

···

       70
        
             description = "Hadoop yarn-site.xml definition";

     

       71
        
           };

     

       72
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       73
        
           log4jProperties = mkOption {

     

       74
        
             default = "${cfg.package}/lib/${cfg.package.untarDir}/etc/hadoop/log4j.properties";

     

       75
        
             type = types.path;

     
···

       118
        
       

     

       119
        
         config = mkMerge [

     

       120
        
           (mkIf (builtins.hasAttr "yarn" config.users.users ||

     

       121
       -
                  builtins.hasAttr "hdfs" config.users.users) {

     

       0
        
       
     

       122
        
             users.groups.hadoop = {

     

       123
        
               gid = config.ids.gids.hadoop;

     

       124
        
             };

···

       70
        
             description = "Hadoop yarn-site.xml definition";

     

       71
        
           };

     

       72
        
       

     

       73
       +
           httpfsSite = mkOption {

     

       74
       +
             default = { };

     

       75
       +
             type = types.attrsOf types.anything;

     

       76
       +
             example = literalExpression ''

     

       77
       +
               {

     

       78
       +
                 "hadoop.http.max.threads" = 500;

     

       79
       +
               }

     

       80
       +
             '';

     

       81
       +
             description = "Hadoop httpfs-site.xml definition";

     

       82
       +
           };

     

       83
       +
       

     

       84
        
           log4jProperties = mkOption {

     

       85
        
             default = "${cfg.package}/lib/${cfg.package.untarDir}/etc/hadoop/log4j.properties";

     

       86
        
             type = types.path;

     
···

       129
        
       

     

       130
        
         config = mkMerge [

     

       131
        
           (mkIf (builtins.hasAttr "yarn" config.users.users ||

     

       132
       +
                  builtins.hasAttr "hdfs" config.users.users ||

     

       133
       +
                  builtins.hasAttr "httpfs" config.users.users) {

     

       134
        
             users.groups.hadoop = {

     

       135
        
               gid = config.ids.gids.hadoop;

     

       136
        
             };

+49 -1

nixos/modules/services/cluster/hadoop/hdfs.nix

···

       86
        
             };

     

       87
        
             inherit restartIfChanged;

     

       88
        
           };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       89
        
         };

     

       90
        
       

     

       91
        
         config = mkMerge [

     
···

       166
        
               };

     

       167
        
             };

     

       168
        
           })

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       169
        
           (mkIf (

     

       170
        
               cfg.hdfs.namenode.enabled || cfg.hdfs.datanode.enabled || cfg.hdfs.journalnode.enabled || cfg.hdfs.zkfc.enabled

     

       171
        
           ) {

     
···

       175
        
               uid = config.ids.uids.hdfs;

     

       176
        
             };

     

       177
        
           })

     

       178
       -
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       179
        
         ];

     

       180
        
       }

···

       86
        
             };

     

       87
        
             inherit restartIfChanged;

     

       88
        
           };

     

       89
       +
           httpfs = {

     

       90
       +
             enabled = mkOption {

     

       91
       +
               type = types.bool;

     

       92
       +
               default = false;

     

       93
       +
               description = ''

     

       94
       +
                 Whether to run the HDFS httpfs failover controller

     

       95
       +
               '';

     

       96
       +
             };

     

       97
       +
             tempPath = mkOption {

     

       98
       +
               type = types.path;

     

       99
       +
               default = "/tmp/hadoop/httpfs";

     

       100
       +
               description = ''

     

       101
       +
                 HTTPFS_TEMP path used by HTTPFS

     

       102
       +
               '';

     

       103
       +
             };

     

       104
       +
             inherit restartIfChanged;

     

       105
       +
           };

     

       106
        
         };

     

       107
        
       

     

       108
        
         config = mkMerge [

     
···

       183
        
               };

     

       184
        
             };

     

       185
        
           })

     

       186
       +
           (mkIf cfg.hdfs.httpfs.enabled {

     

       187
       +
             systemd.services.hdfs-httpfs = {

     

       188
       +
               description = "Hadoop httpfs";

     

       189
       +
               wantedBy = [ "multi-user.target" ];

     

       190
       +
               inherit (cfg.hdfs.httpfs) restartIfChanged;

     

       191
       +
       

     

       192
       +
               environment = {

     

       193
       +
                 HTTPFS_TEMP = cfg.hdfs.httpfs.tempPath;

     

       194
       +
               };

     

       195
       +
       

     

       196
       +
               preStart = ''

     

       197
       +
                 mkdir -p $HTTPFS_TEMP

     

       198
       +
               '';

     

       199
       +
       

     

       200
       +
               serviceConfig = {

     

       201
       +
                 User = "httpfs";

     

       202
       +
                 SyslogIdentifier = "hdfs-httpfs";

     

       203
       +
                 ExecStart = "${cfg.package}/bin/hdfs --config ${hadoopConf} httpfs";

     

       204
       +
                 Restart = "always";

     

       205
       +
               };

     

       206
       +
             };

     

       207
       +
             networking.firewall.allowedTCPPorts = (mkIf cfg.hdfs.datanode.openFirewall [

     

       208
       +
               14000 # httpfs.http.port

     

       209
       +
             ]);

     

       210
       +
           })

     

       211
        
           (mkIf (

     

       212
        
               cfg.hdfs.namenode.enabled || cfg.hdfs.datanode.enabled || cfg.hdfs.journalnode.enabled || cfg.hdfs.zkfc.enabled

     

       213
        
           ) {

     
···

       217
        
               uid = config.ids.uids.hdfs;

     

       218
        
             };

     

       219
        
           })

     

       220
       +
           (mkIf cfg.hdfs.httpfs.enabled {

     

       221
       +
             users.users.httpfs = {

     

       222
       +
               description = "Hadoop HTTPFS user";

     

       223
       +
               group = "hadoop";

     

       224
       +
               isSystemUser = true;

     

       225
       +
             };

     

       226
       +
           })

     

       227
        
         ];

     

       228
        
       }

+15 -8

nixos/tests/hadoop/hdfs.nix

···

       2
        
       import ../make-test-python.nix ({...}: {

     

       3
        
         nodes = {

     

       4
        
           namenode = {pkgs, ...}: {

     

       0
        
       
     

       5
        
             services.hadoop = {

     

       6
        
               package = pkgs.hadoop;

     

       7
       -
               hdfs.namenode = {

     

       8
       -
                 enabled = true;

     

       9
       -
                 formatOnInit = true;

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       10
        
               };

     

       11
        
               coreSite = {

     

       12
        
                 "fs.defaultFS" = "hdfs://namenode:8020";

     

       13
       -
               };

     

       14
       -
               hdfsSite = {

     

       15
       -
                 "dfs.replication" = 1;

     

       16
       -
                 "dfs.namenode.rpc-bind-host" = "0.0.0.0";

     

       17
       -
                 "dfs.namenode.http-bind-host" = "0.0.0.0";

     

       18
        
               };

     

       19
        
             };

     

       20
        
           };

     
···

       24
        
               hdfs.datanode.enabled = true;

     

       25
        
               coreSite = {

     

       26
        
                 "fs.defaultFS" = "hdfs://namenode:8020";

     

       0
        
       
     

       0
        
       
     

       27
        
               };

     

       28
        
             };

     

       29
        
           };

     
···

       49
        
           datanode.succeed("sudo -u hdfs hdfs dfsadmin -safemode wait")

     

       50
        
           datanode.succeed("echo testfilecontents | sudo -u hdfs hdfs dfs -put - /testfile")

     

       51
        
           assert "testfilecontents" in datanode.succeed("sudo -u hdfs hdfs dfs -cat /testfile")

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       52
        
         '';

     

       53
        
       })

···

       2
        
       import ../make-test-python.nix ({...}: {

     

       3
        
         nodes = {

     

       4
        
           namenode = {pkgs, ...}: {

     

       5
       +
             virtualisation.memorySize = 1024;

     

       6
        
             services.hadoop = {

     

       7
        
               package = pkgs.hadoop;

     

       8
       +
               hdfs = {

     

       9
       +
                 namenode = {

     

       10
       +
                   enabled = true;

     

       11
       +
                   formatOnInit = true;

     

       12
       +
                 };

     

       13
       +
                 httpfs.enabled = true;

     

       14
        
               };

     

       15
        
               coreSite = {

     

       16
        
                 "fs.defaultFS" = "hdfs://namenode:8020";

     

       17
       +
                 "hadoop.proxyuser.httpfs.groups" = "*";

     

       18
       +
                 "hadoop.proxyuser.httpfs.hosts" = "*";

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       19
        
               };

     

       20
        
             };

     

       21
        
           };

     
···

       25
        
               hdfs.datanode.enabled = true;

     

       26
        
               coreSite = {

     

       27
        
                 "fs.defaultFS" = "hdfs://namenode:8020";

     

       28
       +
                 "hadoop.proxyuser.httpfs.groups" = "*";

     

       29
       +
                 "hadoop.proxyuser.httpfs.hosts" = "*";

     

       30
        
               };

     

       31
        
             };

     

       32
        
           };

     
···

       52
        
           datanode.succeed("sudo -u hdfs hdfs dfsadmin -safemode wait")

     

       53
        
           datanode.succeed("echo testfilecontents | sudo -u hdfs hdfs dfs -put - /testfile")

     

       54
        
           assert "testfilecontents" in datanode.succeed("sudo -u hdfs hdfs dfs -cat /testfile")

     

       55
       +
       

     

       56
       +
           namenode.wait_for_unit("hdfs-httpfs")

     

       57
       +
           namenode.wait_for_open_port(14000)

     

       58
       +
           assert "testfilecontents" in datanode.succeed("curl -f \"http://namenode:14000/webhdfs/v1/testfile?user.name=hdfs&op=OPEN\" 2>&1")

     

       59
        
         '';

     

       60
        
       })