123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148 |
- package store
- import (
- "context"
- "database/sql"
- "strings"
- )
- type SystemSetting struct {
- Name string
- Value string
- Description string
- }
- type FindSystemSetting struct {
- Name string
- }
- func (s *Store) UpsertSystemSetting(ctx context.Context, upsert *SystemSetting) (*SystemSetting, error) {
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return nil, err
- }
- defer tx.Rollback()
- query := `
- INSERT INTO system_setting (
- name, value, description
- )
- VALUES (?, ?, ?)
- ON CONFLICT(name) DO UPDATE
- SET
- value = EXCLUDED.value,
- description = EXCLUDED.description
- `
- if _, err := tx.ExecContext(ctx, query, upsert.Name, upsert.Value, upsert.Description); err != nil {
- return nil, err
- }
- if err := tx.Commit(); err != nil {
- return nil, err
- }
- systemSetting := upsert
- return systemSetting, nil
- }
- func (s *Store) ListSystemSettings(ctx context.Context, find *FindSystemSetting) ([]*SystemSetting, error) {
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return nil, err
- }
- defer tx.Rollback()
- list, err := listSystemSettings(ctx, tx, find)
- if err != nil {
- return nil, err
- }
- if err := tx.Commit(); err != nil {
- return nil, err
- }
- for _, systemSettingMessage := range list {
- s.systemSettingCache.Store(systemSettingMessage.Name, systemSettingMessage)
- }
- return list, nil
- }
- func (s *Store) GetSystemSetting(ctx context.Context, find *FindSystemSetting) (*SystemSetting, error) {
- if find.Name != "" {
- if cache, ok := s.systemSettingCache.Load(find.Name); ok {
- return cache.(*SystemSetting), nil
- }
- }
- tx, err := s.db.BeginTx(ctx, nil)
- if err != nil {
- return nil, err
- }
- defer tx.Rollback()
- list, err := listSystemSettings(ctx, tx, find)
- if err != nil {
- return nil, err
- }
- if len(list) == 0 {
- return nil, nil
- }
- if err := tx.Commit(); err != nil {
- return nil, err
- }
- systemSettingMessage := list[0]
- s.systemSettingCache.Store(systemSettingMessage.Name, systemSettingMessage)
- return systemSettingMessage, nil
- }
- func (s *Store) GetSystemSettingValueWithDefault(ctx *context.Context, settingName string, defaultValue string) string {
- if setting, err := s.GetSystemSetting(*ctx, &FindSystemSetting{
- Name: settingName,
- }); err == nil && setting != nil {
- return setting.Value
- }
- return defaultValue
- }
- func listSystemSettings(ctx context.Context, tx *sql.Tx, find *FindSystemSetting) ([]*SystemSetting, error) {
- where, args := []string{"1 = 1"}, []any{}
- if find.Name != "" {
- where, args = append(where, "name = ?"), append(args, find.Name)
- }
- query := `
- SELECT
- name,
- value,
- description
- FROM system_setting
- WHERE ` + strings.Join(where, " AND ")
- rows, err := tx.QueryContext(ctx, query, args...)
- if err != nil {
- return nil, err
- }
- defer rows.Close()
- list := []*SystemSetting{}
- for rows.Next() {
- systemSettingMessage := &SystemSetting{}
- if err := rows.Scan(
- &systemSettingMessage.Name,
- &systemSettingMessage.Value,
- &systemSettingMessage.Description,
- ); err != nil {
- return nil, err
- }
- list = append(list, systemSettingMessage)
- }
- if err := rows.Err(); err != nil {
- return nil, err
- }
- return list, nil
- }
|