commit 39c007ce9cc3f46db70bde794323b6cbfccfe62a · pyrox.dev/nixpkgs

+73 -3

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

···

       24
        
                 Whether to run the HDFS NameNode

     

       25
        
               '';

     

       26
        
             };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       27
        
             inherit restartIfChanged;

     

       28
        
             openFirewall = mkOption {

     

       29
        
               type = types.bool;

     
···

       50
        
               '';

     

       51
        
             };

     

       52
        
           };

     

       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
        
       
     

       0
        
       
     

       0
        
       
     

       53
        
         };

     

       54
        
       

     

       55
        
         config = mkMerge [

     
···

       59
        
               wantedBy = [ "multi-user.target" ];

     

       60
        
               inherit (cfg.hdfs.namenode) restartIfChanged;

     

       61
        
       

     

       62
       -
               preStart = ''

     

       63
        
                 ${cfg.package}/bin/hdfs --config ${hadoopConf} namenode -format -nonInteractive || true

     

       64
       -
               '';

     

       65
        
       

     

       66
        
               serviceConfig = {

     

       67
        
                 User = "hdfs";

     
···

       74
        
             networking.firewall.allowedTCPPorts = (mkIf cfg.hdfs.namenode.openFirewall [

     

       75
        
               9870 # namenode.http-address

     

       76
        
               8020 # namenode.rpc-address

     

       0
        
       
     

       77
        
             ]);

     

       78
        
           })

     

       79
        
           (mkIf cfg.hdfs.datanode.enabled {

     
···

       96
        
               9867 # datanode.ipc.address

     

       97
        
             ]);

     

       98
        
           })

     

       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
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       99
        
           (mkIf (

     

       100
       -
               cfg.hdfs.namenode.enabled || cfg.hdfs.datanode.enabled

     

       101
        
           ) {

     

       102
        
             users.users.hdfs = {

     

       103
        
               description = "Hadoop HDFS user";

···

       24
        
                 Whether to run the HDFS NameNode

     

       25
        
               '';

     

       26
        
             };

     

       27
       +
             formatOnInit = mkOption {

     

       28
       +
               type = types.bool;

     

       29
       +
               default = false;

     

       30
       +
               description = ''

     

       31
       +
                 Format HDFS namenode on first start. This is useful for quickly spinning up ephemeral HDFS clusters with a single namenode.

     

       32
       +
                 For HA clusters, initialization involves multiple steps across multiple nodes. Follow [this guide](https://hadoop.apache.org/docs/stable/hadoop-project-dist/hadoop-hdfs/HDFSHighAvailabilityWithQJM.html)

     

       33
       +
                 to initialize an HA cluster manually.

     

       34
       +
               '';

     

       35
       +
             };

     

       36
        
             inherit restartIfChanged;

     

       37
        
             openFirewall = mkOption {

     

       38
        
               type = types.bool;

     
···

       59
        
               '';

     

       60
        
             };

     

       61
        
           };

     

       62
       +
           journalnode = {

     

       63
       +
             enabled = mkOption {

     

       64
       +
               type = types.bool;

     

       65
       +
               default = false;

     

       66
       +
               description = ''

     

       67
       +
                 Whether to run the HDFS JournalNode

     

       68
       +
               '';

     

       69
       +
             };

     

       70
       +
             inherit restartIfChanged;

     

       71
       +
             openFirewall = mkOption {

     

       72
       +
               type = types.bool;

     

       73
       +
               default = true;

     

       74
       +
               description = ''

     

       75
       +
                 Open firewall ports for journalnode

     

       76
       +
               '';

     

       77
       +
             };

     

       78
       +
           };

     

       79
       +
           zkfc = {

     

       80
       +
             enabled = mkOption {

     

       81
       +
               type = types.bool;

     

       82
       +
               default = false;

     

       83
       +
               description = ''

     

       84
       +
                 Whether to run the HDFS ZooKeeper failover controller

     

       85
       +
               '';

     

       86
       +
             };

     

       87
       +
             inherit restartIfChanged;

     

       88
       +
           };

     

       89
        
         };

     

       90
        
       

     

       91
        
         config = mkMerge [

     
···

       95
        
               wantedBy = [ "multi-user.target" ];

     

       96
        
               inherit (cfg.hdfs.namenode) restartIfChanged;

     

       97
        
       

     

       98
       +
               preStart = (mkIf cfg.hdfs.namenode.formatOnInit ''

     

       99
        
                 ${cfg.package}/bin/hdfs --config ${hadoopConf} namenode -format -nonInteractive || true

     

       100
       +
               '');

     

       101
        
       

     

       102
        
               serviceConfig = {

     

       103
        
                 User = "hdfs";

     
···

       110
        
             networking.firewall.allowedTCPPorts = (mkIf cfg.hdfs.namenode.openFirewall [

     

       111
        
               9870 # namenode.http-address

     

       112
        
               8020 # namenode.rpc-address

     

       113
       +
               8022 # namenode. servicerpc-address

     

       114
        
             ]);

     

       115
        
           })

     

       116
        
           (mkIf cfg.hdfs.datanode.enabled {

     
···

       133
        
               9867 # datanode.ipc.address

     

       134
        
             ]);

     

       135
        
           })

     

       136
       +
           (mkIf cfg.hdfs.journalnode.enabled {

     

       137
       +
             systemd.services.hdfs-journalnode = {

     

       138
       +
               description = "Hadoop HDFS JournalNode";

     

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

     

       140
       +
               inherit (cfg.hdfs.journalnode) restartIfChanged;

     

       141
       +
       

     

       142
       +
               serviceConfig = {

     

       143
       +
                 User = "hdfs";

     

       144
       +
                 SyslogIdentifier = "hdfs-journalnode";

     

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

     

       146
       +
                 Restart = "always";

     

       147
       +
               };

     

       148
       +
             };

     

       149
       +
       

     

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

     

       151
       +
               8480 # dfs.journalnode.http-address

     

       152
       +
               8485 # dfs.journalnode.rpc-address

     

       153
       +
             ]);

     

       154
       +
           })

     

       155
       +
           (mkIf cfg.hdfs.zkfc.enabled {

     

       156
       +
             systemd.services.hdfs-zkfc = {

     

       157
       +
               description = "Hadoop HDFS ZooKeeper failover controller";

     

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

     

       159
       +
               inherit (cfg.hdfs.zkfc) restartIfChanged;

     

       160
       +
       

     

       161
       +
               serviceConfig = {

     

       162
       +
                 User = "hdfs";

     

       163
       +
                 SyslogIdentifier = "hdfs-zkfc";

     

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

     

       165
       +
                 Restart = "always";

     

       166
       +
               };

     

       167
       +
             };

     

       168
       +
           })

     

       169
        
           (mkIf (

     

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

     

       171
        
           ) {

     

       172
        
             users.users.hdfs = {

     

       173
        
               description = "Hadoop HDFS user";

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

···

       91
        
               8030 # resourcemanager.scheduler.address

     

       92
        
               8031 # resourcemanager.resource-tracker.address

     

       93
        
               8032 # resourcemanager.address

     

       0
        
       
     

       94
        
             ]);

     

       95
        
           })

     

       96

···

       91
        
               8030 # resourcemanager.scheduler.address

     

       92
        
               8031 # resourcemanager.resource-tracker.address

     

       93
        
               8032 # resourcemanager.address

     

       94
       +
               8033 # resourcemanager.admin.address

     

       95
        
             ]);

     

       96
        
           })

     

       97

+198 -38

nixos/tests/hadoop/hadoop.nix

···

       0
        
       
     

       0
        
       
     

       0
        
       
     

       1
        
       import ../make-test-python.nix ({pkgs, ...}: {

     

       2
        
       

     

       3
        
         nodes = let

     

       4
        
           package = pkgs.hadoop;

     

       5
        
           coreSite = {

     

       6
       -
             "fs.defaultFS" = "hdfs://master";

     

       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
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       7
        
           };

     

       8
        
         in {

     

       9
       -
           master = {pkgs, options, ...}: {

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       10
        
             services.hadoop = {

     

       11
       -
               inherit package coreSite;

     

       12
        
               hdfs.namenode.enabled = true;

     

       13
       -
               yarn.resourcemanager.enabled = true;

     

       14
        
             };

     

       15
       -
             virtualisation.memorySize = 1024;

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       16
        
           };

     

       17
        
       

     

       18
       -
           worker = {pkgs, options, ...}: {

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       19
        
             services.hadoop = {

     

       20
       -
               inherit package coreSite;

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       21
        
               hdfs.datanode.enabled = true;

     

       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
        
       
     

       22
        
               yarn.nodemanager.enabled = true;

     

       23
       -
               yarnSite = options.services.hadoop.yarnSite.default // {

     

       24
       -
                 "yarn.resourcemanager.hostname" = "master";

     

       25
       -
               };

     

       26
        
             };

     

       27
       -
             virtualisation.memorySize = 2048;

     

       28
        
           };

     

       29
        
         };

     

       30
        
       

     

       31
        
         testScript = ''

     

       32
        
           start_all()

     

       33
        
       

     

       34
       -
           master.wait_for_unit("network.target")

     

       35
       -
           master.wait_for_unit("hdfs-namenode")

     

       36
        
       

     

       37
       -
           master.wait_for_open_port(8020)

     

       38
       -
           master.wait_for_open_port(9870)

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       39
        
       

     

       40
       -
           worker.wait_for_unit("network.target")

     

       41
       -
           worker.wait_for_unit("hdfs-datanode")

     

       42
       -
           worker.wait_for_open_port(9864)

     

       43
       -
           worker.wait_for_open_port(9866)

     

       44
       -
           worker.wait_for_open_port(9867)

     

       45
        
       

     

       46
       -
           master.succeed("curl -f http://worker:9864")

     

       47
       -
           worker.succeed("curl -f http://master:9870")

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       48
        
       

     

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

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       50
        
       

     

       51
       -
           master.wait_for_unit("yarn-resourcemanager")

     

       0
        
       
     

       52
        
       

     

       53
       -
           master.wait_for_open_port(8030)

     

       54
       -
           master.wait_for_open_port(8031)

     

       55
       -
           master.wait_for_open_port(8032)

     

       56
       -
           master.wait_for_open_port(8088)

     

       57
       -
           worker.succeed("curl -f http://master:8088")

     

       0
        
       
     

       58
        
       

     

       59
       -
           worker.wait_for_unit("yarn-nodemanager")

     

       60
       -
           worker.wait_for_open_port(8042)

     

       61
       -
           worker.wait_for_open_port(8040)

     

       62
       -
           master.succeed("curl -f http://worker:8042")

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       63
        
       

     

       64
       -
           assert "Total Nodes:1" in worker.succeed("yarn node -list")

     

       0
        
       
     

       0
        
       
     

       65
        
       

     

       66
       -
           assert "Estimated value of Pi is" in worker.succeed("HADOOP_USER_NAME=hdfs yarn jar $(readlink $(which yarn) | sed -r 's~bin/yarn~lib/hadoop-*/share/hadoop/mapreduce/hadoop-mapreduce-examples-*.jar~g') pi 2 10")

     

       67
       -
           assert "SUCCEEDED" in worker.succeed("yarn application -list -appStates FINISHED")

     

       68
       -
           worker.succeed("sudo -u hdfs hdfs dfs -ls / | systemd-cat")

     

       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
        
       
     

       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
        
       
     

       0
        
       
     

       69
        
         '';

     

       70
       -
        })

···

       1
       +
       # This test is very comprehensive. It tests whether all hadoop services work well with each other.

     

       2
       +
       # Run this when updating the Hadoop package or making significant changes to the hadoop module.

     

       3
       +
       # For a more basic test, see hdfs.nix and yarn.nix

     

       4
        
       import ../make-test-python.nix ({pkgs, ...}: {

     

       5
        
       

     

       6
        
         nodes = let

     

       7
        
           package = pkgs.hadoop;

     

       8
        
           coreSite = {

     

       9
       +
             "fs.defaultFS" = "hdfs://ns1";

     

       10
       +
           };

     

       11
       +
           hdfsSite = {

     

       12
       +
             "dfs.namenode.rpc-bind-host" = "0.0.0.0";

     

       13
       +
             "dfs.namenode.http-bind-host" = "0.0.0.0";

     

       14
       +
             "dfs.namenode.servicerpc-bind-host" = "0.0.0.0";

     

       15
       +
       

     

       16
       +
             # HA Quorum Journal Manager configuration

     

       17
       +
             "dfs.nameservices" = "ns1";

     

       18
       +
             "dfs.ha.namenodes.ns1" = "nn1,nn2";

     

       19
       +
             "dfs.namenode.shared.edits.dir.ns1.nn1" = "qjournal://jn1:8485;jn2:8485;jn3:8485/ns1";

     

       20
       +
             "dfs.namenode.shared.edits.dir.ns1.nn2" = "qjournal://jn1:8485;jn2:8485;jn3:8485/ns1";

     

       21
       +
             "dfs.namenode.rpc-address.ns1.nn1" = "nn1:8020";

     

       22
       +
             "dfs.namenode.rpc-address.ns1.nn2" = "nn2:8020";

     

       23
       +
             "dfs.namenode.servicerpc-address.ns1.nn1" = "nn1:8022";

     

       24
       +
             "dfs.namenode.servicerpc-address.ns1.nn2" = "nn2:8022";

     

       25
       +
             "dfs.namenode.http-address.ns1.nn1" = "nn1:9870";

     

       26
       +
             "dfs.namenode.http-address.ns1.nn2" = "nn2:9870";

     

       27
       +
       

     

       28
       +
             # Automatic failover configuration

     

       29
       +
             "dfs.client.failover.proxy.provider.ns1" = "org.apache.hadoop.hdfs.server.namenode.ha.ConfiguredFailoverProxyProvider";

     

       30
       +
             "dfs.ha.automatic-failover.enabled.ns1" = "true";

     

       31
       +
             "dfs.ha.fencing.methods" = "shell(true)";

     

       32
       +
             "ha.zookeeper.quorum" = "zk1:2181";

     

       33
       +
           };

     

       34
       +
           yarnSiteHA = {

     

       35
       +
             "yarn.resourcemanager.zk-address" = "zk1:2181";

     

       36
       +
             "yarn.resourcemanager.ha.enabled" = "true";

     

       37
       +
             "yarn.resourcemanager.ha.rm-ids" = "rm1,rm2";

     

       38
       +
             "yarn.resourcemanager.hostname.rm1" = "rm1";

     

       39
       +
             "yarn.resourcemanager.hostname.rm2" = "rm2";

     

       40
       +
             "yarn.resourcemanager.ha.automatic-failover.enabled" = "true";

     

       41
       +
             "yarn.resourcemanager.cluster-id" = "cluster1";

     

       42
       +
             # yarn.resourcemanager.webapp.address needs to be defined even though yarn.resourcemanager.hostname is set. This shouldn't be necessary, but there's a bug in

     

       43
       +
             # hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-web-proxy/src/main/java/org/apache/hadoop/yarn/server/webproxy/amfilter/AmFilterInitializer.java:70

     

       44
       +
             # that causes AM containers to fail otherwise.

     

       45
       +
             "yarn.resourcemanager.webapp.address.rm1" = "rm1:8088";

     

       46
       +
             "yarn.resourcemanager.webapp.address.rm2" = "rm2:8088";

     

       47
        
           };

     

       48
        
         in {

     

       49
       +
           zk1 = { ... }: {

     

       50
       +
             services.zookeeper.enable = true;

     

       51
       +
             networking.firewall.allowedTCPPorts = [ 2181 ];

     

       52
       +
           };

     

       53
       +
       

     

       54
       +
           # HDFS cluster

     

       55
       +
           nn1 = {pkgs, options, ...}: {

     

       56
        
             services.hadoop = {

     

       57
       +
               inherit package coreSite hdfsSite;

     

       58
        
               hdfs.namenode.enabled = true;

     

       59
       +
               hdfs.zkfc.enabled = true;

     

       60
        
             };

     

       61
       +
           };

     

       62
       +
           nn2 = {pkgs, options, ...}: {

     

       63
       +
             services.hadoop = {

     

       64
       +
               inherit package coreSite hdfsSite;

     

       65
       +
               hdfs.namenode.enabled = true;

     

       66
       +
               hdfs.zkfc.enabled = true;

     

       67
       +
             };

     

       68
        
           };

     

       69
        
       

     

       70
       +
           jn1 = {pkgs, options, ...}: {

     

       71
       +
             services.hadoop = {

     

       72
       +
               inherit package coreSite hdfsSite;

     

       73
       +
               hdfs.journalnode.enabled = true;

     

       74
       +
             };

     

       75
       +
           };

     

       76
       +
           jn2 = {pkgs, options, ...}: {

     

       77
        
             services.hadoop = {

     

       78
       +
               inherit package coreSite hdfsSite;

     

       79
       +
               hdfs.journalnode.enabled = true;

     

       80
       +
             };

     

       81
       +
           };

     

       82
       +
           jn3 = {pkgs, options, ...}: {

     

       83
       +
             services.hadoop = {

     

       84
       +
               inherit package coreSite hdfsSite;

     

       85
       +
               hdfs.journalnode.enabled = true;

     

       86
       +
             };

     

       87
       +
           };

     

       88
       +
       

     

       89
       +
           dn1 = {pkgs, options, ...}: {

     

       90
       +
             services.hadoop = {

     

       91
       +
               inherit package coreSite hdfsSite;

     

       92
        
               hdfs.datanode.enabled = true;

     

       93
       +
             };

     

       94
       +
           };

     

       95
       +
       

     

       96
       +
           # YARN cluster

     

       97
       +
           rm1 = {pkgs, options, ...}: {

     

       98
       +
             virtualisation.memorySize = 1024;

     

       99
       +
             services.hadoop = {

     

       100
       +
               inherit package coreSite hdfsSite;

     

       101
       +
               yarnSite = options.services.hadoop.yarnSite.default // yarnSiteHA;

     

       102
       +
               yarn.resourcemanager.enabled = true;

     

       103
       +
             };

     

       104
       +
           };

     

       105
       +
           rm2 = {pkgs, options, ...}: {

     

       106
       +
             virtualisation.memorySize = 1024;

     

       107
       +
             services.hadoop = {

     

       108
       +
               inherit package coreSite hdfsSite;

     

       109
       +
               yarnSite = options.services.hadoop.yarnSite.default // yarnSiteHA;

     

       110
       +
               yarn.resourcemanager.enabled = true;

     

       111
       +
             };

     

       112
       +
           };

     

       113
       +
           nm1 = {pkgs, options, ...}: {

     

       114
       +
             virtualisation.memorySize = 2048;

     

       115
       +
             services.hadoop = {

     

       116
       +
               inherit package coreSite hdfsSite;

     

       117
       +
               yarnSite = options.services.hadoop.yarnSite.default // yarnSiteHA;

     

       118
        
               yarn.nodemanager.enabled = true;

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       119
        
             };

     

       0
        
       
     

       120
        
           };

     

       121
        
         };

     

       122
        
       

     

       123
        
         testScript = ''

     

       124
        
           start_all()

     

       125
        
       

     

       126
       +
           #### HDFS tests ####

     

       0
        
       
     

       127
        
       

     

       128
       +
           zk1.wait_for_unit("network.target")

     

       129
       +
           jn1.wait_for_unit("network.target")

     

       130
       +
           jn2.wait_for_unit("network.target")

     

       131
       +
           jn3.wait_for_unit("network.target")

     

       132
       +
           nn1.wait_for_unit("network.target")

     

       133
       +
           nn2.wait_for_unit("network.target")

     

       134
       +
           dn1.wait_for_unit("network.target")

     

       135
        
       

     

       136
       +
           zk1.wait_for_unit("zookeeper")

     

       137
       +
           jn1.wait_for_unit("hdfs-journalnode")

     

       138
       +
           jn2.wait_for_unit("hdfs-journalnode")

     

       139
       +
           jn3.wait_for_unit("hdfs-journalnode")

     

       0
        
       
     

       140
        
       

     

       141
       +
           zk1.wait_for_open_port(2181)

     

       142
       +
           jn1.wait_for_open_port(8480)

     

       143
       +
           jn1.wait_for_open_port(8485)

     

       144
       +
           jn2.wait_for_open_port(8480)

     

       145
       +
           jn2.wait_for_open_port(8485)

     

       146
        
       

     

       147
       +
           # Namenodes must be stopped before initializing the cluster

     

       148
       +
           nn1.succeed("systemctl stop hdfs-namenode")

     

       149
       +
           nn2.succeed("systemctl stop hdfs-namenode")

     

       150
       +
           nn1.succeed("systemctl stop hdfs-zkfc")

     

       151
       +
           nn2.succeed("systemctl stop hdfs-zkfc")

     

       152
        
       

     

       153
       +
           # Initialize zookeeper for failover controller

     

       154
       +
           nn1.succeed("sudo -u hdfs hdfs zkfc -formatZK 2>&1 | systemd-cat")

     

       155
        
       

     

       156
       +
           # Format NN1 and start it

     

       157
       +
           nn1.succeed("sudo -u hdfs hadoop namenode -format 2>&1 | systemd-cat")

     

       158
       +
           nn1.succeed("systemctl start hdfs-namenode")

     

       159
       +
           nn1.wait_for_open_port(9870)

     

       160
       +
           nn1.wait_for_open_port(8022)

     

       161
       +
           nn1.wait_for_open_port(8020)

     

       162
        
       

     

       163
       +
           # Bootstrap NN2 from NN1 and start it

     

       164
       +
           nn2.succeed("sudo -u hdfs hdfs namenode -bootstrapStandby 2>&1 | systemd-cat")

     

       165
       +
           nn2.succeed("systemctl start hdfs-namenode")

     

       166
       +
           nn2.wait_for_open_port(9870)

     

       167
       +
           nn2.wait_for_open_port(8022)

     

       168
       +
           nn2.wait_for_open_port(8020)

     

       169
       +
           nn1.succeed("netstat -tulpne | systemd-cat")

     

       170
        
       

     

       171
       +
           # Start failover controllers

     

       172
       +
           nn1.succeed("systemctl start hdfs-zkfc")

     

       173
       +
           nn2.succeed("systemctl start hdfs-zkfc")

     

       174
        
       

     

       175
       +
           # DN should have started by now, but confirm anyway

     

       176
       +
           dn1.wait_for_unit("hdfs-datanode")

     

       177
       +
           # Print states of namenodes

     

       178
       +
           dn1.succeed("sudo -u hdfs hdfs haadmin -getAllServiceState | systemd-cat")

     

       179
       +
           # Wait for cluster to exit safemode

     

       180
       +
           dn1.succeed("sudo -u hdfs hdfs dfsadmin -safemode wait")

     

       181
       +
           dn1.succeed("sudo -u hdfs hdfs haadmin -getAllServiceState | systemd-cat")

     

       182
       +
           # test R/W

     

       183
       +
           dn1.succeed("echo testfilecontents | sudo -u hdfs hdfs dfs -put - /testfile")

     

       184
       +
           assert "testfilecontents" in dn1.succeed("sudo -u hdfs hdfs dfs -cat /testfile")

     

       185
       +
       

     

       186
       +
           # Test NN failover

     

       187
       +
           nn1.succeed("systemctl stop hdfs-namenode")

     

       188
       +
           assert "active" in dn1.succeed("sudo -u hdfs hdfs haadmin -getAllServiceState")

     

       189
       +
           dn1.succeed("sudo -u hdfs hdfs haadmin -getAllServiceState | systemd-cat")

     

       190
       +
           assert "testfilecontents" in dn1.succeed("sudo -u hdfs hdfs dfs -cat /testfile")

     

       191
       +
       

     

       192
       +
           nn1.succeed("systemctl start hdfs-namenode")

     

       193
       +
           nn1.wait_for_open_port(9870)

     

       194
       +
           nn1.wait_for_open_port(8022)

     

       195
       +
           nn1.wait_for_open_port(8020)

     

       196
       +
           assert "standby" in dn1.succeed("sudo -u hdfs hdfs haadmin -getAllServiceState")

     

       197
       +
           dn1.succeed("sudo -u hdfs hdfs haadmin -getAllServiceState | systemd-cat")

     

       198
       +
       

     

       199
       +
           #### YARN tests ####

     

       200
       +
       

     

       201
       +
           rm1.wait_for_unit("network.target")

     

       202
       +
           rm2.wait_for_unit("network.target")

     

       203
       +
           nm1.wait_for_unit("network.target")

     

       204
       +
       

     

       205
       +
           rm1.wait_for_unit("yarn-resourcemanager")

     

       206
       +
           rm1.wait_for_open_port(8088)

     

       207
       +
           rm2.wait_for_unit("yarn-resourcemanager")

     

       208
       +
           rm2.wait_for_open_port(8088)

     

       209
       +
       

     

       210
       +
           nm1.wait_for_unit("yarn-nodemanager")

     

       211
       +
           nm1.wait_for_open_port(8042)

     

       212
       +
           nm1.wait_for_open_port(8040)

     

       213
       +
           nm1.wait_until_succeeds("yarn node -list | grep Nodes:1")

     

       214
       +
           nm1.succeed("sudo -u yarn yarn rmadmin -getAllServiceState | systemd-cat")

     

       215
       +
           nm1.succeed("sudo -u yarn yarn node -list | systemd-cat")

     

       216
       +
       

     

       217
       +
           # Test RM failover

     

       218
       +
           rm1.succeed("systemctl stop yarn-resourcemanager")

     

       219
       +
           assert "standby" not in nm1.succeed("sudo -u yarn yarn rmadmin -getAllServiceState")

     

       220
       +
           nm1.succeed("sudo -u yarn yarn rmadmin -getAllServiceState | systemd-cat")

     

       221
       +
           rm1.succeed("systemctl start yarn-resourcemanager")

     

       222
       +
           rm1.wait_for_unit("yarn-resourcemanager")

     

       223
       +
           rm1.wait_for_open_port(8088)

     

       224
       +
           assert "standby" in nm1.succeed("sudo -u yarn yarn rmadmin -getAllServiceState")

     

       225
       +
           nm1.succeed("sudo -u yarn yarn rmadmin -getAllServiceState | systemd-cat")

     

       226
       +
       

     

       227
       +
           assert "Estimated value of Pi is" in nm1.succeed("HADOOP_USER_NAME=hdfs yarn jar $(readlink $(which yarn) | sed -r 's~bin/yarn~lib/hadoop-*/share/hadoop/mapreduce/hadoop-mapreduce-examples-*.jar~g') pi 2 10")

     

       228
       +
           assert "SUCCEEDED" in nm1.succeed("yarn application -list -appStates FINISHED")

     

       229
        
         '';

     

       230
       +
       })

+9 -10

nixos/tests/hadoop/hdfs.nix

···

       0
        
       
     

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

     

       2
        
         nodes = {

     

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

     

       4
        
             services.hadoop = {

     

       5
        
               package = pkgs.hadoop;

     

       6
       -
               hdfs.namenode.enabled = true;

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       7
        
               coreSite = {

     

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

     

       9
        
               };

     
···

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

     

       14
        
               };

     

       15
        
             };

     

       16
       -
             networking.firewall.allowedTCPPorts = [

     

       17
       -
               9870 # namenode.http-address

     

       18
       -
               8020 # namenode.rpc-address

     

       19
       -
             ];

     

       20
        
           };

     

       21
        
           datanode = {pkgs, ...}: {

     

       22
        
             services.hadoop = {

     
···

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

     

       27
        
               };

     

       28
        
             };

     

       29
       -
             networking.firewall.allowedTCPPorts = [

     

       30
       -
               9864 # datanode.http.address

     

       31
       -
               9866 # datanode.address

     

       32
       -
               9867 # datanode.ipc.address

     

       33
       -
             ];

     

       34
        
           };

     

       35
        
         };

     

       36
        
       

     
···

       50
        
       

     

       51
        
           namenode.succeed("curl -f http://namenode:9870")

     

       52
        
           datanode.succeed("curl -f http://datanode:9864")

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       53
        
         '';

     

       54
        
       })

···

       1
       +
       # Test a minimal HDFS cluster with no HA

     

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

     

       3
        
         nodes = {

     

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

     

       5
        
             services.hadoop = {

     

       6
        
               package = pkgs.hadoop;

     

       7
       +
               hdfs.namenode = {

     

       8
       +
                 enabled = true;

     

       9
       +
                 formatOnInit = true;

     

       10
       +
               };

     

       11
        
               coreSite = {

     

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

     

       13
        
               };

     
···

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

     

       18
        
               };

     

       19
        
             };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       20
        
           };

     

       21
        
           datanode = {pkgs, ...}: {

     

       22
        
             services.hadoop = {

     
···

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

     

       27
        
               };

     

       28
        
             };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       29
        
           };

     

       30
        
         };

     

       31
        
       

     
···

       45
        
       

     

       46
        
           namenode.succeed("curl -f http://namenode:9870")

     

       47
        
           datanode.succeed("curl -f http://datanode:9864")

     

       48
       +
       

     

       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")

     

       52
        
         '';

     

       53
        
       })

+1 -10

nixos/tests/hadoop/yarn.nix

···

       0
        
       
     

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

     

       2
        
         nodes = {

     

       3
        
           resourcemanager = {pkgs, ...}: {

     
···

       6
        
             services.hadoop.yarnSite = {

     

       7
        
               "yarn.resourcemanager.scheduler.class" = "org.apache.hadoop.yarn.server.resourcemanager.scheduler.fifo.FifoScheduler";

     

       8
        
             };

     

       9
       -
             networking.firewall.allowedTCPPorts = [

     

       10
       -
               8088 # resourcemanager.webapp.address

     

       11
       -
               8031 # resourcemanager.resource-tracker.address

     

       12
       -
             ];

     

       13
        
           };

     

       14
        
           nodemanager = {pkgs, ...}: {

     

       15
        
             services.hadoop.package = pkgs.hadoop;

     
···

       17
        
             services.hadoop.yarnSite = {

     

       18
        
               "yarn.resourcemanager.hostname" = "resourcemanager";

     

       19
        
               "yarn.nodemanager.log-dirs" = "/tmp/userlogs";

     

       20
       -
               "yarn.nodemanager.address" = "0.0.0.0:8041";

     

       21
        
             };

     

       22
       -
             networking.firewall.allowedTCPPorts = [

     

       23
       -
               8042 # nodemanager.webapp.address

     

       24
       -
               8041 # nodemanager.address

     

       25
       -
             ];

     

       26
        
           };

     

       27
        
       

     

       28
        
         };

     
···

       38
        
           nodemanager.wait_for_unit("yarn-nodemanager")

     

       39
        
           nodemanager.wait_for_unit("network.target")

     

       40
        
           nodemanager.wait_for_open_port(8042)

     

       41
       -
           nodemanager.wait_for_open_port(8041)

     

       42
        
       

     

       43
        
           resourcemanager.succeed("curl -f http://localhost:8088")

     

       44
        
           nodemanager.succeed("curl -f http://localhost:8042")

···

       1
       +
       # This only tests if YARN is able to start its services

     

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

     

       3
        
         nodes = {

     

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

     
···

       7
        
             services.hadoop.yarnSite = {

     

       8
        
               "yarn.resourcemanager.scheduler.class" = "org.apache.hadoop.yarn.server.resourcemanager.scheduler.fifo.FifoScheduler";

     

       9
        
             };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       10
        
           };

     

       11
        
           nodemanager = {pkgs, ...}: {

     

       12
        
             services.hadoop.package = pkgs.hadoop;

     
···

       14
        
             services.hadoop.yarnSite = {

     

       15
        
               "yarn.resourcemanager.hostname" = "resourcemanager";

     

       16
        
               "yarn.nodemanager.log-dirs" = "/tmp/userlogs";

     

       0
        
       
     

       17
        
             };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       18
        
           };

     

       19
        
       

     

       20
        
         };

     
···

       30
        
           nodemanager.wait_for_unit("yarn-nodemanager")

     

       31
        
           nodemanager.wait_for_unit("network.target")

     

       32
        
           nodemanager.wait_for_open_port(8042)

     

       0
        
       
     

       33
        
       

     

       34
        
           resourcemanager.succeed("curl -f http://localhost:8088")

     

       35
        
           nodemanager.succeed("curl -f http://localhost:8042")