ci: make runner input flag (#2071)

ariga · Sep 12, 2023 · ade1fef · ade1fef
1 parent 1d3fbc7
commit ade1fef
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 11 deletions.
diff --git a/internal/ci/ci_dialect.tmpl b/internal/ci/ci_dialect.tmpl
@@ -15,7 +15,7 @@ on:
 jobs:
 {{- range $.Jobs }}
   integration-{{ .Version }}:
-    runs-on: ubuntu-latest
+    runs-on: {{ $.Runner }}
     {{ if .Image }}services:
       {{ .Version }}:
         image: {{ .Image }}

diff --git a/internal/ci/ci_go.tmpl b/internal/ci/ci_go.tmpl
@@ -14,7 +14,7 @@ on:
 
 jobs:
   lint:
-    runs-on: ubuntu-latest
+    runs-on: {{ $.Runner }}
     steps:
       - uses: actions/setup-go@v4
         with:
@@ -26,7 +26,7 @@ jobs:
           args: --verbose
 
   generate-cmp:
-    runs-on: ubuntu-latest
+    runs-on: {{ $.Runner }}
     steps:
       - uses: actions/checkout@v2.4.0
       - uses: actions/setup-go@v4
@@ -48,7 +48,7 @@ jobs:
             exit 1
           fi
   unit:
-    runs-on: ubuntu-latest
+    runs-on: {{ $.Runner }}
     strategy:
       matrix:
         go: [ '1.20', '1.21' ]
@@ -65,7 +65,7 @@ jobs:
         working-directory: schemahcl
 
   cli:
-    runs-on: ubuntu-latest
+    runs-on: {{ $.Runner }}
     steps:
       - uses: actions/checkout@v2
       - uses: actions/setup-go@v4
@@ -76,7 +76,7 @@ jobs:
         working-directory: cmd/atlas
 
   integration:
-    runs-on: ubuntu-latest
+    runs-on: {{ $.Runner }}
     steps:
       - uses: actions/checkout@v2.3.4
       - uses: actions/setup-go@v4

diff --git a/internal/ci/ci_revisions.tmpl b/internal/ci/ci_revisions.tmpl
@@ -13,7 +13,7 @@ on:
       - 'cmd/atlas/internal/migrate/ent/**'
 jobs:
   revisions:
-    runs-on: ubuntu-latest
+    runs-on: {{ $.Runner }}
     steps:
       - uses: actions/checkout@v2.3.4
         with:

diff --git a/internal/ci/main.go b/internal/ci/main.go
@@ -165,17 +165,18 @@ var (
 )
 
 func main() {
-	var flavor, tags, suffix string
+	var flavor, tags, suffix, runner string
 	flag.StringVar(&flavor, "flavor", "", "")
 	flag.StringVar(&tags, "tags", "", "")
 	flag.StringVar(&suffix, "suffix", "", "")
+	flag.StringVar(&runner, "runner", "ubuntu-latest", "")
 	flag.Parse()
 	for _, n := range []string{"dialect", "go", "revisions"} {
 		var buf bytes.Buffer
 		if err := tpl.ExecuteTemplate(&buf, fmt.Sprintf("ci_%s.tmpl", n), struct {
-			Jobs         []Job
-			Flavor, Tags string
-		}{jobs, flavor, tags}); err != nil {
+			Jobs                 []Job
+			Flavor, Tags, Runner string
+		}{jobs, flavor, tags, runner}); err != nil {
 			log.Fatalln(err)
 		}
 		err := os.WriteFile(filepath.Clean(fmt.Sprintf("../../.github/workflows/ci-%s_%s.yaml", n, suffix)), buf.Bytes(), 0600)