gopostgresqlarrayenum的null处理

OGeek|极客世界-中国程序员成长平台 › 门户 › 编程› Go›Go教程

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

将postgresql数组字段的初始值定为空串时报错，应设置为'{}'

pq: 有缺陷的数组常量:""

若数组字段rows.Scan用interface{}输入，会是base64编码，但如果字段可为null，scan到string又会报错

最后将表里面的字段设为not null，然后go读出的string为image=="{... , ... , ...}"

strings.Split(image[1:len(image)-1],",")

postgresql的enum类型读出到go的interface{}时也是base64编码，所以尽量将非字符串类型设为not null，可以直接scan到string

最后用[]byte处理

//这个函数会close rows	rows中的字段全部用字符串处理，如果rows中有null会报错,keys指要将string转为string array的key
func rowsToMapArray(rows *sql.Rows, keys ...string) (code int, msg, debugMsg string, res []map[string]interface{}) {
	defer rows.Close()
	cols, err := rows.Columns()
	if err != nil {
		return j.CodeRowsScanError, j.MsgRowsScanError, err.Error(), nil
	}
	num := len(cols)
	for rows.Next() {
		str := make([][]byte, num)
		row := map[string]interface{}{}
		var args []interface{}
		for i := 0; i < num; i++ {
			//interface{}时数据库的enum 和 array类型 Scan会乱码,但能自动识别int/string类型,
			args = append(args, &str[i])
		}
		err = rows.Scan(args...)
		if err != nil {
			return j.CodeRowsScanError, j.MsgRowsScanError, err.Error(), nil
		}
		for i := 0; i < num; i++ {
			row[cols[i]] = string(str[i]) //row[cols[i]]=args[i]得到的会是base64
		}
		for _, val := range keys {
			err = stringToStringArray(row, val)
			if err != nil {
				return j.CodeStringArrayError, j.MsgStringArrayError, err.Error(), nil
			}
		}
		res = append(res, row)
	}
	return j.CodeSuccess, j.MsgSuccess, "", res
}

//如果某个字段的数据库类型是数组，用这个函数将string转化为[]string
func stringToStringArray(mp map[string]interface{}, key string) error {
	val, ok := mp[key].(string)
	if !ok {
		return errors.New("value type is not string")
	}
	l := len(val)
	if l <= 2 {
		mp[key] = make([]string, 0)
		return nil
	}
	mp[key] = strings.Split(val[1:l-1], ",")
	return nil
}