mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-11 02:15:48 +02:00
chore: rm unused functions in model.go
This commit is contained in:
0xJacky
parent
38c9727e15
commit
18f2b4aba8
1 changed files with 66 additions and 270 deletions
336
model/model.go
336
model/model.go
|
|
@ -1,307 +1,103 @@
|
|||
package model
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/0xJacky/Nginx-UI/internal/logger"
|
||||
"github.com/0xJacky/Nginx-UI/settings"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/spf13/cast"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gen"
|
||||
"gorm.io/gorm"
|
||||
gormlogger "gorm.io/gorm/logger"
|
||||
"gorm.io/gorm/schema"
|
||||
"path"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
"fmt"
|
||||
"github.com/0xJacky/Nginx-UI/internal/logger"
|
||||
"github.com/0xJacky/Nginx-UI/settings"
|
||||
"github.com/gin-gonic/gin"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gen"
|
||||
"gorm.io/gorm"
|
||||
gormlogger "gorm.io/gorm/logger"
|
||||
"path"
|
||||
"time"
|
||||
)
|
||||
|
||||
var db *gorm.DB
|
||||
|
||||
type Model struct {
|
||||
ID int `gorm:"primary_key" json:"id"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
DeletedAt *gorm.DeletedAt `gorm:"index" json:"deleted_at"`
|
||||
ID int `gorm:"primary_key" json:"id"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
DeletedAt *gorm.DeletedAt `gorm:"index" json:"deleted_at"`
|
||||
}
|
||||
|
||||
func GenerateAllModel() []any {
|
||||
return []any{
|
||||
ConfigBackup{},
|
||||
Auth{},
|
||||
AuthToken{},
|
||||
Cert{},
|
||||
ChatGPTLog{},
|
||||
Site{},
|
||||
Stream{},
|
||||
DnsCredential{},
|
||||
Environment{},
|
||||
Notification{},
|
||||
}
|
||||
return []any{
|
||||
ConfigBackup{},
|
||||
Auth{},
|
||||
AuthToken{},
|
||||
Cert{},
|
||||
ChatGPTLog{},
|
||||
Site{},
|
||||
Stream{},
|
||||
DnsCredential{},
|
||||
Environment{},
|
||||
Notification{},
|
||||
}
|
||||
}
|
||||
|
||||
func logMode() gormlogger.Interface {
|
||||
switch settings.ServerSettings.RunMode {
|
||||
case gin.ReleaseMode:
|
||||
return gormlogger.Default.LogMode(gormlogger.Warn)
|
||||
default:
|
||||
fallthrough
|
||||
case gin.DebugMode:
|
||||
return gormlogger.Default.LogMode(gormlogger.Info)
|
||||
}
|
||||
switch settings.ServerSettings.RunMode {
|
||||
case gin.ReleaseMode:
|
||||
return gormlogger.Default.LogMode(gormlogger.Warn)
|
||||
default:
|
||||
fallthrough
|
||||
case gin.DebugMode:
|
||||
return gormlogger.Default.LogMode(gormlogger.Info)
|
||||
}
|
||||
}
|
||||
|
||||
func UseDB() *gorm.DB {
|
||||
return db
|
||||
return db
|
||||
}
|
||||
|
||||
func Init() *gorm.DB {
|
||||
dbPath := path.Join(path.Dir(settings.ConfPath), fmt.Sprintf("%s.db", settings.ServerSettings.Database))
|
||||
dbPath := path.Join(path.Dir(settings.ConfPath), fmt.Sprintf("%s.db", settings.ServerSettings.Database))
|
||||
|
||||
var err error
|
||||
db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{
|
||||
Logger: logMode(),
|
||||
PrepareStmt: true,
|
||||
DisableForeignKeyConstraintWhenMigrating: true,
|
||||
})
|
||||
var err error
|
||||
db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{
|
||||
Logger: logMode(),
|
||||
PrepareStmt: true,
|
||||
DisableForeignKeyConstraintWhenMigrating: true,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
logger.Fatal(err.Error())
|
||||
}
|
||||
if err != nil {
|
||||
logger.Fatal(err.Error())
|
||||
}
|
||||
|
||||
// Migrate the schema
|
||||
err = db.AutoMigrate(GenerateAllModel()...)
|
||||
if err != nil {
|
||||
logger.Fatal(err.Error())
|
||||
}
|
||||
// Migrate the schema
|
||||
err = db.AutoMigrate(GenerateAllModel()...)
|
||||
if err != nil {
|
||||
logger.Fatal(err.Error())
|
||||
}
|
||||
|
||||
return db
|
||||
return db
|
||||
}
|
||||
|
||||
type Pagination struct {
|
||||
Total int64 `json:"total"`
|
||||
PerPage int `json:"per_page"`
|
||||
CurrentPage int `json:"current_page"`
|
||||
TotalPages int64 `json:"total_pages"`
|
||||
Total int64 `json:"total"`
|
||||
PerPage int `json:"per_page"`
|
||||
CurrentPage int `json:"current_page"`
|
||||
TotalPages int64 `json:"total_pages"`
|
||||
}
|
||||
|
||||
type DataList struct {
|
||||
Data interface{} `json:"data"`
|
||||
Pagination Pagination `json:"pagination,omitempty"`
|
||||
}
|
||||
|
||||
func SortOrder(c *gin.Context) func(db *gorm.DB) *gorm.DB {
|
||||
return func(db *gorm.DB) *gorm.DB {
|
||||
sort := c.DefaultQuery("order", "desc")
|
||||
order := fmt.Sprintf("`%s` %s", DefaultQuery(c, "sort_by", "id"), sort)
|
||||
return db.Order(order)
|
||||
}
|
||||
}
|
||||
|
||||
func OrderAndPaginate(c *gin.Context) func(db *gorm.DB) *gorm.DB {
|
||||
return func(db *gorm.DB) *gorm.DB {
|
||||
sort := c.DefaultQuery("order", "desc")
|
||||
if sort != "desc" && sort != "asc" {
|
||||
sort = "desc"
|
||||
}
|
||||
|
||||
// check if the order field is valid
|
||||
order := c.DefaultQuery("sort_by", "id")
|
||||
s, _ := schema.Parse(db.Model, &sync.Map{}, schema.NamingStrategy{})
|
||||
if _, ok := s.FieldsByName[order]; ok {
|
||||
order = fmt.Sprintf("%s %s", order, sort)
|
||||
db = db.Order(order)
|
||||
} else {
|
||||
logger.Error("invalid order field: ", order)
|
||||
}
|
||||
|
||||
page := cast.ToInt(c.Query("page"))
|
||||
if page == 0 {
|
||||
page = 1
|
||||
}
|
||||
pageSize := settings.ServerSettings.PageSize
|
||||
reqPageSize := c.Query("page_size")
|
||||
if reqPageSize != "" {
|
||||
pageSize = cast.ToInt(reqPageSize)
|
||||
}
|
||||
offset := (page - 1) * pageSize
|
||||
|
||||
return db.Offset(offset).Limit(pageSize)
|
||||
}
|
||||
}
|
||||
|
||||
func QueryToInSearch(c *gin.Context, db *gorm.DB, keys ...string) *gorm.DB {
|
||||
for _, v := range keys {
|
||||
queryArray := c.QueryArray(v + "[]")
|
||||
if len(queryArray) == 0 {
|
||||
queryArray = c.QueryArray(v)
|
||||
}
|
||||
if len(queryArray) > 0 {
|
||||
var sb strings.Builder
|
||||
|
||||
_, err := fmt.Fprintf(&sb, "`%s` IN ?", v)
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
db = db.Where(sb.String(), queryArray)
|
||||
}
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func QueryToEqualSearch(c *gin.Context, db *gorm.DB, keys ...string) *gorm.DB {
|
||||
for _, v := range keys {
|
||||
if c.Query(v) != "" {
|
||||
var sb strings.Builder
|
||||
|
||||
_, err := fmt.Fprintf(&sb, "`%s` = ?", v)
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
db = db.Where(sb.String(), c.Query(v))
|
||||
}
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func QueryToFussySearch(c *gin.Context, db *gorm.DB, keys ...string) *gorm.DB {
|
||||
for _, v := range keys {
|
||||
if c.Query(v) != "" {
|
||||
var sb strings.Builder
|
||||
|
||||
_, err := fmt.Fprintf(&sb, "`%s` LIKE ?", v)
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
var sbValue strings.Builder
|
||||
|
||||
_, err = fmt.Fprintf(&sbValue, "%%%s%%", c.Query(v))
|
||||
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
db = db.Where(sb.String(), sbValue.String())
|
||||
}
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func QueryToFussyKeysSearch(c *gin.Context, db *gorm.DB, value string, keys ...string) *gorm.DB {
|
||||
if c.Query(value) == "" {
|
||||
return db
|
||||
}
|
||||
|
||||
var condition *gorm.DB
|
||||
for i, v := range keys {
|
||||
sb := v + " LIKE ?"
|
||||
sv := "%" + c.Query(value) + "%"
|
||||
|
||||
switch i {
|
||||
case 0:
|
||||
condition = db.Where(db.Where(sb, sv))
|
||||
default:
|
||||
condition = condition.Or(sb, sv)
|
||||
}
|
||||
}
|
||||
|
||||
return db.Where(condition)
|
||||
}
|
||||
|
||||
func QueryToOrInSearch(c *gin.Context, db *gorm.DB, keys ...string) *gorm.DB {
|
||||
for _, v := range keys {
|
||||
queryArray := c.QueryArray(v + "[]")
|
||||
if len(queryArray) == 0 {
|
||||
queryArray = c.QueryArray(v)
|
||||
}
|
||||
if len(queryArray) > 0 {
|
||||
var sb strings.Builder
|
||||
|
||||
_, err := fmt.Fprintf(&sb, "`%s` IN ?", v)
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
db = db.Or(sb.String(), queryArray)
|
||||
}
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func QueryToOrEqualSearch(c *gin.Context, db *gorm.DB, keys ...string) *gorm.DB {
|
||||
for _, v := range keys {
|
||||
if c.Query(v) != "" {
|
||||
var sb strings.Builder
|
||||
|
||||
_, err := fmt.Fprintf(&sb, "`%s` = ?", v)
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
db = db.Or(sb.String(), c.Query(v))
|
||||
}
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func QueryToOrFussySearch(c *gin.Context, db *gorm.DB, keys ...string) *gorm.DB {
|
||||
for _, v := range keys {
|
||||
if c.Query(v) != "" {
|
||||
var sb strings.Builder
|
||||
|
||||
_, err := fmt.Fprintf(&sb, "`%s` LIKE ?", v)
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
var sbValue strings.Builder
|
||||
|
||||
_, err = fmt.Fprintf(&sbValue, "%%%s%%", c.Query(v))
|
||||
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
continue
|
||||
}
|
||||
|
||||
db = db.Or(sb.String(), sbValue.String())
|
||||
}
|
||||
}
|
||||
return db
|
||||
Data interface{} `json:"data"`
|
||||
Pagination Pagination `json:"pagination,omitempty"`
|
||||
}
|
||||
|
||||
func TotalPage(total int64, pageSize int) int64 {
|
||||
n := total / int64(pageSize)
|
||||
if total%int64(pageSize) > 0 {
|
||||
n++
|
||||
}
|
||||
return n
|
||||
}
|
||||
|
||||
func DefaultValue(c *gin.Context, key string, defaultValue any) any {
|
||||
if value, ok := c.Get(key); ok {
|
||||
return value
|
||||
}
|
||||
return defaultValue
|
||||
}
|
||||
|
||||
func DefaultQuery(c *gin.Context, key string, defaultValue any) string {
|
||||
return c.DefaultQuery(key, DefaultValue(c, key, defaultValue).(string))
|
||||
n := total / int64(pageSize)
|
||||
if total%int64(pageSize) > 0 {
|
||||
n++
|
||||
}
|
||||
return n
|
||||
}
|
||||
|
||||
type Method interface {
|
||||
// FirstByID Where("id=@id")
|
||||
FirstByID(id int) (*gen.T, error)
|
||||
// DeleteByID update @@table set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=@id
|
||||
DeleteByID(id int) error
|
||||
// FirstByID Where("id=@id")
|
||||
FirstByID(id int) (*gen.T, error)
|
||||
// DeleteByID update @@table set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=@id
|
||||
DeleteByID(id int) error
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue