60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
|
|
"git.gobha.me/xcaliber/chat-switchboard/models"
|
|
)
|
|
|
|
type GlobalConfigStore struct{}
|
|
|
|
func NewGlobalConfigStore() *GlobalConfigStore { return &GlobalConfigStore{} }
|
|
|
|
func (s *GlobalConfigStore) Get(ctx context.Context, key string) (models.JSONMap, error) {
|
|
var valueJSON []byte
|
|
err := DB.QueryRowContext(ctx,
|
|
"SELECT value FROM global_settings WHERE key = $1", key).Scan(&valueJSON)
|
|
if err == sql.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var result models.JSONMap
|
|
json.Unmarshal(valueJSON, &result)
|
|
return result, nil
|
|
}
|
|
|
|
func (s *GlobalConfigStore) Set(ctx context.Context, key string, value models.JSONMap, updatedBy string) error {
|
|
valueJSON := ToJSON(value)
|
|
_, err := DB.ExecContext(ctx, `
|
|
INSERT INTO global_settings (key, value, updated_by, updated_at)
|
|
VALUES ($1, $2, $3, NOW())
|
|
ON CONFLICT (key) DO UPDATE SET value = $2, updated_by = $3, updated_at = NOW()`,
|
|
key, valueJSON, updatedBy)
|
|
return err
|
|
}
|
|
|
|
func (s *GlobalConfigStore) GetAll(ctx context.Context) (map[string]models.JSONMap, error) {
|
|
rows, err := DB.QueryContext(ctx, "SELECT key, value FROM global_settings")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
result := make(map[string]models.JSONMap)
|
|
for rows.Next() {
|
|
var key string
|
|
var valueJSON []byte
|
|
if err := rows.Scan(&key, &valueJSON); err != nil {
|
|
continue
|
|
}
|
|
var m models.JSONMap
|
|
json.Unmarshal(valueJSON, &m)
|
|
result[key] = m
|
|
}
|
|
return result, rows.Err()
|
|
}
|