use a temporary folder on the D: drive in Windows CI for postgres

hugodutka · hugodutka · commit 988c8935a210 · 2024-12-02T14:50:32.000Z
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -426,6 +426,12 @@ jobs:
 
           if [ "${{ runner.os }}" == "Linux" ]; then
             make test-postgres
+          elif [ "${{ runner.os }}" == "Windows" ]; then
+            # Create temp dir on D: drive for Windows. The default C: drive is extremely
+            # slow: https://github.com/actions/runner-images/issues/8755
+            mkdir -p "D:/temp/embedded-pg"
+            go run scripts/embedded-pg/main.go -path "D:/temp/embedded-pg"
+            DB=ci gotestsum --format standard-quiet -- -v -short -count=1 ./...
           else
             go run scripts/embedded-pg/main.go
             DB=ci gotestsum --format standard-quiet -- -v -short -count=1 ./...
diff --git a/scripts/embedded-pg/main.go b/scripts/embedded-pg/main.go
@@ -3,14 +3,23 @@ package main
 
 import (
 	"database/sql"
+	"flag"
 	"os"
 	"path/filepath"
 
 	embeddedpostgres "github.com/fergusstrange/embedded-postgres"
 )
 
 func main() {
+	var customPath string
+	flag.StringVar(&customPath, "path", "", "Optional custom path for postgres data directory")
+	flag.Parse()
+
 	postgresPath := filepath.Join(os.TempDir(), "coder-test-postgres")
+	if customPath != "" {
+		postgresPath = customPath
+	}
+
 	ep := embeddedpostgres.NewDatabase(
 		embeddedpostgres.DefaultConfig().
 			Version(embeddedpostgres.V16).
