finish world initialization, set up init on startup and on shutdown, move all models to main directory

internal/foundry/models/db/utils.go

@@ -0,0 +1,239 @@
+package db
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/jmoiron/sqlx"
+	"github.com/mattn/go-sqlite3"
+	"golang.org/x/sync/errgroup"
+)
+
+var (
+	ErrNoQuery          = errors.New("Query has not been set")
+	ErrorRecordNotFound = errors.New("Record not found")
+)
+
+type AllowedIds interface {
+	~uint | ~string
+}
+
+type InsertId[T AllowedIds] struct {
+	id        T
+	fieldName string
+	tableName string
+	query     string
+}
+
+type Insertable[T AllowedIds] interface {
+	Query(data *InsertId[T])
+	Insert(tx *sqlx.Tx, relId *InsertId[T]) error
+	InsertCtx(ctx context.Context, tx *sqlx.Tx, relId *InsertId[T]) error
+}
+
+func InsertWithCtx[T AllowedIds, I Insertable[T]](tx *sqlx.Tx, data I, relId InsertId[T]) error {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	data.Query(&relId)
+	return data.InsertCtx(ctx, tx, &relId)
+}
+
+func InsertWithCtxParallel[T AllowedIds, I Insertable[T]](g *errgroup.Group, ctx context.Context, tx *sqlx.Tx, data I, relId InsertId[T]) {
+	g.Go(func() error {
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+
+		data.Query(&relId)
+		err := data.InsertCtx(ctx, tx, &relId)
+		return err
+	})
+}
+
+func InsertSlice[T AllowedIds, I Insertable[T]](tx *sqlx.Tx, data []I, relId InsertId[T]) error {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	if len(data) > 0 {
+		data[0].Query(&relId)
+	}
+	for i := range data {
+		err := data[i].InsertCtx(ctx, tx, &relId)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func InsertSliceParallelTimeout[T AllowedIds, I Insertable[T]](g *errgroup.Group, tx *sqlx.Tx, data []I, relId InsertId[T], timeout time.Duration) {
+	g.Go(func() error {
+		wg, _ := errgroup.WithContext(context.Background())
+
+		var err error
+		if len(data) > 0 {
+			data[0].Query(&relId)
+		}
+		for i := range data {
+			wg.Go(func() error {
+				ctx, cancel := context.WithTimeout(context.Background(), timeout)
+				defer cancel()
+
+				err = data[i].InsertCtx(ctx, tx, &relId)
+				return err
+			})
+		}
+
+		return wg.Wait()
+	})
+}
+
+func InsertSliceParallel[T AllowedIds, I Insertable[T]](g *errgroup.Group, tx *sqlx.Tx, data []I, relId InsertId[T]) {
+	g.Go(func() error {
+		wg, _ := errgroup.WithContext(context.Background())
+
+		var err error
+		if len(data) > 0 {
+			data[0].Query(&relId)
+		}
+		for i := range data {
+			wg.Go(func() error {
+				ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+				defer cancel()
+
+				err = data[i].InsertCtx(ctx, tx, &relId)
+				return err
+			})
+		}
+
+		return wg.Wait()
+	})
+}
+
+func InsertSimpleSlice[T AllowedIds, I any](tx *sqlx.Tx, data []I, relId *InsertId[T]) error {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	query := fmt.Sprintf(`
+		INSERT INTO %s (%s, value)
+		VALUES ($1, $2)`, relId.tableName, relId.fieldName)
+	for i := range data {
+		_, err := tx.ExecContext(ctx, query, relId.id, data[i])
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func InsertSimpleSliceParallel[T AllowedIds, I any](g *errgroup.Group, tx *sqlx.Tx, data []I, relId *InsertId[T]) {
+	g.Go(func() error {
+		wg, _ := errgroup.WithContext(context.Background())
+
+		query := fmt.Sprintf(`
+			INSERT INTO %s (%s, value)
+			VALUES ($1, $2)`, relId.tableName, relId.fieldName)
+		for i := range data {
+			wg.Go(func() error {
+				ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+				defer cancel()
+
+				_, err := tx.ExecContext(ctx, query, relId.id, data[i])
+				return err
+			})
+		}
+
+		return wg.Wait()
+	})
+}
+
+func DeleteSetupAll(db *sqlx.DB) error {
+	const query = `DELETE FROM setup`
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	tx := db.MustBegin()
+	defer tx.Rollback()
+
+	result, err := tx.ExecContext(ctx, query)
+	if err != nil {
+		return err
+	}
+	tx.Commit()
+
+	rowsAffected, err := result.RowsAffected()
+	if err != nil {
+		return err
+	}
+	if rowsAffected == 0 {
+		return ErrorRecordNotFound
+	}
+
+	return nil
+}
+
+func DeleteGameAll(db *sqlx.DB) error {
+	const query = `DELETE FROM game`
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	tx := db.MustBegin()
+	defer tx.Rollback()
+
+	result, err := tx.ExecContext(ctx, query)
+	if err != nil {
+		return err
+	}
+	tx.Commit()
+
+	rowsAffected, err := result.RowsAffected()
+	if err != nil {
+		return err
+	}
+	if rowsAffected == 0 {
+		return ErrorRecordNotFound
+	}
+
+	return nil
+}
+
+func DeleteSeqAll(db *sqlx.DB) error {
+	const query = `DELETE FROM sqlite_sequence`
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	tx := db.MustBegin()
+	defer tx.Rollback()
+
+	result, err := tx.ExecContext(ctx, query)
+	if err != nil {
+		return err
+	}
+	tx.Commit()
+
+	rowsAffected, err := result.RowsAffected()
+	if err != nil {
+		return err
+	}
+	if rowsAffected == 0 {
+		return ErrorRecordNotFound
+	}
+
+	return nil
+}
+
+func IsErrUniqueConstraint(err error) {
+	if sqliteErr, ok := err.(sqlite3.Error); ok {
+		// SQLITE_CONSTRAINT_UNIQUE (extended code 2067)
+		// or SQLITE_CONSTRAINT (basic code 19)
+		if sqliteErr.ExtendedCode == sqlite3.ErrConstraintUnique ||
+			sqliteErr.Code == sqlite3.ErrConstraint {
+		}
+	}
+}