117 lines
3.8 KiB
Go
117 lines
3.8 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.zaynet.fr/DoGaMa/DoGaMa-serv/internal/instance"
|
|
)
|
|
|
|
const globalLabelsKey = "game_container_labels"
|
|
|
|
func (r *Repository) GetGlobalLabels(ctx context.Context) (map[string]string, error) {
|
|
var body string
|
|
err := r.db.QueryRowContext(ctx, `SELECT value_json FROM system_settings WHERE key=?`, globalLabelsKey).Scan(&body)
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return map[string]string{}, nil
|
|
}
|
|
if err != nil {
|
|
return nil, fmt.Errorf("load global game-container labels: %w", err)
|
|
}
|
|
var labels map[string]string
|
|
if err := json.Unmarshal([]byte(body), &labels); err != nil {
|
|
return nil, fmt.Errorf("decode global game-container labels: %w", err)
|
|
}
|
|
return labels, nil
|
|
}
|
|
|
|
func (r *Repository) SetGlobalLabels(ctx context.Context, labels map[string]string, pending bool) (int, int, error) {
|
|
tx, err := r.db.BeginTx(ctx, nil)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
now := r.now().UTC().Format(time.RFC3339Nano)
|
|
body := string(mustJSON(labels))
|
|
if _, err := tx.ExecContext(ctx, `INSERT INTO system_settings(key,value_json,revision,updated_at) VALUES(?,?,1,?) ON CONFLICT(key) DO UPDATE SET value_json=excluded.value_json, revision=system_settings.revision+1, updated_at=excluded.updated_at`, globalLabelsKey, body, now); err != nil {
|
|
return 0, 0, err
|
|
}
|
|
rows, err := tx.QueryContext(ctx, `SELECT id, preview_json, desired_running FROM instances WHERE deleted_at IS NULL AND lifecycle_state!='draft'`)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
type update struct {
|
|
id, body string
|
|
running bool
|
|
}
|
|
var updates []update
|
|
for rows.Next() {
|
|
var id, previewBody string
|
|
var running int
|
|
if err := rows.Scan(&id, &previewBody, &running); err != nil {
|
|
rows.Close()
|
|
return 0, 0, err
|
|
}
|
|
var preview instance.Preview
|
|
if json.Unmarshal([]byte(previewBody), &preview) != nil {
|
|
rows.Close()
|
|
return 0, 0, errors.New("decode instance configuration")
|
|
}
|
|
preview.GlobalLabels = labels
|
|
updates = append(updates, update{id: id, body: string(mustJSON(preview)), running: running != 0})
|
|
}
|
|
rows.Close()
|
|
running := 0
|
|
pendingValue := 0
|
|
if pending {
|
|
pendingValue = 1
|
|
}
|
|
for _, update := range updates {
|
|
if update.running {
|
|
running++
|
|
}
|
|
if _, err := tx.ExecContext(ctx, `UPDATE instances SET preview_json=?, container_config_pending=?, revision=revision+1, updated_at=? WHERE id=?`, update.body, pendingValue, now, update.id); err != nil {
|
|
return 0, 0, err
|
|
}
|
|
}
|
|
if err := tx.Commit(); err != nil {
|
|
return 0, 0, err
|
|
}
|
|
return len(updates), running, nil
|
|
}
|
|
|
|
func (r *Repository) SaveInstanceConfiguration(ctx context.Context, id string, preview instance.Preview, pending bool) error {
|
|
body := string(mustJSON(preview))
|
|
labels := string(mustJSON(preview.CustomLabels))
|
|
pendingValue := 0
|
|
if pending {
|
|
pendingValue = 1
|
|
}
|
|
result, err := r.db.ExecContext(ctx, `UPDATE instances SET preview_json=?, custom_labels_json=?, image_tag_mode=?, image_tag=?, container_config_pending=?, revision=revision+1, updated_at=? WHERE id=? AND deleted_at IS NULL`, body, labels, preview.ImageTag.Mode, preview.ImageTag.Tag, pendingValue, r.now().UTC().Format(time.RFC3339Nano), id)
|
|
if err != nil {
|
|
return fmt.Errorf("save instance container configuration: %w", err)
|
|
}
|
|
changed, _ := result.RowsAffected()
|
|
if changed != 1 {
|
|
return instance.ErrInstanceNotFound
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *Repository) ClearContainerConfigPending(ctx context.Context, id, planDigest string) error {
|
|
_, err := r.db.ExecContext(ctx, `UPDATE instances SET container_config_pending=0, plan_digest=?, updated_at=? WHERE id=?`, planDigest, r.now().UTC().Format(time.RFC3339Nano), id)
|
|
return err
|
|
}
|
|
|
|
func mustJSON(value any) []byte {
|
|
body, err := json.Marshal(value)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return body
|
|
}
|