Add Cluster Status Management Demon

Signed-off-by: Luca Carrogu <[email protected]>

Author: lukeseawalker

attributes/default.rb

@@ -32,6 +32,7 @@
 default['cluster']['previous_cluster_config_path'] = "#{node['cluster']['shared_dir']}/previous-cluster-config.yaml"
 default['cluster']['launch_templates_config_path'] = "#{node['cluster']['shared_dir']}/launch-templates-config.json"
 default['cluster']['instance_types_data_path'] = "#{node['cluster']['shared_dir']}/instance-types-data.json"
+default['cluster']['computefleet_status_path'] = "#{node['cluster']['shared_dir']}/computefleet-status.json"
 default['cluster']['reserved_base_uid'] = 400
 
 # Python Version

cookbooks/aws-parallelcluster-config/recipes/update_computefleet_status.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+#
+# Cookbook:: aws-parallelcluster-config
+# Recipe:: update_computefleet_status
+#
+# Copyright:: 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with the
+# License. A copy of the License is located at
+#
+# http://aws.amazon.com/apache2.0/
+#
+# or in the "LICENSE.txt" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions and
+# limitations under the License.
+
+unless node['cluster']['scheduler'] == 'awsbatch'
+  load_cluster_config
+end
+
+include_recipe 'aws-parallelcluster-scheduler-plugin::update_computefleet_status' if node['cluster']['scheduler'] == 'plugin'

cookbooks/aws-parallelcluster-config/templates/default/base/parallelcluster_supervisord.conf.erb

@@ -17,7 +17,14 @@ environment = HOME="/home/<%= node['cluster']['cluster_admin_user'] %>",USER="<%
 redirect_stderr = true
 stdout_logfile = /var/log/parallelcluster/clustermgtd
 <% end -%>
-
+<% if node['cluster']['scheduler'] == 'plugin' -%>
+[program:clusterstatusmgtd]
+command = <%= node['cluster']['cookbook_virtualenv_path'] %>/bin/python /opt/parallelcluster/scripts/scheduler_plugin/clusterstatusmgtd.py
+user = <%= node['cluster']['cluster_admin_user'] %>
+environment = HOME="/home/<%= node['cluster']['cluster_admin_user'] %>",USER="<%= node['cluster']['cluster_admin_user'] %>"
+redirect_stderr = true
+stdout_logfile = /var/log/parallelcluster/clusterstatusmgtd
+<% end -%>
 <% if node['cluster']['dcv_enabled'] == "head_node" && node['conditions']['dcv_supported'] -%>
 [program:pcluster_dcv_authenticator]
 command = <%= node['cluster']['dcv']['authenticator']['virtualenv_path'] %>/bin/python <%= node['cluster']['dcv']['authenticator']['user_home'] %>/pcluster_dcv_authenticator.py