Add runs-on-high api

Signed-off-by: Tamal Saha <[email protected]>
appscodelabs · Oct 5, 2023 · fdfe1b4 · fdfe1b4
1 parent 2c5fbc5
commit fdfe1b4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/pkg/backend/processor.go b/pkg/backend/processor.go
@@ -112,7 +112,7 @@ func SubmitPayload(gh *github.Client, nc *nats.Conn, r *http.Request, secretToke
 	return nil
 }
 
-func DefaultJobLabel(gh *github.Client, org string, private bool) string {
+func UseRegularRunner(gh *github.Client, org string, private bool) string {
 	initCache(gh)
 
 	if private && mustUsedUpFreeMinutes(actionsBillingCache.Get(org)) {
@@ -121,6 +121,15 @@ func DefaultJobLabel(gh *github.Client, org string, private bool) string {
 	return "ubuntu-20.04"
 }
 
+func UseHighPriorityRunner(gh *github.Client, org string, private bool) string {
+	initCache(gh)
+
+	if private && mustUsedUpFreeMinutes(actionsBillingCache.Get(org)) {
+		return RunnerHigh
+	}
+	return "ubuntu-20.04"
+}
+
 func usedUpFreeMinutes(gh *github.Client, org string) (bool, error) {
 	ab, _, err := gh.Billing.GetActionsBillingOrg(context.Background(), org)
 	if err != nil {

diff --git a/pkg/cmds/webhook.go b/pkg/cmds/webhook.go
@@ -132,7 +132,13 @@ func runServer(gh *github.Client, nc *nats.Conn, sp *backend.StatusReporter) err
 	r.Get("/runs-on/{org}", func(w http.ResponseWriter, r *http.Request) {
 		org := chi.URLParam(r, "org")
 		private := r.URL.Query().Get("visibility") == "private"
-		label := backend.DefaultJobLabel(gh, org, private)
+		label := backend.UseRegularRunner(gh, org, private)
+		_, _ = w.Write([]byte(label))
+	})
+	r.Get("/runs-on-high/{org}", func(w http.ResponseWriter, r *http.Request) {
+		org := chi.URLParam(r, "org")
+		private := r.URL.Query().Get("visibility") == "private"
+		label := backend.UseHighPriorityRunner(gh, org, private)
 		_, _ = w.Write([]byte(label))
 	})