Hass/Message-Push-Nest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add gormv2 and support sqlite

Browse Source
This commit is contained in:
engigu
2024-04-29 17:31:13 +08:00
parent 44054ae157
commit 33e44ce596
22 changed files with 236 additions and 279 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/send_ways.go
+7 -7
View File
@@ -3,16 +3,16 @@ package models
import (
"errors"
"fmt"
"github.com/jinzhu/gorm"
"gorm.io/gorm"
"message-nest/pkg/util"
)
type SendWays struct {
UUIDModel
Name string `json:"name" gorm:"type:varchar(100) comment '渠道名称';default:'';"`
Type string `json:"type" gorm:"type:varchar(100) comment '渠道类型';default:'';index:type"`
Auth string `json:"auth" gorm:"type:varchar(2048) comment '认证信息';default:'';"`
Name string `json:"name" gorm:"type:varchar(100) ;default:'';"`
Type string `json:"type" gorm:"type:varchar(100) ;default:'';index"`
Auth string `json:"auth" gorm:"type:varchar(2048) ;default:'';"`
}
func GenerateWayUniqueID() string {
@@ -63,10 +63,10 @@ func GetSendWays(pageNum int, pageSize int, name string, type_ string, maps inte
return ways, nil
}
func GetSendWaysTotal(name string, type_ string, maps interface{}) (int, error) {
func GetSendWaysTotal(name string, type_ string, maps interface{}) (int64, error) {
var (
err error
total int
total int64
)
query := db.Model(&SendWays{}).Where(maps)
if name != "" {
@@ -85,7 +85,7 @@ func GetSendWaysTotal(name string, type_ string, maps interface{}) (int, error)
func GetWayByID(id string) (SendWays, error) {
var way SendWays
err := db.Where("id = ? ", id).Find(&way).Error
err := db.Where("id = ? ", id).Take(&way).Error
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
return way, err
}