Function main

coderd/database/gen/dump/main.go:51–126

()

}

func main() {
	t := &mockTB{}

	// Ensure cleanups run on both normal exit and SIGINT/SIGTERM.
	// Go's default signal handlers call os.Exit, which skips deferred
	// funcs and would leave an embedded-postgres daemon orphaned.
	var cleanupOnce sync.Once
	runCleanup := func() {
		cleanupOnce.Do(func() {
			for _, f := range t.cleanup {
				f()
			}
		})
	}
	defer runCleanup()

	sigCh := make(chan os.Signal, 1)
	signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
	go func() {
		<-sigCh
		runCleanup()
		os.Exit(130)
	}()

	connection := os.Getenv("DB_DUMP_CONNECTION_URL")
	if connection == "" {
		var cleanup func()
		var err error
		connection, cleanup, err = dbtestutil.OpenContainerized(t, dbtestutil.DBContainerOptions{})
		if err != nil {
			_, _ = fmt.Fprintf(os.Stderr, "containerized postgres unavailable (%s); falling back to embedded postgres\n", err)
			connection, cleanup, err = openEmbeddedPostgres()
			if err != nil {
				panic(err)
			}
		}
		t.Cleanup(cleanup)
	}

	db, err := sql.Open("postgres", connection)
	if err != nil {
		err = xerrors.Errorf("open database failed: %w", err)
		panic(err)
	}
	defer db.Close()

	err = migrations.Up(db)
	if err != nil {
		err = xerrors.Errorf("run migrations failed: %w", err)
		panic(err)
	}

	dumpBytes, err := dbtestutil.PGDumpSchemaOnly(connection)
	if err != nil {
		if !pgDumpUsable() {
			_, _ = fmt.Fprintf(os.Stderr,
				"\nThis step needs pg_dump (PostgreSQL v13 or later) on PATH OR a Docker-compatible daemon.\n"+
					"Install pg_dump locally to avoid Docker:\n"+