popertotsandSisyphus
bc6a5461c1
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
//go:build dev
|
|
// +build dev
|
|
|
|
package dav
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
var Service *BaseService
|
|
|
|
func init() {
|
|
if err := connect(); err != nil {
|
|
// Don't fatal in test mode - tests will use testdb instead
|
|
if os.Getenv("GO_TESTING") != "" {
|
|
return
|
|
}
|
|
log.Fatalf("failed to initialize dev service: %v", err)
|
|
}
|
|
}
|
|
|
|
func connect() error {
|
|
dsn := fmt.Sprintf(
|
|
"postgres://%s:%s@localhost:5432/%s?sslmode=disable&require_auth=scram-sha-256",
|
|
getEnv("POSTGRES_USER"),
|
|
getEnv("POSTGRES_PASSWORD"),
|
|
getEnv("POSTGRES_DB"),
|
|
)
|
|
pool, err := pgxpool.New(context.Background(), dsn)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
Service = newBaseService(pool)
|
|
return testDB(pool)
|
|
}
|
|
|
|
func testDB(pool *pgxpool.Pool) error {
|
|
var n int
|
|
err := pool.QueryRow(context.Background(), "SELECT 1").Scan(&n)
|
|
if err != nil || n != 1 {
|
|
return fmt.Errorf("db test failed: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func getEnv(key string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
// Return empty string instead of fatal error - allows tests to run without prod env vars
|
|
return ""
|
|
}
|