105 lines
2.1 KiB
Go
105 lines
2.1 KiB
Go
package models
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/google/uuid"
|
|
"github.com/jinzhu/gorm"
|
|
)
|
|
|
|
type SendWays struct {
|
|
UUIDModel
|
|
|
|
Name string `json:"name"`
|
|
Type string `json:"type"`
|
|
Auth string `gorm:"not null" json:"auth"`
|
|
}
|
|
|
|
func AddSendWay(name string, auth string, wayType string, createdBy string, modifiedBy string) error {
|
|
newUUID := uuid.New()
|
|
way := SendWays{
|
|
UUIDModel: UUIDModel{
|
|
ID: newUUID,
|
|
CreatedBy: createdBy,
|
|
ModifiedBy: modifiedBy,
|
|
},
|
|
Name: name,
|
|
Type: wayType,
|
|
Auth: auth,
|
|
}
|
|
if err := db.Create(&way).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetSendWays(pageNum int, pageSize int, name string, type_ string, maps interface{}) ([]SendWays, error) {
|
|
var (
|
|
ways []SendWays
|
|
err error
|
|
)
|
|
query := db.Where(maps)
|
|
if name != "" {
|
|
query = query.Where("name like ?", fmt.Sprintf("%%%s%%", name))
|
|
}
|
|
if type_ != "" {
|
|
query = query.Where("type = ?", type_)
|
|
}
|
|
|
|
query = query.Order("created_on DESC")
|
|
if pageSize > 0 || pageNum > 0 {
|
|
query = query.Offset(pageNum).Limit(pageSize)
|
|
}
|
|
|
|
err = query.Find(&ways).Error
|
|
if err != nil && err != gorm.ErrRecordNotFound {
|
|
return nil, err
|
|
}
|
|
return ways, nil
|
|
}
|
|
|
|
func GetSendWaysTotal(name string, type_ string, maps interface{}) (int, error) {
|
|
var (
|
|
err error
|
|
total int
|
|
)
|
|
query := db.Model(&SendWays{}).Where(maps)
|
|
if name != "" {
|
|
query = query.Where("name like ?", fmt.Sprintf("%%%s%%", name))
|
|
}
|
|
if type_ != "" {
|
|
query = query.Where("type = ?", type_)
|
|
}
|
|
|
|
err = query.Count(&total).Error
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return total, nil
|
|
}
|
|
|
|
func GetWayByID(id string) (SendWays, error) {
|
|
var way SendWays
|
|
err := db.Where("id = ? ", id).Find(&way).Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return way, err
|
|
}
|
|
return way, nil
|
|
}
|
|
|
|
func DeleteMsgWay(id string) error {
|
|
if err := db.Where("id = ?", id).Delete(&SendWays{}).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func EditSendWay(id string, data interface{}) error {
|
|
if err := db.Model(&SendWays{}).Where("id = ? ", id).Updates(data).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|