You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
882 B
39 lines
882 B
package models
|
|
|
|
import "database/sql"
|
|
|
|
type ConfigStore struct {
|
|
DB *sql.DB
|
|
}
|
|
|
|
func (s *ConfigStore) GetAll() (map[string]string, error) {
|
|
rows, err := s.DB.Query(`SELECT key, value FROM site_config`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
config := make(map[string]string)
|
|
for rows.Next() {
|
|
var k, v string
|
|
if err := rows.Scan(&k, &v); err != nil {
|
|
return nil, err
|
|
}
|
|
config[k] = v
|
|
}
|
|
return config, rows.Err()
|
|
}
|
|
|
|
func (s *ConfigStore) Get(key string) (string, error) {
|
|
var value string
|
|
err := s.DB.QueryRow(`SELECT value FROM site_config WHERE key = ?`, key).Scan(&value)
|
|
if err == sql.ErrNoRows {
|
|
return "", nil
|
|
}
|
|
return value, err
|
|
}
|
|
|
|
func (s *ConfigStore) Set(key, value string) error {
|
|
_, err := s.DB.Exec(`INSERT INTO site_config (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = ?`, key, value, value)
|
|
return err
|
|
}
|
|
|