|
@@ -3,7 +3,6 @@ package sqlstore
|
|
|
import (
|
|
import (
|
|
|
"bytes"
|
|
"bytes"
|
|
|
"fmt"
|
|
"fmt"
|
|
|
- "strconv"
|
|
|
|
|
"time"
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/go-xorm/xorm"
|
|
"github.com/go-xorm/xorm"
|
|
@@ -31,11 +30,11 @@ func DeleteAlertNotification(cmd *m.DeleteAlertNotificationCommand) error {
|
|
|
})
|
|
})
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func AlertNotificationQuery(query *m.GetAlertNotificationQuery) error {
|
|
|
|
|
|
|
+func AlertNotificationQuery(query *m.GetAlertNotificationsQuery) error {
|
|
|
return getAlertNotifications(query, x.NewSession())
|
|
return getAlertNotifications(query, x.NewSession())
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func getAlertNotifications(query *m.GetAlertNotificationQuery, sess *xorm.Session) error {
|
|
|
|
|
|
|
+func getAlertNotifications(query *m.GetAlertNotificationsQuery, sess *xorm.Session) error {
|
|
|
var sql bytes.Buffer
|
|
var sql bytes.Buffer
|
|
|
params := make([]interface{}, 0)
|
|
params := make([]interface{}, 0)
|
|
|
|
|
|
|
@@ -43,16 +42,15 @@ func getAlertNotifications(query *m.GetAlertNotificationQuery, sess *xorm.Sessio
|
|
|
alert_notification.id,
|
|
alert_notification.id,
|
|
|
alert_notification.org_id,
|
|
alert_notification.org_id,
|
|
|
alert_notification.name,
|
|
alert_notification.name,
|
|
|
- alert_notification.type,
|
|
|
|
|
|
|
+ alert_notification.type,
|
|
|
alert_notification.created,
|
|
alert_notification.created,
|
|
|
- alert_notification.updated,
|
|
|
|
|
- alert_notification.settings,
|
|
|
|
|
- alert_notification.always_execute
|
|
|
|
|
|
|
+ alert_notification.updated,
|
|
|
|
|
+ alert_notification.settings
|
|
|
FROM alert_notification
|
|
FROM alert_notification
|
|
|
`)
|
|
`)
|
|
|
|
|
|
|
|
sql.WriteString(` WHERE alert_notification.org_id = ?`)
|
|
sql.WriteString(` WHERE alert_notification.org_id = ?`)
|
|
|
- params = append(params, query.OrgID)
|
|
|
|
|
|
|
+ params = append(params, query.OrgId)
|
|
|
|
|
|
|
|
if query.Name != "" {
|
|
if query.Name != "" {
|
|
|
sql.WriteString(` AND alert_notification.name = ?`)
|
|
sql.WriteString(` AND alert_notification.name = ?`)
|
|
@@ -61,60 +59,26 @@ func getAlertNotifications(query *m.GetAlertNotificationQuery, sess *xorm.Sessio
|
|
|
|
|
|
|
|
if query.Id != 0 {
|
|
if query.Id != 0 {
|
|
|
sql.WriteString(` AND alert_notification.id = ?`)
|
|
sql.WriteString(` AND alert_notification.id = ?`)
|
|
|
- params = append(params, strconv.Itoa(int(query.Id)))
|
|
|
|
|
|
|
+ params = append(params, query.Id)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
if len(query.Ids) > 0 {
|
|
if len(query.Ids) > 0 {
|
|
|
- sql.WriteString(` AND (`)
|
|
|
|
|
-
|
|
|
|
|
- for i, id := range query.Ids {
|
|
|
|
|
- if i != 0 {
|
|
|
|
|
- sql.WriteString(` OR`)
|
|
|
|
|
- }
|
|
|
|
|
- sql.WriteString(` alert_notification.id = ?`)
|
|
|
|
|
- params = append(params, id)
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- sql.WriteString(`)`)
|
|
|
|
|
|
|
+ sql.WriteString(` AND alert_notification.id IN (?)`)
|
|
|
|
|
+ params = append(params, query.Ids)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- var searches []*m.AlertNotification
|
|
|
|
|
- if err := sess.Sql(sql.String(), params...).Find(&searches); err != nil {
|
|
|
|
|
|
|
+ results := make([]*m.AlertNotification, 0)
|
|
|
|
|
+ if err := sess.Sql(sql.String(), params...).Find(&results); err != nil {
|
|
|
return err
|
|
return err
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- var result []*m.AlertNotification
|
|
|
|
|
- var def []*m.AlertNotification
|
|
|
|
|
- if query.IncludeAlwaysExecute {
|
|
|
|
|
-
|
|
|
|
|
- if err := sess.Where("org_id = ? AND always_execute = 1", query.OrgID).Find(&def); err != nil {
|
|
|
|
|
- return err
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- result = append(result, def...)
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- for _, s := range searches {
|
|
|
|
|
- canAppend := true
|
|
|
|
|
- for _, d := range result {
|
|
|
|
|
- if d.Id == s.Id {
|
|
|
|
|
- canAppend = false
|
|
|
|
|
- break
|
|
|
|
|
- }
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if canAppend {
|
|
|
|
|
- result = append(result, s)
|
|
|
|
|
- }
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- query.Result = result
|
|
|
|
|
|
|
+ query.Result = results
|
|
|
return nil
|
|
return nil
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func CreateAlertNotificationCommand(cmd *m.CreateAlertNotificationCommand) error {
|
|
func CreateAlertNotificationCommand(cmd *m.CreateAlertNotificationCommand) error {
|
|
|
return inTransaction(func(sess *xorm.Session) error {
|
|
return inTransaction(func(sess *xorm.Session) error {
|
|
|
- existingQuery := &m.GetAlertNotificationQuery{OrgID: cmd.OrgID, Name: cmd.Name, IncludeAlwaysExecute: false}
|
|
|
|
|
|
|
+ existingQuery := &m.GetAlertNotificationsQuery{OrgId: cmd.OrgId, Name: cmd.Name}
|
|
|
err := getAlertNotifications(existingQuery, sess)
|
|
err := getAlertNotifications(existingQuery, sess)
|
|
|
|
|
|
|
|
if err != nil {
|
|
if err != nil {
|
|
@@ -126,18 +90,15 @@ func CreateAlertNotificationCommand(cmd *m.CreateAlertNotificationCommand) error
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
alertNotification := &m.AlertNotification{
|
|
alertNotification := &m.AlertNotification{
|
|
|
- OrgId: cmd.OrgID,
|
|
|
|
|
- Name: cmd.Name,
|
|
|
|
|
- Type: cmd.Type,
|
|
|
|
|
- Created: time.Now(),
|
|
|
|
|
- Settings: cmd.Settings,
|
|
|
|
|
- Updated: time.Now(),
|
|
|
|
|
- AlwaysExecute: cmd.AlwaysExecute,
|
|
|
|
|
|
|
+ OrgId: cmd.OrgId,
|
|
|
|
|
+ Name: cmd.Name,
|
|
|
|
|
+ Type: cmd.Type,
|
|
|
|
|
+ Settings: cmd.Settings,
|
|
|
|
|
+ Created: time.Now(),
|
|
|
|
|
+ Updated: time.Now(),
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- _, err = sess.Insert(alertNotification)
|
|
|
|
|
-
|
|
|
|
|
- if err != nil {
|
|
|
|
|
|
|
+ if _, err = sess.Insert(alertNotification); err != nil {
|
|
|
return err
|
|
return err
|
|
|
}
|
|
}
|
|
|
|
|
|
|
@@ -148,38 +109,34 @@ func CreateAlertNotificationCommand(cmd *m.CreateAlertNotificationCommand) error
|
|
|
|
|
|
|
|
func UpdateAlertNotification(cmd *m.UpdateAlertNotificationCommand) error {
|
|
func UpdateAlertNotification(cmd *m.UpdateAlertNotificationCommand) error {
|
|
|
return inTransaction(func(sess *xorm.Session) (err error) {
|
|
return inTransaction(func(sess *xorm.Session) (err error) {
|
|
|
- current := &m.AlertNotification{}
|
|
|
|
|
- _, err = sess.Id(cmd.Id).Get(current)
|
|
|
|
|
|
|
+ current := m.AlertNotification{}
|
|
|
|
|
|
|
|
- if err != nil {
|
|
|
|
|
|
|
+ if _, err = sess.Id(cmd.Id).Get(¤t); err != nil {
|
|
|
return err
|
|
return err
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- alertNotification := &m.AlertNotification{
|
|
|
|
|
- Id: cmd.Id,
|
|
|
|
|
- OrgId: cmd.OrgID,
|
|
|
|
|
- Name: cmd.Name,
|
|
|
|
|
- Type: cmd.Type,
|
|
|
|
|
- Settings: cmd.Settings,
|
|
|
|
|
- Updated: time.Now(),
|
|
|
|
|
- Created: current.Created,
|
|
|
|
|
- AlwaysExecute: cmd.AlwaysExecute,
|
|
|
|
|
|
|
+ // check if name exists
|
|
|
|
|
+ sameNameQuery := &m.GetAlertNotificationsQuery{OrgId: cmd.OrgId, Name: cmd.Name}
|
|
|
|
|
+ if err := getAlertNotifications(sameNameQuery, sess); err != nil {
|
|
|
|
|
+ return err
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- sess.UseBool("always_execute")
|
|
|
|
|
|
|
+ if len(sameNameQuery.Result) > 0 && sameNameQuery.Result[0].Id != current.Id {
|
|
|
|
|
+ return fmt.Errorf("Alert notification name %s already exists", cmd.Name)
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- var affected int64
|
|
|
|
|
- affected, err = sess.Id(alertNotification.Id).Update(alertNotification)
|
|
|
|
|
|
|
+ current.Updated = time.Now()
|
|
|
|
|
+ current.Settings = cmd.Settings
|
|
|
|
|
+ current.Name = cmd.Name
|
|
|
|
|
+ current.Type = cmd.Type
|
|
|
|
|
|
|
|
- if err != nil {
|
|
|
|
|
|
|
+ if affected, err := sess.Id(cmd.Id).Update(current); err != nil {
|
|
|
return err
|
|
return err
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if affected == 0 {
|
|
|
|
|
|
|
+ } else if affected == 0 {
|
|
|
return fmt.Errorf("Could not find alert notification")
|
|
return fmt.Errorf("Could not find alert notification")
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- cmd.Result = alertNotification
|
|
|
|
|
|
|
+ cmd.Result = ¤t
|
|
|
return nil
|
|
return nil
|
|
|
})
|
|
})
|
|
|
}
|
|
}
|