Golang MySQL读取到Map中,bit数据类型处理

2020-08-14  本文已影响0人  承诺一时的华丽
func QueryRows(db *DB, sqlStr string, val ...interface{}) (list []map[string]interface{}, err error) {
    var rows *sql.Rows
    rows, err = db.Raw(sqlStr, val...).Rows()
    if err != nil {
        return
    }
    defer rows.Close()
    var columns []string
    columns, err = rows.Columns()
    if err != nil {
        return
    }
    values := make([]interface{}, len(columns))
    scanArgs := make([]interface{}, len(values))
    for i := range values {
        scanArgs[i] = &values[i]
    }
    // 这里需要初始化为空数组,否则在查询结果为空的时候,返回的会是一个未初始化的指针
    for rows.Next() {
        err = rows.Scan(scanArgs...)
        if err != nil {
            return
        }

        ret := make(map[string]interface{})
        for i, col := range values {
            if col == nil {
                ret[columns[i]] = nil
            } else {
                switch val := (*scanArgs[i].(*interface{})).(type) {
                case byte:
                    ret[columns[i]] = val
                    break
                case []byte:
                    v := string(val)
                    switch v {
                    case "\x00": // 处理数据类型为bit的情况
                        ret[columns[i]] = 0
                    case "\x01": // 处理数据类型为bit的情况
                        ret[columns[i]] = 1
                    default:
                        ret[columns[i]] = v
                        break
                    }
                    break
                case time.Time:
                    if val.IsZero() {
                        ret[columns[i]] = nil
                    } else {
                        ret[columns[i]] = val.Format("2006-01-02 15:04:05")
                    }
                    break
                default:
                    ret[columns[i]] = val
                }
            }
        }
        list = append(list, ret)
    }
    if err = rows.Err(); err != nil {
        return
    }
    return
}
    switch v {
        case "\x00": // 处理数据类型为bit的情况
            ret[columns[i]] = 0
        case "\x01": // 处理数据类型为bit的情况
            ret[columns[i]] = 1
        default:
            ret[columns[i]] = v
            break
        }
上一篇 下一篇

猜你喜欢

热点阅读