add setup insert methods
This commit is contained in:
@@ -1,21 +1,72 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type Setup struct {
|
||||
ID uint
|
||||
ID uint
|
||||
CreatedAt time.Time
|
||||
|
||||
IsAdmin bool
|
||||
IsSetup bool
|
||||
CoreUpdate CoreUpdate
|
||||
FeaturedContent FeaturedContent
|
||||
Files Files
|
||||
Options SetupOptions
|
||||
Release Release
|
||||
Languages []SetupLanguage
|
||||
Modules []Module
|
||||
News []News
|
||||
PackageWarnings []PackageWarning
|
||||
Systems []System
|
||||
Worlds []World
|
||||
CoreUpdate CoreUpdate //+
|
||||
FeaturedContent FeaturedContent //+
|
||||
Files Files //+
|
||||
Options *SetupOptions //+
|
||||
Release Release //+
|
||||
Languages []SetupLanguage //+
|
||||
Modules []*Module
|
||||
News []News //+
|
||||
PackageWarnings []*PackageWarning
|
||||
Systems []*System
|
||||
Worlds []*World
|
||||
}
|
||||
|
||||
func (s *Setup) Insert(db *sqlx.DB) error {
|
||||
tx := db.MustBegin()
|
||||
defer tx.Rollback()
|
||||
|
||||
const query = `
|
||||
INSERT INTO setup (is_admin, is_setup)
|
||||
VALUES ($1, $2)
|
||||
RETURNING id, created_at`
|
||||
|
||||
args := []any{s.IsAdmin, s.IsSetup}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
|
||||
err := tx.QueryRowContext(ctx, query, args...).Scan(&s.ID, &s.CreatedAt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
id := InsertId[uint]{id: s.ID, fieldName: "setup_id"}
|
||||
|
||||
dataSync := SyncDbOperations{
|
||||
wg: sync.WaitGroup{},
|
||||
errChan: make(chan error),
|
||||
}
|
||||
defer close(dataSync.errChan)
|
||||
|
||||
go InsertSliceParallel(&dataSync, tx, s.News, &id)
|
||||
go InsertSliceParallel(&dataSync, tx, s.Languages, &id)
|
||||
|
||||
InsertWithCtx(tx, &s.FeaturedContent, &id)
|
||||
InsertWithCtx(tx, &s.Files, &id)
|
||||
InsertWithCtx(tx, &s.Release, &id)
|
||||
InsertWithCtx(tx, &s.CoreUpdate, &id)
|
||||
|
||||
err = dataSync.Wait()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
type FeaturedContent struct {
|
||||
@@ -27,6 +78,38 @@ type FeaturedContent struct {
|
||||
Image string
|
||||
}
|
||||
|
||||
func (f *FeaturedContent) Insert(tx *sqlx.Tx, setup *InsertId[uint]) error {
|
||||
const query = `
|
||||
INSERT INTO featured_content (setup_id, title, caption, url, image)
|
||||
VALUES ($1, $2, $3, $4, $5)
|
||||
RETURNING id`
|
||||
|
||||
args := []any{setup.id, f.Title, f.Caption, f.URL, f.Image}
|
||||
|
||||
err := tx.QueryRowx(query, args...).Scan(&f.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *FeaturedContent) InsertCtx(ctx context.Context, tx *sqlx.Tx, setup *InsertId[uint]) error {
|
||||
const query = `
|
||||
INSERT INTO featured_content (setup_id, title, caption, url, image)
|
||||
VALUES ($1, $2, $3, $4, $5)
|
||||
RETURNING id`
|
||||
|
||||
args := []any{setup.id, f.Title, f.Caption, f.URL, f.Image}
|
||||
|
||||
err := tx.QueryRowxContext(ctx, query, args...).Scan(&f.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type News struct {
|
||||
ID uint
|
||||
|
||||
@@ -35,3 +118,35 @@ type News struct {
|
||||
URL string
|
||||
Image string
|
||||
}
|
||||
|
||||
func (n News) Insert(tx *sqlx.Tx, setup *InsertId[uint]) error {
|
||||
const insertNews = `
|
||||
INSERT INTO news (setup_id, title, caption, url, image)
|
||||
VALUES ($1, $2, $3, $4, $5)
|
||||
RETURNING id`
|
||||
|
||||
args := []any{setup.id, n.Title, n.Caption, n.URL, n.Image}
|
||||
|
||||
err := tx.QueryRow(insertNews, args...).Scan(&n.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (n News) InsertCtx(ctx context.Context, tx *sqlx.Tx, setup *InsertId[uint]) error {
|
||||
const insertNews = `
|
||||
INSERT INTO news (setup_id, title, caption, url, image)
|
||||
VALUES ($1, $2, $3, $4, $5)
|
||||
RETURNING id`
|
||||
|
||||
args := []any{setup.id, n.Title, n.Caption, n.URL, n.Image}
|
||||
|
||||
err := tx.QueryRowContext(ctx, insertNews, args...).Scan(&n.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user