Merge pull request 'add param refresh to method orm.save' (#19) from orm_save_refresh into master

This commit is contained in:
medcl 2022-12-08 20:54:33 +08:00
commit 16bda2977c
8 changed files with 18 additions and 18 deletions

View File

@ -25,7 +25,7 @@ func (h *AlertAPI) createChannel(w http.ResponseWriter, req *http.Request, ps ht
return return
} }
err = orm.Create(obj) err = orm.Create(nil, obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)
@ -93,7 +93,7 @@ func (h *AlertAPI) updateChannel(w http.ResponseWriter, req *http.Request, ps ht
//protect //protect
obj.ID = id obj.ID = id
obj.Created = create obj.Created = create
err = orm.Update(&obj) err = orm.Update(nil, &obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)

View File

@ -63,7 +63,7 @@ func (alertAPI *AlertAPI) createRule(w http.ResponseWriter, req *http.Request, p
rule.Schedule.Interval = "1m" rule.Schedule.Interval = "1m"
} }
err = orm.Save(rule) err = orm.Save(nil, rule)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
alertAPI.WriteJSON(w, util.MapStr{ alertAPI.WriteJSON(w, util.MapStr{
@ -282,7 +282,7 @@ func (alertAPI *AlertAPI) updateRule(w http.ResponseWriter, req *http.Request, p
rule.Created = create rule.Created = create
rule.Updated = time.Now() rule.Updated = time.Now()
err = orm.Update(rule) err = orm.Update(nil, rule)
if err != nil { if err != nil {
alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError) alertAPI.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)
@ -598,7 +598,7 @@ func (alertAPI *AlertAPI) enableRule(w http.ResponseWriter, req *http.Request, p
clearKV(id) clearKV(id)
} }
obj.Enabled = reqObj.Enabled obj.Enabled = reqObj.Enabled
err = orm.Save(obj) err = orm.Save(nil, obj)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
alertAPI.WriteError(w, fmt.Sprintf("save rule error:%v", err), http.StatusInternalServerError) alertAPI.WriteError(w, fmt.Sprintf("save rule error:%v", err), http.StatusInternalServerError)

View File

@ -51,7 +51,7 @@ func (h *GatewayAPI) createInstance(w http.ResponseWriter, req *http.Request, ps
h.WriteError(w, "gateway instance already registered", http.StatusInternalServerError) h.WriteError(w, "gateway instance already registered", http.StatusInternalServerError)
return return
} }
err = orm.Create(obj) err = orm.Create(nil, obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)
@ -119,7 +119,7 @@ func (h *GatewayAPI) updateInstance(w http.ResponseWriter, req *http.Request, ps
//protect //protect
obj.ID = id obj.ID = id
obj.Created = create obj.Created = create
err = orm.Update(&obj) err = orm.Update(nil, &obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)

View File

@ -67,7 +67,7 @@ func (handler APIHandler) CreateDictItemAction(w http.ResponseWriter, req *http.
return return
} }
err = orm.Create(&dict) err = orm.Create(nil, &dict)
if err != nil { if err != nil {
resp["status"] = false resp["status"] = false
resp["error"] = err resp["error"] = err
@ -107,7 +107,7 @@ func (handler APIHandler) UpdateDictItemAction(w http.ResponseWriter, req *http.
} }
dict.UpdatedAt = time.Now() dict.UpdatedAt = time.Now()
err = orm.Update(dict) err = orm.Update(nil, dict)
if err != nil { if err != nil {
resp["status"] = false resp["status"] = false
resp["error"] = err resp["error"] = err

View File

@ -22,7 +22,7 @@ func (h *InsightAPI) createDashboard(w http.ResponseWriter, req *http.Request, p
log.Error(err) log.Error(err)
return return
} }
err = orm.Create(obj) err = orm.Create(nil, obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)
@ -102,7 +102,7 @@ func (h *InsightAPI) updateDashboard(w http.ResponseWriter, req *http.Request, p
//protect //protect
obj.ID = id obj.ID = id
obj.Created = create obj.Created = create
err = orm.Update(&obj) err = orm.Update(nil, &obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)

View File

@ -6,8 +6,8 @@ package insight
import ( import (
"fmt" "fmt"
"infini.sh/framework/core/insight"
httprouter "infini.sh/framework/core/api/router" httprouter "infini.sh/framework/core/api/router"
"infini.sh/framework/core/insight"
"infini.sh/framework/core/orm" "infini.sh/framework/core/orm"
"infini.sh/framework/core/util" "infini.sh/framework/core/util"
"net/http" "net/http"
@ -24,7 +24,7 @@ func (h *InsightAPI) createVisualization(w http.ResponseWriter, req *http.Reques
log.Error(err) log.Error(err)
return return
} }
err = orm.Create(obj) err = orm.Create(nil, obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)
@ -87,7 +87,7 @@ func (h *InsightAPI) updateVisualization(w http.ResponseWriter, req *http.Reques
//protect //protect
obj.ID = id obj.ID = id
obj.Created = create obj.Created = create
err = orm.Update(&obj) err = orm.Update(nil, &obj)
if err != nil { if err != nil {
h.WriteError(w, err.Error(), http.StatusInternalServerError) h.WriteError(w, err.Error(), http.StatusInternalServerError)
log.Error(err) log.Error(err)

View File

@ -411,7 +411,7 @@ func (module *Module) initialize(w http.ResponseWriter, r *http.Request, ps http
security.InitSecurity() security.InitSecurity()
//保存默认集群 //保存默认集群
err=orm.Save(&cfg) err=orm.Save(nil, &cfg)
if err!=nil{ if err!=nil{
panic(err) panic(err)
} }
@ -435,7 +435,7 @@ func (module *Module) initialize(w http.ResponseWriter, r *http.Request, ps http
Name: rbac.RoleAdminName, Name: rbac.RoleAdminName,
}) })
user.Roles=role user.Roles=role
err=orm.Save(&user) err=orm.Save(nil, &user)
if err!=nil{ if err!=nil{
panic(err) panic(err)
} }

View File

@ -626,7 +626,7 @@ func (engine *Engine) Do(rule *alerting.Rule) error {
} }
} }
err = orm.Save(alertItem) err = orm.Save(nil, alertItem)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
} }
@ -1131,7 +1131,7 @@ func getLastAlertMessage(ruleID string, duration time.Duration) (*alerting.Alert
func saveAlertMessageToES(message *alerting.AlertMessage) error { func saveAlertMessageToES(message *alerting.AlertMessage) error {
message.Updated = time.Now() message.Updated = time.Now()
return orm.Save(message) return orm.Save(nil, message)
} }
func saveAlertMessage(message *alerting.AlertMessage) error { func saveAlertMessage(message *alerting.AlertMessage) error {