fetch full coder binary

johnstcn · johnstcn · commit ead423e87abc · 2023-10-03T11:50:36.000+01:00
diff --git a/cli/exp_scaletest_slim.go b/cli/exp_scaletest_slim.go
@@ -9,7 +9,7 @@ func (r *RootCmd) scaletestCmd() *clibase.Cmd {
 		Use:   "scaletest",
 		Short: "Run a scale test against the Coder API",
 		Handler: func(inv *clibase.Invocation) error {
-			SlimUnsupported(inv.Stderr, "scaletest")
+			SlimUnsupported(inv.Stderr, "exp scaletest")
 			return nil
 		},
 	}
diff --git a/scaletest/lib/coder_init.sh b/scaletest/lib/coder_init.sh
@@ -20,7 +20,6 @@ ARCH="$(arch)"
 if [[ "$ARCH" == "x86_64" ]]; then
 	ARCH="amd64"
 fi
-PLATFORM="$(uname | tr '[:upper:]' '[:lower:]')"
 
 if [[ -f "${CONFIG_DIR}/coder.env" ]]; then
 	echo "Found existing coder.env in ${CONFIG_DIR}!"
@@ -29,8 +28,20 @@ if [[ -f "${CONFIG_DIR}/coder.env" ]]; then
 fi
 
 maybedryrun "$DRY_RUN" mkdir -p "${CONFIG_DIR}"
-echo "Fetching Coder CLI for first-time setup!"
-maybedryrun "$DRY_RUN" curl -fsSLk "${CODER_URL}/bin/coder-${PLATFORM}-${ARCH}" -o "${CONFIG_DIR}/coder"
+echo "Fetching Coder for first-time setup!"
+pod=$(kubectl get pods \
+	--namespace="${NAMESPACE}" \
+	--selector="app.kubernetes.io/name=coder,app.kubernetes.io/part-of=coder" \
+	--output="jsonpath='{.items[0].metadata.name}'")
+if [[ -z ${pod} ]]; then
+	log "Could not find coder pod!"
+	exit 1
+fi
+maybedryrun "$DRY_RUN" kubectl \
+	--namespace="${NAMESPACE}" \
+	cp \
+	--container=coder \
+	"${pod}:/opt/coder" "${CONFIG_DIR}/coder"
 maybedryrun "$DRY_RUN" chmod +x "${CONFIG_DIR}/coder"
 
 set +o pipefail
diff --git a/scaletest/templates/scaletest-runner/scripts/lib.sh b/scaletest/templates/scaletest-runner/scripts/lib.sh
@@ -24,7 +24,12 @@ SCALETEST_PPROF_DIR="${SCALETEST_RUN_DIR}/pprof"
 mkdir -p "${SCALETEST_STATE_DIR}" "${SCALETEST_RESULTS_DIR}" "${SCALETEST_PPROF_DIR}"
 
 coder() {
-	maybedryrun "${DRY_RUN}" command coder "${@}"
+	if [[ -z "${SCALETEST_CODER_BINARY}" ]]; then
+		echo "Fetching full coder binary..."
+		fetch_coder_full
+		echo "SCALETEST_CODER_BINARY=${SCALETEST_CODER_BINARY}"
+	fi
+	maybedryrun "${DRY_RUN}" "${SCALETEST_CODER_BINARY}" "${@}"
 }
 
 show_json() {
@@ -240,3 +245,36 @@ set_appearance() {
 		--data "${newjson}" \
 		"${CODER_URL}/api/v2/appearance"
 }
+
+# fetch_coder_full fetches the full (non-slim) coder binary from one of the coder pods
+# running in the same namespace as the current pod.
+fetch_coder_full() {
+	local mkdir_dry_run_arg=""
+	if [[ "${DRY_RUN}" == "1" ]]; then
+		mkdir_dry_run_arg="--dry-run"
+	fi
+	target_dir=$(mktemp ${mkdir_dry_run_arg} --directory -t scaletest-coder-full-XXXXXX)
+	target_path="${target_dir}/coder"
+	local pod
+	local namespace
+	namespace=<(/var/run/secrets/kubernetes.io/serviceaccount/namespace)
+	if [[ -z ${namespace} ]]; then
+		log "Could not determine namespace!"
+		exit 1
+	fi
+	pod=$(kubectl get pods \
+		--namespace "${namespace}" \
+		--selector "app.kubernetes.io/name=coder,app.kubernetes.io/part-of=coder" \
+		--output jsonpath='{.items[0].metadata.name}')
+	if [[ -z ${pod} ]]; then
+		log "Could not find coder pod!"
+		exit 1
+	fi
+	maybedryrun "${DRY_RUN}" kubectl \
+		--namespace "${namespace}" \
+		cp \
+		--container coder \
+		"${pod}:/opt/coder" "${target_path}"
+	maybedryrun chmod +x "${target_path}"
+	export SCALETEST_CODER_BINARY="${target_path}"
+}

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ func (r RootCmd) scaletestCmd() clibase.Cmd {`
`9`	`9`	`Use: "scaletest",`
`10`	`10`	`Short: "Run a scale test against the Coder API",`
`11`	`11`	`Handler: func(inv *clibase.Invocation) error {`
`12`		`- SlimUnsupported(inv.Stderr, "scaletest")`
	`12`	`+ SlimUnsupported(inv.Stderr, "exp scaletest")`
`13`	`13`	`return nil`
`14`	`14`	`},`
`15`	`15`	`}`