mirror of https://github.com/muety/wakapi.git
63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
package services
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/muety/wakapi/models"
|
|
)
|
|
|
|
type KeyValueService struct {
|
|
Config *models.Config
|
|
Db *gorm.DB
|
|
}
|
|
|
|
func NewKeyValueService(db *gorm.DB) *KeyValueService {
|
|
return &KeyValueService{
|
|
Config: models.GetConfig(),
|
|
Db: db,
|
|
}
|
|
}
|
|
|
|
func (srv *KeyValueService) GetString(key string) (*models.KeyStringValue, error) {
|
|
kv := &models.KeyStringValue{}
|
|
if err := srv.Db.
|
|
Where(&models.KeyStringValue{Key: key}).
|
|
First(&kv).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return kv, nil
|
|
}
|
|
|
|
func (srv *KeyValueService) PutString(kv *models.KeyStringValue) error {
|
|
result := srv.Db.
|
|
Where(&models.KeyStringValue{Key: kv.Key}).
|
|
Assign(kv).
|
|
FirstOrCreate(kv)
|
|
|
|
if err := result.Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
if result.RowsAffected != 1 {
|
|
return errors.New("nothing updated")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (srv *KeyValueService) DeleteString(key string) error {
|
|
result := srv.Db.
|
|
Delete(&models.KeyStringValue{}, &models.KeyStringValue{Key: key})
|
|
|
|
if err := result.Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
if result.RowsAffected != 1 {
|
|
return errors.New("nothing deleted")
|
|
}
|
|
|
|
return nil
|
|
}
|