rapl capping via ssh command in rapl package

pcp/ssh/raplcap.go

@@ -1,27 +0,0 @@
-package ssh
-
-import (
-	"golang.org/x/crypto/ssh"
-	"fmt"
-)
-
-func main() {
-	sshConfig := &ssh.ClientConfig{
-		User: "rapl",
-		Auth: []ssh.AuthMethod{
-			ssh.Password("pankajlikesdanceswithwolves#!@#"),
-		},
-	}
-
-	connection, err := ssh.Dial("tcp", "host:port", sshConfig)
-	if err != nil {
-		return nil, fmt.Errorf("Failed to dial: %s", err)
-	}
-
-	session, err := connection.NewSession()
-	if err != nil {
-		return nil, fmt.Errorf("Failed to create session: %s", err)
-	}
-
-	err = session.Run("sudo /misc/shared_data/rdelval1/RAPL_PKG_Throttle.py 100")
-}

pcp/test/power.go

@@ -109,7 +109,7 @@ func main() {
 				fmt.Println("Index: ", i)
 				powerIndexes = append(powerIndexes, i)
 				indexToHost[i] = split[0]
-				powerAverage[split[0]] = ring.New(10) // Two PKS per node, 10 = 5 seconds tracking
+				powerAverage[split[0]] = ring.New(10) // Two PKGS per node, 10 = 5 seconds tracking
 			}
 		}
 
@@ -152,15 +152,7 @@ func main() {
 				log.Printf("Current Victim %s Avg. Wattage: %f", victims[0].Host, victims[0].Watts * RAPLUnits)
 			}
 
-			/*
-				fmt.Printf("Second: %d\n", seconds)
-				for i, val := range strings.Split(scanner.Text(), ",") {
-					fmt.Printf("host metric: %s val: %s\n", headers[i], val)
-				}*/
-
 			seconds++
-
-			// fmt.Println("--------------------------------")
 		}
 	}(logging)
 

rapl/cap.go

@@ -0,0 +1,39 @@
+package rapl
+
+import (
+	"golang.org/x/crypto/ssh"
+	"github.com/pkg/errors"
+	"strconv"
+)
+
+func Cap(host, username string,  percentage int) (error) {
+
+	if percentage > 100 || percentage < 0 {
+		return errors.New("Percentage is out of range")
+	}
+
+	sshConfig := &ssh.ClientConfig{
+		User: username,
+		Auth: []ssh.AuthMethod{
+			// TODO: CHANGE and MAKE THIS USE SSH KEY BEFORE MAKING PUBLIC!!!!
+			ssh.Password("pankajlikesdanceswithwolves#!@#"),
+		},
+	}
+
+	connection, err := ssh.Dial("tcp", host+":22", sshConfig)
+	if err != nil {
+		return errors.Wrap(err, "Failed to dial")
+	}
+
+	session, err := connection.NewSession()
+	if err != nil {
+		return errors.Wrap(err, "Failed to create session")
+	}
+
+	err = session.Run("sudo /misc/shared_data/rdelval1/RAPL_PKG_Throttle.py " + strconv.Itoa(percentage))
+	if err != nil {
+		return errors.Wrap(err, "Failed to run RAPL script")
+	}
+
+	return nil
+}