初始

api/v1/common/coryCommon/basics.go

@@ -0,0 +1,107 @@
+package coryCommon
+
+import (
+	"context"
+	"errors"
+	"github.com/tiger1103/gfast/v3/internal/app/system/dao"
+	"reflect"
+	"strconv"
+)
+
+func New() *coryCom {
+	return &coryCom{}
+}
+
+type coryCom struct{}
+
+// CreateByOrUpdateBy 专门用来反射结构体中的创建人和更新人，然后返回回去，避免重复造轮子去写重复代码
+/**
+*使用实例代码
+	* by := coryCommon.New().CreateByOrUpdateBy(ctx, res)
+	* infoRes := by.(model.BusCompanyInfoRes)
+	* res = &infoRes
+	*
+	*其中 	updateByFieldVal.Set(reflect.ValueOf(updateByValue.Interface().(*gvar.Var).String()))   必须类型一致
+*/
+func (s *coryCom) CreateByOrUpdateBy(ctx context.Context, data interface{}) (dataRes interface{}) {
+	// 使用反射获取 data 的值和类型信息
+	val := reflect.ValueOf(data)
+	typ := reflect.TypeOf(data)
+
+	// 判断 data 是否为指针类型，并获取实际值
+	if val.Kind() == reflect.Ptr {
+		val = val.Elem()
+		typ = typ.Elem()
+	}
+
+	flagCreate := true
+	flagUpdate := true
+	// 获取 createBy 字段在结构体中的索引
+	createByField, ok := typ.FieldByName("CreateBy")
+	if !ok {
+		flagCreate = false
+		//return data  		// 如果结构体中不存在 createBy 字段，则直接返回原始值
+		createByField2, ok2 := typ.FieldByName("CreatedBy")
+		if ok2 {
+			createByField = createByField2
+			flagCreate = true
+		}
+	}
+	updateByField, ok := typ.FieldByName("UpdateBy")
+	if !ok {
+		flagUpdate = false
+		//return data 		// 如果结构体中不存在 createBy 字段，则直接返回原始值
+		updateByField2, ok2 := typ.FieldByName("UpdatedBy")
+		if ok2 {
+			updateByField = updateByField2
+			flagCreate = true
+		}
+	}
+	if flagCreate {
+		// 判断 createBy 字段的类型是否为 string
+		if createByField.Type.Kind() != reflect.String {
+			// 如果 createBy 字段的类型不是 string，请根据需要进行相应的处理或返回错误
+			// 此处假设 createBy 必须为 string 类型，如果不是则返回错误
+			return errors.New("createBy字段类型不匹配")
+		}
+		// 获取原始的 createBy 字段的值并获取创建人
+		createId := val.FieldByIndex(createByField.Index).String()
+		ve := SelectByString(ctx, IsNumeric(createId), createId)
+		// 设置 createBy 字段的值为指定参数
+		createByFieldVal := val.FieldByIndex(createByField.Index)
+		createByFieldVal.SetString(ve)
+	}
+	if flagUpdate {
+		// 判断 updateBy 字段的类型是否为 string
+		if updateByField.Type.Kind() != reflect.String {
+			// 如果 createBy 字段的类型不是 string，请根据需要进行相应的处理或返回错误
+			// 此处假设 createBy 必须为 string 类型，如果不是则返回错误
+			return errors.New("updateBy字段类型不匹配")
+		}
+		// 获取原始的 createBy 字段的值并获取创建人
+		updateId := val.FieldByIndex(updateByField.Index).String()
+		ve := SelectByString(ctx, IsNumeric(updateId), updateId)
+		// 设置 createBy 字段的值为指定参数
+		updateByFieldVal := val.FieldByIndex(updateByField.Index)
+		updateByFieldVal.SetString(ve)
+	}
+	// 返回修改后的结构体
+	return val.Interface()
+}
+
+func IsNumeric(str string) bool {
+	_, err := strconv.ParseFloat(str, 64)
+	return err == nil
+}
+
+// SelectByString 根据字符串查询到具体人对应的名称
+func SelectByString(ctx context.Context, flag bool, str string) (ve string) {
+	if flag {
+		value, _ := dao.SysUser.Ctx(ctx).Fields("user_nickname").Where("id", str).Value()
+		ve = value.String()
+	} else {
+		value, _ := dao.BusConstructionUser.Ctx(ctx).Fields("user_name").Where("openid", str).Value()
+		ve = value.String()
+	}
+	return
+}