doushu7588 2017-09-30 16:14
浏览 101
已采纳

Golang抽象函数从db获取数据并填充数组

I want to create an abstract function, that gets data from DB and fills array by this data. Types of array can be different. And I want to do it without reflect, due to performance issues. I just want to call everywhere some function like GetDBItems() and get array of data from DB with desired type. But all implementations that I create are owful.

Here is this function implementation:

type AbstractArrayGetter func(size int) []interface{}

func GetItems(arrayGetter AbstractArrayGetter) {
    res := DBResponse{}
    DB.Get(&res)
    arr := arrayGetter(len(res.Rows))
    for i := 0; i < len(res.Rows); i++ {
        json.Unmarshal(res.Rows[i].Value, &obj[i])
    }
}

Here I call this function:

var events []Event
GetFullItems("events", "events_list", map[string]interface{}{}, func(size int) []interface{} {
        events = make([]Event, size, size)
        proxyEnt := make([]interface{}, size, size)
        for i, _ := range events {
            proxyEnt[i] = &events[i]
        }
        return proxyEnt
    })

It works, but there are to much code to call this function, also there is some perfomance issue about copying events array to interfaces array.

How can I do it without reflect and do it with a short function call code? Or reflect not to slow in this case?

  • 写回答

1条回答 默认 最新

  • drkjzk3359 2017-10-05 17:13
    关注

    I tested performance with reflect, and it is similar to the mentioned above solution. So here is solution with reflect, if someone needs it. This function gets data from DB and fills abstract array

    func GetItems(design string, viewName string, opts map[string]interface{}, arrayType interface{}) (interface{}, error) {
        res := couchResponse{}
        opts["limit"] = 100000
        bytes, err := CouchView(design, viewName, opts)
        if err != nil {
            return nil, err
        }
        err = json.Unmarshal(bytes, &res)
        if err != nil {
            return nil, err
        }
        dataType := reflect.TypeOf(arrayType)
        slice := reflect.MakeSlice(reflect.SliceOf(dataType), len(res.Rows), len(res.Rows))
        for i := 0; i < len(res.Rows); i++ {
            if opts["include_docs"] == true {
                err = json.Unmarshal(res.Rows[i].Doc, slice.Index(i).Addr().Interface())
            } else {
                err = json.Unmarshal(res.Rows[i].Value, slice.Index(i).Addr().Interface())
            }
            if err != nil {
                return nil, err
            }
        }
        x := reflect.New(slice.Type())
        x.Elem().Set(slice)
        return x.Interface(), nil
    }
    

    and getting data using this function:

    var e Event
    res, err := GetItems("data", "data_list", map[string]interface{}{}, e)
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 安卓adb backup备份应用数据失败
  • ¥15 eclipse运行项目时遇到的问题
  • ¥15 关于#c##的问题:最近需要用CAT工具Trados进行一些开发
  • ¥15 南大pa1 小游戏没有界面,并且报了如下错误,尝试过换显卡驱动,但是好像不行
  • ¥15 没有证书,nginx怎么反向代理到只能接受https的公网网站
  • ¥50 成都蓉城足球俱乐部小程序抢票
  • ¥15 yolov7训练自己的数据集
  • ¥15 esp8266与51单片机连接问题(标签-单片机|关键词-串口)(相关搜索:51单片机|单片机|测试代码)
  • ¥15 电力市场出清matlab yalmip kkt 双层优化问题
  • ¥30 ros小车路径规划实现不了,如何解决?(操作系统-ubuntu)