database.go

package main

import (
	"context"
	"database/sql"
	"fmt"
	"log"
	"os"

	_ "github.com/jackc/pgx/v4/stdlib"
	"go.opentelemetry.io/otel"
)

var db *sql.DB

type Test struct {
	ID     string `json:"id"`
	Name   string `json:"name"`
	Status string `json:"status"`
}

func initDB() {
	connStr := os.Getenv("POSTGRES_CONN_STRING")
	if connStr == "" {
		log.Fatal("POSTGRES_CONN_STRING environment variable not set")
	}

	var err error
	db, err = sql.Open("pgx", connStr)
	if err != nil {
		log.Fatalf("Unable to connect to database: %v\n", err)
	}

	// Create table if it doesn't exist
	createTable()
}

func createTable() {
	_, err := db.Exec(`
		CREATE TABLE IF NOT EXISTS tests (
			id UUID PRIMARY KEY,
			name TEXT NOT NULL,
			status TEXT NOT NULL
		);
	`)
	if err != nil {
		log.Fatalf("Unable to create table: %v\n", err)
	}
}

func createTest(ctx context.Context, name string) (string, error) {
	_, span := otel.Tracer("").Start(ctx, "createTestInDB")
	defer span.End()

	var id string
	err := db.QueryRowContext(ctx, "INSERT INTO tests (id, name, status) VALUES (gen_random_uuid(), $1, 'created') RETURNING id", name).Scan(&id)
	if err != nil {
		return "", fmt.Errorf("could not create test: %w", err)
	}
	return id, nil
}

func updateTestStatus(ctx context.Context, id, status string) error {
	_, span := otel.Tracer("").Start(ctx, "updateTestStatusInDB")
	defer span.End()

	res, err := db.ExecContext(ctx, "UPDATE tests SET status = $1 WHERE id = $2", status, id)
	if err != nil {
		return fmt.Errorf("could not update test status: %w", err)
	}

	rowsAffected, err := res.RowsAffected()
	if err != nil {
		return fmt.Errorf("could not get rows affected: %w", err)
	}

	if rowsAffected == 0 {
		return fmt.Errorf("no test found with id %s", id)
	}

	return nil
}