8bfeb7d90d
- update fatih/color (fix windows issue) - update mongo-driver (fix build issue) - go.mod: merge two "require" blocks - update semver dependency (same version as indirect dep), fix test checks in cscli setup - remove gotest.tools dependency (use testify, cstest) - update x/ exp, mod, sys dependencies
34 lines
1,019 B
Go
34 lines
1,019 B
Go
package database
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/database/ent"
|
|
"github.com/crowdsecurity/crowdsec/pkg/database/ent/configitem"
|
|
)
|
|
|
|
func (c *Client) GetConfigItem(key string) (*string, error) {
|
|
result, err := c.Ent.ConfigItem.Query().Where(configitem.NameEQ(key)).First(c.CTX)
|
|
if err != nil && ent.IsNotFound(err) {
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
return nil, errors.Wrapf(QueryFail, "select config item: %s", err)
|
|
}
|
|
|
|
return &result.Value, nil
|
|
}
|
|
|
|
func (c *Client) SetConfigItem(key string, value string) error {
|
|
|
|
nbUpdated, err := c.Ent.ConfigItem.Update().SetValue(value).Where(configitem.NameEQ(key)).Save(c.CTX)
|
|
if (err != nil && ent.IsNotFound(err)) || nbUpdated == 0 { //not found, create
|
|
err := c.Ent.ConfigItem.Create().SetName(key).SetValue(value).Exec(c.CTX)
|
|
if err != nil {
|
|
return errors.Wrapf(QueryFail, "insert config item: %s", err)
|
|
}
|
|
} else if err != nil {
|
|
return errors.Wrapf(QueryFail, "update config item: %s", err)
|
|
}
|
|
return nil
|
|
}
|