go gorm StructField动态天生结构体查询多条表记录 开辟者神器、DB通用微服 ...

water/goweb

1. type PagedbRequest struct {
2.     page.PageRequest
4.     //表名
5.     TableName string `json:"table_name"`
6.     //字段列表，分隔
7.     FieldsName string `json:"fields_name"`
8.     //返回日期转为int64
9.     TimeToInt  bool `json:"time_to_int"`
10.     Camel2Case bool `json:"-"`
12.     SubTable  *SubTableDto `json:"sub_table,omitempty"`
13.     JoinTable *SubTableDto `json:"join_table,omitempty"`
14. }

复制代码

1. type PageRequest struct {
2.     basedto.BaseEntity `json:"-"`
3.     RequestId          string `json:"request_id"`
4.     //每页记录数
5.     PageSize int `json:"page_size"`
6.     //当前页码
7.     PageCurrent int `json:"current"`
8.     //排序字段数组
9.     OrderBys []*dto.OrderByDto `json:"order_by"`
10.     //查询字段条件
11.     Fields     []*dto.QueryField `json:"fields"`
12.     EsBoolType int               `json:"es_bool_type"`
14.     *baseconfig.DbClientDto `json:"-"`
15. }

复制代码

1. type PageResult struct {
2.     basedto.BaseEntity
4.     Code int    `json:"code"`
5.     Msg  string `json:"msg"`
7.     PageSize    int `json:"page_size,omitempty"`
8.     PageCurrent int `json:"current,omitempty"`
10.     Total         int             `json:"total,omitempty"`
11.     Data          any             `json:"data"`
12.    
13. }
14. 测试用例

复制代码

1. func Test013_GeneralScanTable(t *testing.T) {
3.     var dbRequest = Default().SetPageSize(3)
5.     dbRequest.TableName = "sys_dept"
6.     dbRequest.FieldsName = "dept_id,dept_name" //dbRequest.SetQueryAll(true)
7.     var result = dbRequest.GeneralScanTable()
8.     golog.Info(result)
10. }

复制代码

2024-10-13 10:51:45.312 [INFO] {
     "code": 200,
     "msg": "乐成",
     "page_size": 3,
     "current": 1,
     "total": 10,
     "data": [
          {
               "DeptId": 100,
               "DeptName": "若依科技"
          },
          {
               "DeptId": 101,
               "DeptName": "深圳总公司"
          },
          {
               "DeptId": 102,
               "DeptName": "长沙分公司"
          }
     ],
  
}

代码实现

1. package pagedb
3. import (
4.     "errors"
5.     "gitee.com/leijmdas/gobase/goconfig/common/base/basedto"
6.     "gitee.com/leijmdas/gobase/goconfig/common/base/baseutils"
7.     "gitee.com/leijmdas/gobase/goconfig/common/base/jsonutils"
8.     "gitee.com/leijmdas/gobase/goconfig/common/base/stringutils"
9.     "gitee.com/leijmdas/gobase/goconfig/common/gocontext"
10.     "gitee.com/leijmdas/gobase/goconfig/common/golog"
11.     "gitee.com/leijmdas/gobase/goconfig/common/ichubconfig"
12.     "gitee.com/leijmdas/goweb/common/meta/metadata"
13.     "gitee.com/leijmdas/goweb/common/meta/service"
14.     "gitee.com/leijmdas/goweb/commonpage/page"
15.     "github.com/huandu/go-clone"
16.     "github.com/jinzhu/gorm"
17.     "reflect"
18.     "time"
20.     "github.com/sirupsen/logrus"
21.     "strings"
22. )

复制代码

1. func (self *PagedbRequest) GeneralScanTable() *page.PageResult {
2.     var metadataFactroy = service.FindBeanMetadataFactroy() //(self.DbClientDto)
3.     var metadata = metadataFactroy.FindMetadata(self.TableName)
4.     if !metadata.TableExist {
5.        return page.ResultFailedPageResultErr(errors.New("table not exist"))
6.     }
7.     var dbentity = self.CreateDbTableStru(metadata)
8.     var records, err = self.ScanTable(dbentity.Addr().Interface())
10.     if err != nil {
11.        return page.ResultFailedPageResultErr(err)
12.     }
14.     var result = page.DefaultResult()
15.     result.PageCurrent = self.PageCurrent
16.     result.PageSize = self.PageSize
17.     result.Total, _ = self.CountTable(self.TableName)
18.     result.Data = records
19.     return result
21. }

复制代码

1. func (self *PagedbRequest) CreateDbTableStru(meta *metadata.MetadataTable) reflect.Value {
2.     // logrus.Info("CreateDbTableStru meta=", meta.ToPrettyString())
3.     var metaColService = service.NewMetadataFactroy()
4.     dbFields := []reflect.StructField{}
6.     for _, col := range meta.Columns {
7.        //logrus.Info("CreateDbTableStru col=", col.ToPrettyString())
8.        var colType = metaColService.FindGoType(col.DataType)
9.        var ColTyp8 = reflect.TypeOf(int8(0))
11.        var ct = func(colType string) *reflect.Type {
12.           var ColTyp = reflect.TypeOf(int8(0))
13.           switch colType {
14.           case "float64":
15.              ColTyp = reflect.TypeOf(float64(0))
16.           case "int64":
17.              ColTyp = reflect.TypeOf(int64(0))
18.           case "int32":
19.              ColTyp = reflect.TypeOf(int64(0))
20.           case "string":
21.              ColTyp = reflect.TypeOf("")
22.           case "bool":
23.              ColTyp = reflect.TypeOf(true)
24.           case "byte":
25.              ColTyp = reflect.TypeOf(byte(0))
26.           case "[]byte":
27.              ColTyp = reflect.TypeOf([]byte{})
29.           case "time.Time":
30.              ColTyp = reflect.TypeOf(time.Now())
32.           }
33.           return &ColTyp
34.        }(colType)
35.        if ColTyp8 != *ct {
36.           var colField = reflect.StructField{
37.              Name: stringutils.Case2Camel(col.ColumnName),
38.              Type: *ct,
39.           }
40.           if self.IfExistField(col.ColumnName) {
41.              //colField.Tag = reflect.StructTag(fmt.Sprintf("json:"%s"", col.ColumnName))
43.              dbFields = append(dbFields, colField)
44.           }
45.        }
46.     }
48.     dbtableType := reflect.StructOf(dbFields)
49.     dbentity := reflect.New(dbtableType).Elem()
50.     return dbentity
51. }

复制代码

1. func (self *PagedbRequest) IfExistField(field string) bool {
2.     if self.FieldsName == "" {
3.        return true
4.     }
5.     if strings.Contains(field, ",") {
6.        return false
7.     }
8.     var fields = strings.Split(self.FieldsName, ",")
9.     for _, v := range fields {
10.        if v == field {
11.           return true
12.        }
13.     }
14.     return false
15. }

复制代码

1. func (self *PagedbRequest) ScanTable(dbentity any) (any, error) {
3.     var records = []any{} //reflect.New(reflect.SliceOf(dbtableType)).Elem()
5.     var dbc = self.FindScanTable()
6.     var rows, err = dbc.Rows()
7.     if err != nil {
8.        golog.Error(err)
9.        return records, err
10.     }
12.     defer func() {
13.        rows.Close()
14.     }()
15.     for rows.Next() {
16.        dbentity = clone.Clone(dbentity) //    dbentity := reflect.New(dbtableType).Elem()
17.        if err := dbc.ScanRows(rows, dbentity); err != nil {
18.           golog.Error("should get no error, but got ", err)
19.           return records, err
20.        }
21.        records = append(records, dbentity)
22.     }
24.     return baseutils.ParseArray2Map(records)
26. }

复制代码

1. func (self *PagedbRequest) FindScanTable() *gorm.DB {
3.     dbc := self.GetDB().Table(self.TableName)
5.     dbc = self.BuildWhere(dbc)
6.     dbc = self.Order(dbc)
7.     dbc = self.SetLimit(dbc)
8.     if self.FieldsName != "" {
9.        dbc = dbc.Select(strings.Split(self.FieldsName, ","))
10.     }
11.     return dbc
13. }

复制代码

1. func ParseArray2Map(stru []any) (any, error) {
2.     var bytes, err = gjson.Encode(stru)
3.     if err != nil {
4.        return nil, err
5.     }
6.     logrus.Info(1)
7.     return gjson.Decode(bytes)
9. }

复制代码

1. func (self *PageRequest) CountTable(table string) (int, error) {
2.     dbc := self.GetDB().Table(table)
3.     dbc = self.BuildWhere(dbc).Offset(0).Limit(1)
4.     var count int
5.     if err := dbc.Count(&count).Error; err != nil {
6.        logrus.Error(err)
7.        return 0, err
8.     }
10.     logrus.Info("\ncount=", count)
11.     return count, nil
13. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。