go练习 day03

最新推荐文章于 2024-06-20 13:44:57 发布

aminwangaa

最新推荐文章于 2024-06-20 13:44:57 发布

阅读量208

点赞数

分类专栏： go 文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/aminwangaa/article/details/132354991

版权

go 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

表格的多条件查询 mongodb数据库

摸瞎自学～

const (
	STRING = "string"
	ARRAY_TYPE = "[]interface {}"
	MAP_TYPE = "map[string]interface {}"
	BOOL = "bool"
	INT = "int"
	INT32 = "int32"
	INT64 = "int64"
	FLOAT = "float"
	FLOAT32 = "float32"
	FLOAT64 = "float64"
)

// 参数处理 根据请求体参数处理成查询所需要的bson结构
func GetDealParams(params map[string]interface{}) primitive.M {
	filter := bson.M{}
	for key, val := range params {
		// 获取字段的类型
		valType := GetType(val)
		// 字段类型转字符串
		sType := fmt.Sprintf("%v", valType)
		fmt.Println(sType, "sType")
		if sType == STRING  && val != "" {
			filter[key] = bson.M{"$regex": val}
		} 
		if sType == BOOL || sType == INT || sType == INT32 || sType == INT64 || sType == FLOAT || sType == FLOAT32 || sType == FLOAT64 {
			filter[key] = val
		} 
		// 对象数组类型处理
		if sType == ARRAY_TYPE {
			f := GetDealArrayParams(val)
			filter[key] = f
		}
		// 对象类型处理
		if sType == MAP_TYPE {
			f := GetDealParams(val.(map[string]interface{}))
			filter[key] = f
		}
	}

	return filter
}


func GetDealArrayParams (params interface{}) primitive.M{
	t := params.([]interface{})
	f := bson.M{}
	for _, iv := range t {
		ff := bson.M{}
		if fmt.Sprintf("%v", GetType(iv)) == STRING {
			// fmt.Println(iv, "iviviviv")
			ff["$eq"] = iv
		} else {
			// 遍历外层数组
			for kk, vv := range iv.(map[string]interface{}) {
				valCType := GetType(vv)
				cType := fmt.Sprintf("%v", valCType)
				if cType == STRING  && vv != "" {
					ff[kk] = bson.M{"$regex": vv}
				} 
				if cType == BOOL || cType == INT || cType == INT32 || cType == INT64 || cType == FLOAT || cType == FLOAT32 || cType == FLOAT64 {
					ff[kk] = vv
				} 
				if cType == ARRAY_TYPE {
					ff[kk] = GetDealArrayParams(vv)
				}
				if cType == MAP_TYPE {
					ff[kk] = GetDealParams(vv.(map[string]interface{}))
				}
			}
		}
		
		f["$elemMatch"] = ff
	}
	return f
}


func (m *RecordDao) GetRecordList(iRecordDTO map[string]interface{}) ([]model.Record, int64, error){
	var giRecordList []model.Record
	
	collection := m.MDB.Database("lowcode").Collection("lowcode")
	// 获取bson格式的请求数据
	filter := utils.GetDealParams(iRecordDTO)
	fmt.Println(filter, "filter")
	// 根据请求数据 获取查询内容
	cursor, err := collection.Find(context.Background(), filter)
	cursor.All(context.Background(), &giRecordList)
	// 根据请求数据 获取数量信息
	count, err := collection.CountDocuments(context.Background(), filter)  

	return giRecordList, count, err
}