63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
|
package infra
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"owl-blogs/app/repository"
|
||
|
|
"owl-blogs/domain/model"
|
||
|
|
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DefaultSiteConfigRepo struct {
|
||
|
|
db *sqlx.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewSiteConfigRepo(db Database) repository.SiteConfigRepository {
|
||
|
|
sqlxdb := db.Get()
|
||
|
|
|
||
|
|
sqlxdb.MustExec(`
|
||
|
|
CREATE TABLE IF NOT EXISTS site_config (
|
||
|
|
config TEXT
|
||
|
|
);
|
||
|
|
`)
|
||
|
|
|
||
|
|
return &DefaultSiteConfigRepo{
|
||
|
|
db: sqlxdb,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get implements repository.SiteConfigRepository.
|
||
|
|
func (r *DefaultSiteConfigRepo) Get() (model.SiteConfig, error) {
|
||
|
|
data := []byte{}
|
||
|
|
err := r.db.Get(&data, "SELECT config FROM site_config LIMIT 1")
|
||
|
|
if err != nil {
|
||
|
|
if err.Error() == "sql: no rows in result set" {
|
||
|
|
return model.SiteConfig{}, nil
|
||
|
|
}
|
||
|
|
return model.SiteConfig{}, err
|
||
|
|
}
|
||
|
|
if len(data) == 0 {
|
||
|
|
return model.SiteConfig{}, nil
|
||
|
|
}
|
||
|
|
config := model.SiteConfig{}
|
||
|
|
err = json.Unmarshal(data, &config)
|
||
|
|
return config, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update implements repository.SiteConfigRepository.
|
||
|
|
func (r *DefaultSiteConfigRepo) Update(siteConfig model.SiteConfig) error {
|
||
|
|
jsonData, err := json.Marshal(siteConfig)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
res, err := r.db.Exec("UPDATE site_config SET config = ?", jsonData)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
rows, err := res.RowsAffected()
|
||
|
|
if rows == 0 {
|
||
|
|
_, err = r.db.Exec("INSERT INTO site_config (config) VALUES (?)", jsonData)
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|