compute_base_config.rb

# frozen_string_literal: true

#
# Cookbook Name:: aws-parallelcluster
# Recipe:: compute_base_config
#
# Copyright 2013-2015 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.

# Retrieve head node info
if node['cfncluster']['cfn_scheduler'] == 'slurm'
  ruby_block "retrieve head_node ip" do
    block do
      head_node_private_ip, head_node_private_dns = hit_head_node_info
      node.force_default['cfncluster']['cfn_master'] = head_node_private_dns
      node.force_default['cfncluster']['cfn_master_private_ip'] = head_node_private_ip
    end
    retries 5
    retry_delay 3
  end
end

# Parse and get RAID shared directory info and turn into an array
raid_shared_dir = node['cfncluster']['cfn_raid_parameters'].split(',')[0]

if raid_shared_dir != "NONE"
  # Path needs to be fully qualified, for example "shared/temp" becomes "/shared/temp"
  raid_shared_dir = "/" + raid_shared_dir unless raid_shared_dir.start_with?("/")

  # Created RAID shared mount point
  directory raid_shared_dir do
    mode '1777'
    owner 'root'
    group 'root'
    action :create
  end

  # Mount RAID directory over NFS
  mount raid_shared_dir do
    device(lazy { "#{node['cfncluster']['cfn_master_private_ip']}:#{raid_shared_dir}" })
    fstype 'nfs'
    options 'hard,intr,noatime,_netdev'
    action %i[mount enable]
    retries 3
    retry_delay 5
  end
end

# Mount /home over NFS
mount '/home' do
  device(lazy { "#{node['cfncluster']['cfn_master_private_ip']}:/home" })
  fstype 'nfs'
  options 'hard,intr,noatime,_netdev'
  action %i[mount enable]
  retries 3
  retry_delay 5
end

# Mount /opt/intel over NFS
mount '/opt/intel' do
  device(lazy { "#{node['cfncluster']['cfn_master_private_ip']}:/opt/intel" })
  fstype 'nfs'
  options 'hard,intr,noatime,_netdev'
  action %i[mount enable]
  retries 3
  retry_delay 5
  only_if { ::File.directory?("/opt/intel") }
end

# Configure Ganglia
include_recipe 'aws-parallelcluster::ganglia_config'

# Setup cluster user
user node['cfncluster']['cfn_cluster_user'] do
  manage_home false
  comment 'AWS ParallelCluster user'
  home "/home/#{node['cfncluster']['cfn_cluster_user']}"
  shell '/bin/bash'
end

# Parse shared directory info and turn into an array
shared_dir_array = node['cfncluster']['cfn_shared_dir'].split(',')

# Mount each volume with NFS
shared_dir_array.each do |dir|
  dirname = dir.strip

  unless dirname == "NONE"
    dirname = "/" + dirname unless dirname.start_with?("/")

    # Created shared mount point
    directory dirname do
      mode '1777'
      owner 'root'
      group 'root'
      recursive true
      action :create
    end

    # Mount shared volume over NFS
    mount dirname do
      device(lazy { "#{node['cfncluster']['cfn_master_private_ip']}:#{dirname}" })
      fstype 'nfs'
      options 'hard,intr,noatime,_netdev'
      action %i[mount enable]
      retries 3
      retry_delay 5
    end
  end
end

# Install nodewatcher.cfg
template '/etc/nodewatcher.cfg' do
  source 'nodewatcher.cfg.erb'
  owner 'root'
  group 'root'
  mode '0644'
end