• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    公众号

Go中的list数组筛选

原作者: [db:作者] 来自: [db:来源] 收藏 邀请

在业务中,我们常常有一些查询是根据某些条件来的,比如在list数据中根据条件匹配相应的数据,常用的做法是通过for循环,然后通过成员单个单个的比较,为方便通用,通过反射做出封装

封装相关方法:

//筛选数组
func GetListByParam(list []interface{}, result interface{}, param map[string]interface{}) error {
	if list == nil {
		return errors.New("source data is nil")
	}
	if param == nil {
		return errors.New("param is nil")
	}
	var resultArr []interface{}
	for _, value := range list {
		refValue := reflect.ValueOf(value)
		isHave := true
		for key, item := range param {
			fieldValue := refValue.FieldByName(key).Interface()
			if fieldValue != item {
				isHave = false
				break
			} else {
				isHave = true
			}
		}
		if isHave {
			resultArr = append(resultArr, value)
		}
	}
	err := InterfaceArrToStructArr(resultArr, result)
	if err != nil {
		return err
	}
	return nil
}

//获取结构体有值的成员
func GetStructMember(model interface{}) (map[string]interface{}, error) {
	if model == nil {
		return nil, errors.New("model is nil")
	}
	resultmap := make(map[string]interface{})
	if model != nil {

		refValue := reflect.ValueOf(model)
		refType := reflect.TypeOf(model)
		fieldCount := refValue.NumField()
		for i := 0; i < fieldCount; i++ {
			fieldType := refType.Field(i)
			fieldValue := refValue.Field(i)
			isadd := false
			switch fieldType.Type.String() {
			case "string":
				if fieldValue.Len() > 0 {
					isadd = true
				}
			case "int", "int8", "int16", "int32", "int64":
				if fieldValue.Int() != 0 {
					isadd = true
				}
			case "time.Time":
				valTime := fieldValue.Interface().(time.Time)
				if !CheckIsDefaultTime(valTime) {
					isadd = true
				}
			}
			if isadd {
				resultmap[refType.Field(i).Name] = fieldValue.Interface()
			}
		}
	}
	return resultmap, nil
}

//InterfaceArrToStructArr Interface数组转为结构体数组
func InterfaceArrToStructArr(src []interface{}, model interface{}) error {
	val := reflect.Indirect(reflect.ValueOf(model))
	typ := val.Type()
	for _, r := range src {
		mVal := reflect.Indirect(reflect.New(typ.Elem().Elem())).Addr()
		fmt.Println(r)
		fmt.Println(mVal.Interface())
		IntfaceToStruct(r, mVal.Interface())
		val = reflect.Append(val, mVal)
	}
	DeepCopy(model, val.Interface())
	return nil
}

//IntfaceToStruct IntfaceToStruct 通过json过渡,结构体互相转化
func IntfaceToStruct(mapSrc interface{}, dst interface{}) error {
	//待优化
	bytes, err := json.Marshal(mapSrc)
	err = json.Unmarshal(bytes, &dst)
	return err
}

func DeepCopy(dst, src interface{}) error {
	var buf bytes.Buffer
	if err := gob.NewEncoder(&buf).Encode(src); err != nil {
		return err
	}
	return gob.NewDecoder(bytes.NewBuffer(buf.Bytes())).Decode(dst)
}

  

调用方式:

model = &hostmodels.HostsModel{Gid: 1, CdnName: ""}
results := make([]*hostmodels.HostsModel, 0)
list := make([]interface{}, 0)
list = append(list,new(hostmodels.HostsModel){})
mapQuery, _ := GetStructMember(*model) 
err = GetListByParam(list, &results, mapQuery) if err != nil { return nil, err } //排序 sort.SliceStable(results, func(i, j int) bool { return results[i].Gid < results[j].Gid })

  


鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
上一篇:
[Go] Template 使用简介发布时间:2022-07-10
下一篇:
Mosquitto源码分析及用go语言的重新实现(一、开篇)发布时间:2022-07-10
热门推荐
热门话题
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap