|
@@ -1,6 +1,12 @@
|
|
|
package data_manage_permission
|
|
|
|
|
|
-import "time"
|
|
|
+import (
|
|
|
+ "eta/eta_api/utils"
|
|
|
+ "fmt"
|
|
|
+ "github.com/beego/beego/v2/client/orm"
|
|
|
+ "strings"
|
|
|
+ "time"
|
|
|
+)
|
|
|
|
|
|
|
|
|
|
|
@@ -16,3 +22,123 @@ type DataPermissionMessage struct {
|
|
|
CreateTime time.Time `json:"create_time"`
|
|
|
ModifyTime time.Time `json:"modify_time"`
|
|
|
}
|
|
|
+
|
|
|
+var DataPermissionMessageCols = struct {
|
|
|
+ DataPermissionMessageId string
|
|
|
+ SendUserId string
|
|
|
+ ReceiveUserId string
|
|
|
+ Content string
|
|
|
+ Remark string
|
|
|
+ OpType string
|
|
|
+ OpUniqueCode string
|
|
|
+ IsRead string
|
|
|
+ CreateTime string
|
|
|
+ ModifyTime string
|
|
|
+}{
|
|
|
+ DataPermissionMessageId: "data_permission_message_id",
|
|
|
+ SendUserId: "send_user_id",
|
|
|
+ ReceiveUserId: "receive_user_id",
|
|
|
+ Content: "content",
|
|
|
+ Remark: "remark",
|
|
|
+ OpType: "op_type",
|
|
|
+ OpUniqueCode: "op_unique_code",
|
|
|
+ IsRead: "is_read",
|
|
|
+ CreateTime: "create_time",
|
|
|
+ ModifyTime: "modify_time",
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) TableName() string {
|
|
|
+ return "data_permission_message"
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) PrimaryId() string {
|
|
|
+ return DataPermissionMessageCols.DataPermissionMessageId
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) Create() (err error) {
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ id, err := o.Insert(m)
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ m.DataPermissionMessageId = id
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) CreateMulti(items []*DataPermissionMessage) (err error) {
|
|
|
+ if len(items) == 0 {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ _, err = o.InsertMulti(500, items)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) Update(cols []string) (err error) {
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ _, err = o.Update(m, cols...)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) Del() (err error) {
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
|
|
|
+ _, err = o.Raw(sql, m.DataPermissionMessageId).Exec()
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) MultiDel(menuIds []int) (err error) {
|
|
|
+ if len(menuIds) == 0 {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ sql := fmt.Sprintf(`DELETE FROM %s WHERE %s IN (%s)`, m.TableName(), m.PrimaryId(), utils.GetOrmInReplace(len(menuIds)))
|
|
|
+ _, err = o.Raw(sql, menuIds).Exec()
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) GetItemById(id int) (item *DataPermissionMessage, err error) {
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
|
|
|
+ err = o.Raw(sql, id).QueryRow(&item)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
|
|
|
+ err = o.Raw(sql, pars).QueryRow(&count)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (m *DataPermissionMessage) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*DataPermissionMessage, err error) {
|
|
|
+ o := orm.NewOrmUsingDB("data")
|
|
|
+ fields := strings.Join(fieldArr, ",")
|
|
|
+ if len(fieldArr) == 0 {
|
|
|
+ fields = `*`
|
|
|
+ }
|
|
|
+ order := `ORDER BY create_time DESC`
|
|
|
+ if orderRule != "" {
|
|
|
+ order = ` ORDER BY ` + orderRule
|
|
|
+ }
|
|
|
+ sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
|
|
|
+ _, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+func FormatDataPermissionMessage2Item(origin *DataPermissionMessage) (item *DataPermissionMessageItem) {
|
|
|
+ item = &DataPermissionMessageItem{
|
|
|
+ DataPermissionMessageId: origin.DataPermissionMessageId,
|
|
|
+ SendUserId: origin.SendUserId,
|
|
|
+ ReceiveUserId: origin.ReceiveUserId,
|
|
|
+ Content: origin.Content,
|
|
|
+ Remark: origin.Remark,
|
|
|
+ OpType: origin.OpType,
|
|
|
+ OpUniqueCode: origin.OpUniqueCode,
|
|
|
+ IsRead: origin.IsRead,
|
|
|
+ CreateTime: utils.TimeTransferString(utils.FormatDateTime, origin.CreateTime),
|
|
|
+ ModifyTime: utils.TimeTransferString(utils.FormatDateTime, origin.ModifyTime),
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|