build files

sreya · sreya · commit 0132fb3d3626 · 2024-11-20T19:00:38.000Z
diff --git a/agent/agentexec/cli_other.go b/agent/agentexec/cli_other.go
@@ -0,0 +1,10 @@
+//go:build !linux
+// +build !linux
+
+package agentexec
+
+import "golang.org/x/xerrors"
+
+func CLI(args []string, environ []string) error {
+	return xerrors.Errorf("agent-exec is only supported on Linux")
+}
diff --git a/agent/agentexec/cli_unix.go b/agent/agentexec/cli_unix.go
@@ -1,3 +1,6 @@
+//go:build linux
+// +build linux
+
 package agentexec
 
 import (
@@ -14,11 +17,6 @@ import (
 	"golang.org/x/xerrors"
 )
 
-const (
-	EnvProcOOMScore  = "CODER_PROC_OOM_SCORE"
-	EnvProcNiceScore = "CODER_PROC_NICE_SCORE"
-)
-
 // CLI runs the agent-exec command. It should only be called by the cli package.
 func CLI(args []string, environ []string) error {
 	runtime.LockOSThread()
@@ -127,30 +125,3 @@ func oomScoreAdj(pid int) (int, error) {
 func writeOOMScoreAdj(pid int, score int) error {
 	return os.WriteFile(fmt.Sprintf("/proc/%d/oom_score_adj", pid), []byte(fmt.Sprintf("%d", score)), 0o600)
 }
-
-// envValInt searches for a key in a list of environment variables and parses it to an int.
-// If the key is not found or cannot be parsed, returns 0 and false.
-func envValInt(env []string, key string) (int, bool) {
-	val, ok := envVal(env, key)
-	if !ok {
-		return 0, false
-	}
-
-	i, err := strconv.Atoi(val)
-	if err != nil {
-		return 0, false
-	}
-	return i, true
-}
-
-// envVal searches for a key in a list of environment variables and returns its value.
-// If the key is not found, returns empty string and false.
-func envVal(env []string, key string) (string, bool) {
-	prefix := key + "="
-	for _, e := range env {
-		if strings.HasPrefix(e, prefix) {
-			return strings.TrimPrefix(e, prefix), true
-		}
-	}
-	return "", false
-}
diff --git a/agent/agentexec/exec.go b/agent/agentexec/exec.go
@@ -6,14 +6,18 @@ import (
 	"os/exec"
 	"path/filepath"
 	"runtime"
+	"strconv"
+	"strings"
 
 	"golang.org/x/xerrors"
 )
 
 const (
 	// EnvProcPrioMgmt is the environment variable that determines whether
 	// we attempt to manage process CPU and OOM Killer priority.
-	EnvProcPrioMgmt = "CODER_PROC_PRIO_MGMT"
+	EnvProcPrioMgmt  = "CODER_PROC_PRIO_MGMT"
+	EnvProcOOMScore  = "CODER_PROC_OOM_SCORE"
+	EnvProcNiceScore = "CODER_PROC_NICE_SCORE"
 )
 
 // CommandContext returns an exec.Cmd that calls "coder agent-exec" prior to exec'ing
@@ -40,3 +44,30 @@ func CommandContext(ctx context.Context, cmd string, args ...string) (*exec.Cmd,
 	args = append([]string{"agent-exec", cmd}, args...)
 	return exec.CommandContext(ctx, bin, args...), nil
 }
+
+// envValInt searches for a key in a list of environment variables and parses it to an int.
+// If the key is not found or cannot be parsed, returns 0 and false.
+func envValInt(env []string, key string) (int, bool) {
+	val, ok := envVal(env, key)
+	if !ok {
+		return 0, false
+	}
+
+	i, err := strconv.Atoi(val)
+	if err != nil {
+		return 0, false
+	}
+	return i, true
+}
+
+// envVal searches for a key in a list of environment variables and returns its value.
+// If the key is not found, returns empty string and false.
+func envVal(env []string, key string) (string, bool) {
+	prefix := key + "="
+	for _, e := range env {
+		if strings.HasPrefix(e, prefix) {
+			return strings.TrimPrefix(e, prefix), true
+		}
+	}
+	return "", false
+}
