im 管理后台

2026-08-11 23:13:51 +08:00
parent 0397fe5f22
commit 331a55751d
1407 changed files with 162069 additions and 2 deletions
@@ -0,0 +1,232 @@
+// Package hgorm
+// @Link  https://github.com/bufanyun/hotgo
+// @Copyright  Copyright (c) 2023 HotGo CLI
+// @Author  Ms <133814250@qq.com>
+// @License  https://github.com/bufanyun/hotgo/blob/master/LICENSE
+package hgorm
+
+// dao.
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"hotgo/utility/convert"
+	"hotgo/utility/db"
+
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/text/gstr"
+)
+
+type daoInstance interface {
+	Table() string
+	Ctx(ctx context.Context) *gdb.Model
+}
+
+// Join 关联表属性
+type Join struct {
+	Dao    daoInstance                // 关联表dao实例
+	Alias  string                     // 别名
+	fields map[string]*gdb.TableField // 表字段列表
+}
+
+func LeftJoin(m *gdb.Model, masterTable, masterField, joinTable, alias, onField string) *gdb.Model {
+	return m.LeftJoin(GenJoinOnRelation(masterTable, masterField, joinTable, alias, onField)...)
+}
+
+// GenJoinOnRelation 生成关联表关联条件
+func GenJoinOnRelation(masterTable, masterField, joinTable, alias, onField string) []string {
+	q := db.GetQuoteChar()
+	relation := fmt.Sprintf("%s%s%s.%s%s%s = %s%s%s.%s%s%s", q, alias, q, q, onField, q, q, masterTable, q, q, masterField, q)
+	return []string{joinTable, alias, relation}
+}
+
+func JoinFields(ctx context.Context, entity interface{}, dao daoInstance, as string) (fs string) {
+	entityFs, err := convert.GetEntityFieldTags(entity)
+	if err != nil {
+		return
+	}
+
+	if len(entityFs) == 0 {
+		return ""
+	}
+
+	fields, err := dao.Ctx(ctx).TableFields(dao.Table())
+	if err != nil {
+		return
+	}
+
+	var columns []string
+	for _, v := range entityFs {
+		if !gstr.HasPrefix(v, as) {
+			continue
+		}
+
+		field := gstr.CaseSnakeFirstUpper(gstr.StrEx(v, as))
+		if _, ok := fields[field]; ok {
+			columns = append(columns, db.QuoteFieldAs(dao.Table(), field, v))
+		}
+	}
+
+	if len(columns) > 0 {
+		return gstr.Implode(",", convert.UniqueSlice(columns))
+	}
+	return
+}
+
+// GenJoinSelect 生成关联表select
+// 这里会将实体中的字段驼峰转为下划线于数据库进行匹配，意味着数据库字段必须全部是小写字母+下划线的格式
+func GenJoinSelect(ctx context.Context, entity interface{}, dao daoInstance, joins []*Join) (allFields string, err error) {
+	var tmpFields []string
+	if len(joins) == 0 {
+		err = gerror.New("JoinFields joins len = 0")
+		return
+	}
+
+	for _, v := range joins {
+		v.fields, err = v.Dao.Ctx(ctx).TableFields(v.Dao.Table())
+		if err != nil {
+			return
+		}
+	}
+
+	masterFields, err := dao.Ctx(ctx).TableFields(dao.Table())
+	if err != nil {
+		return
+	}
+
+	entityFields, err := convert.GetEntityFieldTags(entity)
+	if err != nil {
+		return
+	}
+
+	if len(entityFields) == 0 {
+		return "*", nil
+	}
+
+	// 是否为关联表字段
+	getJoinAttribute := func(field string) (*Join, string) {
+		for _, v := range joins {
+			if gstr.HasPrefix(field, v.Alias) {
+				return v, gstr.CaseSnakeFirstUpper(gstr.StrEx(field, v.Alias))
+			}
+		}
+		return nil, ""
+	}
+
+	for _, field := range entityFields {
+		// 关联表
+		jd, joinField := getJoinAttribute(field)
+		if jd != nil {
+			if _, ok := jd.fields[joinField]; ok {
+				tmpFields = append(tmpFields, db.QuoteFieldAs(jd.Alias, joinField, field))
+				continue
+			}
+		}
+
+		// 主表
+		originalField := gstr.CaseSnakeFirstUpper(field)
+		if _, ok := masterFields[originalField]; ok {
+			tmpFields = append(tmpFields, db.QuoteField(dao.Table(), originalField))
+			continue
+		}
+	}
+	return gstr.Implode(",", convert.UniqueSlice(tmpFields)), nil
+}
+
+// GetPkField 获取dao实例中的主键名称
+func GetPkField(ctx context.Context, dao daoInstance) (string, error) {
+	fields, err := dao.Ctx(ctx).TableFields(dao.Table())
+	if err != nil {
+		return "", err
+	}
+	if len(fields) == 0 {
+		return "", gerror.New("field not found")
+	}
+
+	for _, field := range fields {
+		if strings.ToUpper(field.Key) == "PRI" {
+			return field.Name, nil
+		}
+	}
+	return "", gerror.New("no primary key")
+}
+
+// GetFieldsToSlice 获取dao实例中的所有字段
+func GetFieldsToSlice(ctx context.Context, dao daoInstance) ([]string, error) {
+	fields, err := dao.Ctx(ctx).TableFields(dao.Table())
+	if err != nil {
+		return nil, err
+	}
+	if len(fields) == 0 {
+		return nil, gerror.New("field not found")
+	}
+
+	var keys []string
+	for _, field := range fields {
+		keys = append(keys, field.Name)
+	}
+	return keys, nil
+}
+
+// IsUnique 是否唯一
+func IsUnique(ctx context.Context, dao daoInstance, where g.Map, message string, pkId ...interface{}) error {
+	if len(where) == 0 {
+		return gerror.New("where condition cannot be empty")
+	}
+
+	m := dao.Ctx(ctx).Where(where)
+	if len(pkId) > 0 {
+		field, err := GetPkField(ctx, dao)
+		if err != nil {
+			return err
+		}
+		m = m.WhereNot(field, pkId[0])
+	}
+
+	exist, err := m.Exist()
+	if err != nil {
+		return err
+	}
+
+	if exist {
+		if message == "" {
+			for k := range where {
+				message = fmt.Sprintf("in the table：%s, %v not uniqued", dao.Table(), where[k])
+				break
+			}
+		}
+		return gerror.New(message)
+	}
+	return nil
+}
+
+// FilterKeywordsWithOr 多条件关键词OR查询
+func FilterKeywordsWithOr(m *gdb.Model, filterColumns map[string]string, keyword string) *gdb.Model {
+	if filterColumns == nil || len(filterColumns) == 0 {
+		return m
+	}
+
+	conditions := make([]string, 0)
+	args := make([]interface{}, 0)
+
+	for col, operator := range filterColumns {
+		val := keyword
+		var condition string
+		switch operator {
+		case "LIKE":
+			condition = fmt.Sprintf("%s LIKE ?", col)
+			val = "%" + keyword + "%"
+		default:
+			condition = fmt.Sprintf("%s = ?", col)
+		}
+
+		conditions = append(conditions, condition)
+		args = append(args, val)
+	}
+
+	filter := fmt.Sprintf("(%s)", strings.Join(conditions, " OR "))
+	return m.Where(filter, args...)
+}