mirror of
https://github.com/coder/coder.git
synced 2025-07-09 11:45:56 +00:00
025b55f7be
* chore: Initial database scaffolding This implements migrations and code generation for interfacing with a PostgreSQL database. A dependency is added for the "postgres" binary on the host, but that seems like an acceptable requirement considering it's our primary database. An in-memory database object can be created for simple cross-OS and fast testing. * Run tests in CI * Use Docker instead of binaries on the host * Skip database tests on non-Linux operating systems * chore: Add golangci-lint and codecov * Use consistent file names
90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/coder/coder/database"
|
|
"github.com/coder/coder/database/postgres"
|
|
)
|
|
|
|
func main() {
|
|
connection, closeFn, err := postgres.Open()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer closeFn()
|
|
db, err := sql.Open("postgres", connection)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = database.Migrate(context.Background(), "postgres", db)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
cmd := exec.Command(
|
|
"pg_dump",
|
|
"--schema-only",
|
|
connection,
|
|
"--no-privileges",
|
|
"--no-owner",
|
|
"--no-comments",
|
|
|
|
// We never want to manually generate
|
|
// queries executing against this table.
|
|
"--exclude-table=schema_migrations",
|
|
)
|
|
cmd.Env = []string{
|
|
"PGTZ=UTC",
|
|
"PGCLIENTENCODING=UTF8",
|
|
}
|
|
var output bytes.Buffer
|
|
cmd.Stdout = &output
|
|
cmd.Stderr = os.Stderr
|
|
err = cmd.Run()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for _, sed := range []string{
|
|
// Remove all comments.
|
|
"/^--/d",
|
|
// Public is implicit in the schema.
|
|
"s/ public\\./ /",
|
|
// Remove database settings.
|
|
"s/SET.*;//g",
|
|
// Remove select statements. These aren't useful
|
|
// to a reader of the dump.
|
|
"s/SELECT.*;//g",
|
|
// Removes multiple newlines.
|
|
"/^$/N;/^\\n$/D",
|
|
} {
|
|
cmd := exec.Command("sed", "-e", sed)
|
|
cmd.Stdin = bytes.NewReader(output.Bytes())
|
|
output = bytes.Buffer{}
|
|
cmd.Stdout = &output
|
|
cmd.Stderr = os.Stderr
|
|
err = cmd.Run()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
dump := fmt.Sprintf("-- Code generated by 'make database/generate'. DO NOT EDIT.\n%s", output.Bytes())
|
|
_, mainPath, _, ok := runtime.Caller(0)
|
|
if !ok {
|
|
panic("couldn't get caller path")
|
|
}
|
|
err = ioutil.WriteFile(filepath.Join(mainPath, "..", "..", "dump.sql"), []byte(dump), 0644)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|