pbs_ssher.go

package main

import (
	"os/exec"
	"encoding/json"
	"os"
	"bufio"
	"fmt"
	"strings"
)

func getHostMap() (map[string]string, error) {
	hostMap := make(map[string]string)

	f, err := os.Open(os.Getenv("PBS_NODEFILE"))
	if err != nil {
		return nil, err
	}
	r := bufio.NewReader(f)
	s, e := r.ReadString('\n')

	for e == nil {
		host := strings.Fields(s)[0]
		out, err := exec.Command("getent", "hosts", host).Output()
		if err != nil {
			return nil, err
		}

		hostMap[host] = strings.Fields(string(out))[0]
    		s, e = r.ReadString('\n')
	}


	return hostMap, nil

}

func main() {
	hostMap, _ := getHostMap()

	hosts := make([]string, len(hostMap))
	idx := 0
    	for  _, value := range hostMap {
       		hosts[idx] = value
       		idx++
   	}

	json, _ := json.Marshal(dict)
	hostMap := make(map[string]string)
	for host, ip := range dict {
    	    	fmt.Println(host, ip)
		//out, _ := exec.Command("ssh", fmt.Sprintf("prl900@%s", ip), "nohup", "/g/data1/sp9/prl900/stampede/server/benchmark_server", "&").Output()
		//fmt.Println(string(out))
   	 }

	fmt.Println(dict)
}