Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
3.1 kB · 122 lines
Go
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123package orm
import ( "context" "database/sql" "fmt" "log/slog" "reflect" "strings")
type migrationFn = func(*sql.Tx) error
func RunMigration(c *sql.Conn, logger *slog.Logger, name string, migrationFn migrationFn) error { logger = logger.With("migration", name)
tx, err := c.BeginTx(context.Background(), nil) if err != nil { return err } defer tx.Rollback()
var exists bool err = tx.QueryRow("select exists (select 1 from migrations where name = ?)", name).Scan(&exists) if err != nil { return err }
if !exists { // run migration err = migrationFn(tx) if err != nil { logger.Error("failed to run migration", "err", err) return err }
// mark migration as complete _, err = tx.Exec("insert into migrations (name) values (?)", name) if err != nil { logger.Error("failed to mark migration as complete", "err", err) return err }
// commit the transaction if err := tx.Commit(); err != nil { return err }
logger.Info("migration applied successfully") } else { logger.Warn("skipped migration, already applied") }
return nil}
type Filter struct { Key string arg any Cmp string}
func newFilter(key, cmp string, arg any) Filter { return Filter{ Key: key, arg: arg, Cmp: cmp, }}
func FilterEq(key string, arg any) Filter { return newFilter(key, "=", arg) }func FilterNotEq(key string, arg any) Filter { return newFilter(key, "<>", arg) }func FilterGte(key string, arg any) Filter { return newFilter(key, ">=", arg) }func FilterLte(key string, arg any) Filter { return newFilter(key, "<=", arg) }func FilterIs(key string, arg any) Filter { return newFilter(key, "is", arg) }func FilterIsNot(key string, arg any) Filter { return newFilter(key, "is not", arg) }func FilterIn(key string, arg any) Filter { return newFilter(key, "in", arg) }func FilterLike(key string, arg any) Filter { return newFilter(key, "like", arg) }func FilterNotLike(key string, arg any) Filter { return newFilter(key, "not like", arg) }func FilterContains(key string, arg any) Filter { return newFilter(key, "like", fmt.Sprintf("%%%v%%", arg))}
func (f Filter) Condition() string { rv := reflect.ValueOf(f.arg) kind := rv.Kind()
// if we have `FilterIn(k, [1, 2, 3])`, compile it down to `k in (?, ?, ?)` if (kind == reflect.Slice && rv.Type().Elem().Kind() != reflect.Uint8) || kind == reflect.Array { if rv.Len() == 0 { // always false return "1 = 0" }
placeholders := make([]string, rv.Len()) for i := range placeholders { placeholders[i] = "?" }
return fmt.Sprintf("%s %s (%s)", f.Key, f.Cmp, strings.Join(placeholders, ", ")) }
return fmt.Sprintf("%s %s ?", f.Key, f.Cmp)}
func (f Filter) Arg() []any { rv := reflect.ValueOf(f.arg) kind := rv.Kind() if (kind == reflect.Slice && rv.Type().Elem().Kind() != reflect.Uint8) || kind == reflect.Array { if rv.Len() == 0 { return nil }
out := make([]any, rv.Len()) for i := range rv.Len() { out[i] = rv.Index(i).Interface() } return out }
return []any{f.arg}}