// BindVariablesToProto3 converts internal type to proto3 BindVariable array
func BindVariablesToProto3(bindVars map[string]interface{}) (map[string]*pb.BindVariable, error) {
if len(bindVars) == 0 {
return nil, nil
}
result := make(map[string]*pb.BindVariable)
for k, v := range bindVars {
bv := new(pb.BindVariable)
switch v := v.(type) {
case []interface{}:
// This is how the list variables will normally appear.
if len(v) == 0 {
continue
}
// This assumes homogenous types, but that is what we support.
val := v[0]
switch valt := val.(type) {
// string and []byte are TYPE_BYTES_LIST
case string:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = []byte(lv.(string))
}
bv.ValueBytesList = listArg
case []byte:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = lv.([]byte)
}
bv.ValueBytesList = listArg
// int, int16, int32, int64 are TYPE_INT_LIST
case int:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv.(int))
}
bv.ValueIntList = listArg
case int16:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv.(int16))
}
bv.ValueIntList = listArg
case int32:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv.(int32))
}
bv.ValueIntList = listArg
case int64:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = lv.(int64)
}
bv.ValueIntList = listArg
// uint, uint16, uint32, uint64 are TYPE_UINT_LIST
case uint:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv.(uint))
}
bv.ValueUintList = listArg
case uint16:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv.(uint16))
}
bv.ValueUintList = listArg
case uint32:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv.(uint32))
}
bv.ValueUintList = listArg
case uint64:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = lv.(uint64)
}
bv.ValueUintList = listArg
// float32, float64 are TYPE_FLOAT_LIST
case float32:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = float64(lv.(float32))
}
bv.ValueFloatList = listArg
case float64:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = lv.(float64)
}
bv.ValueFloatList = listArg
case nil:
// do nothing
default:
return nil, fmt.Errorf("unexpected type %T for variable %q", valt, k)
}
case string:
bv.Type = pb.BindVariable_TYPE_BYTES
bv.ValueBytes = []byte(v)
case []string:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = []byte(lv)
}
bv.ValueBytesList = listArg
case []byte:
bv.Type = pb.BindVariable_TYPE_BYTES
bv.ValueBytes = v
case [][]byte:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueBytesList = listArg
case int:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = int64(v)
case int16:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = int64(v)
case int32:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = int64(v)
case int64:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = v
case []int:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv)
}
bv.ValueIntList = listArg
case []int16:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv)
}
bv.ValueIntList = listArg
case []int32:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv)
}
bv.ValueIntList = listArg
case []int64:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueIntList = listArg
case uint:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = uint64(v)
case uint16:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = uint64(v)
case uint32:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = uint64(v)
case uint64:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = v
case []uint:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv)
}
bv.ValueUintList = listArg
case []uint16:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv)
}
bv.ValueUintList = listArg
case []uint32:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv)
}
bv.ValueUintList = listArg
case []uint64:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueUintList = listArg
case float32:
bv.Type = pb.BindVariable_TYPE_FLOAT
bv.ValueFloat = float64(v)
case float64:
bv.Type = pb.BindVariable_TYPE_FLOAT
bv.ValueFloat = float64(v)
case []float32:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = float64(lv)
}
bv.ValueFloatList = listArg
case []float64:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueFloatList = listArg
case nil:
// do nothing
default:
return nil, fmt.Errorf("unexpected type %T for variable %q", v, k)
}
result[k] = bv
}
return result, nil
}
// BindVariablesToProto3 converts internal type to proto3 BindVariable array
func BindVariablesToProto3(bindVars map[string]interface{}) (map[string]*pb.BindVariable, error) {
if len(bindVars) == 0 {
return nil, nil
}
result := make(map[string]*pb.BindVariable)
for k, v := range bindVars {
bv := new(pb.BindVariable)
switch v := v.(type) {
case []interface{}:
// This is how the list variables will normally appear.
if len(v) == 0 {
return nil, fmt.Errorf("empty list not allowed: %s", k)
}
bv.Type = sqltypes.Tuple
bv.Values = make([]*pb.Value, len(v))
values := make([]pb.Value, len(v))
for i, lv := range v {
val, err := BindVariableToValue(lv)
if err != nil {
return nil, fmt.Errorf("key: %s: %v", k, err)
}
if val.Type != sqltypes.Null {
values[i] = val
bv.Values[i] = &values[i]
}
}
case []string:
if len(v) == 0 {
return nil, fmt.Errorf("empty list not allowed: %s", k)
}
bv.Type = sqltypes.Tuple
bv.Values = make([]*pb.Value, len(v))
values := make([]pb.Value, len(v))
for i, lv := range v {
values[i].Type = sqltypes.VarChar
values[i].Value = []byte(lv)
bv.Values[i] = &values[i]
}
case [][]byte:
if len(v) == 0 {
return nil, fmt.Errorf("empty list not allowed: %s", k)
}
bv.Type = sqltypes.Tuple
bv.Values = make([]*pb.Value, len(v))
values := make([]pb.Value, len(v))
for i, lv := range v {
values[i].Type = sqltypes.VarBinary
values[i].Value = lv
bv.Values[i] = &values[i]
}
case []int:
if len(v) == 0 {
return nil, fmt.Errorf("empty list not allowed: %s", k)
}
bv.Type = sqltypes.Tuple
bv.Values = make([]*pb.Value, len(v))
values := make([]pb.Value, len(v))
for i, lv := range v {
values[i].Type = sqltypes.Int64
values[i].Value = strconv.AppendInt(nil, int64(lv), 10)
bv.Values[i] = &values[i]
}
case []int64:
if len(v) == 0 {
return nil, fmt.Errorf("empty list not allowed: %s", k)
}
bv.Type = sqltypes.Tuple
bv.Values = make([]*pb.Value, len(v))
values := make([]pb.Value, len(v))
for i, lv := range v {
values[i].Type = sqltypes.Int64
values[i].Value = strconv.AppendInt(nil, lv, 10)
bv.Values[i] = &values[i]
}
case []uint64:
if len(v) == 0 {
return nil, fmt.Errorf("empty list not allowed: %s", k)
}
bv.Type = sqltypes.Tuple
bv.Values = make([]*pb.Value, len(v))
values := make([]pb.Value, len(v))
for i, lv := range v {
values[i].Type = sqltypes.Uint64
values[i].Value = strconv.AppendUint(nil, lv, 10)
bv.Values[i] = &values[i]
}
default:
val, err := BindVariableToValue(v)
if err != nil {
return nil, fmt.Errorf("key: %s: %v", k, err)
}
bv.Type = val.Type
bv.Value = val.Value
}
result[k] = bv
}
return result, nil
}
// BoundQueryToProto3 converts internal types to proto3 BoundQuery
func BoundQueryToProto3(sql string, bindVars map[string]interface{}) *pb.BoundQuery {
result := &pb.BoundQuery{
Sql: []byte(sql),
}
if len(bindVars) > 0 {
result.BindVariables = make(map[string]*pb.BindVariable)
for k, v := range bindVars {
bv := new(pb.BindVariable)
switch v := v.(type) {
case []interface{}:
// This is how the list variables will normally appear.
if len(v) == 0 {
continue
}
// This assumes homogenous types, but that is what we support.
val := v[0]
switch val.(type) {
// string and []byte are TYPE_BYTES_LIST
case string:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = []byte(lv.(string))
}
bv.ValueBytesList = listArg
case []byte:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = lv.([]byte)
}
bv.ValueBytesList = listArg
// int, int16, int32, int64 are TYPE_INT_LIST
case int:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv.(int))
}
bv.ValueIntList = listArg
case int16:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv.(int16))
}
bv.ValueIntList = listArg
case int32:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv.(int32))
}
bv.ValueIntList = listArg
case int64:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = lv.(int64)
}
bv.ValueIntList = listArg
// uint, uint16, uint32, uint64 are TYPE_UINT_LIST
case uint:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv.(uint))
}
bv.ValueUintList = listArg
case uint16:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv.(uint16))
}
bv.ValueUintList = listArg
case uint32:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv.(uint32))
}
bv.ValueUintList = listArg
case uint64:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = lv.(uint64)
}
bv.ValueUintList = listArg
// float32, float64 are TYPE_FLOAT_LIST
case float32:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = float64(lv.(float32))
}
bv.ValueFloatList = listArg
case float64:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = lv.(float64)
}
bv.ValueFloatList = listArg
}
case string:
bv.Type = pb.BindVariable_TYPE_BYTES
bv.ValueBytes = []byte(v)
case []string:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = []byte(lv)
}
bv.ValueBytesList = listArg
case []byte:
bv.Type = pb.BindVariable_TYPE_BYTES
bv.ValueBytes = v
case [][]byte:
bv.Type = pb.BindVariable_TYPE_BYTES_LIST
listArg := make([][]byte, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueBytesList = listArg
case int:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = int64(v)
case int16:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = int64(v)
case int32:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = int64(v)
case int64:
bv.Type = pb.BindVariable_TYPE_INT
bv.ValueInt = v
case []int:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv)
}
bv.ValueIntList = listArg
case []int16:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv)
}
bv.ValueIntList = listArg
case []int32:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = int64(lv)
}
bv.ValueIntList = listArg
case []int64:
bv.Type = pb.BindVariable_TYPE_INT_LIST
listArg := make([]int64, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueIntList = listArg
case uint:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = uint64(v)
case uint16:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = uint64(v)
case uint32:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = uint64(v)
case uint64:
bv.Type = pb.BindVariable_TYPE_UINT
bv.ValueUint = v
case []uint:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv)
}
bv.ValueUintList = listArg
case []uint16:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv)
}
bv.ValueUintList = listArg
case []uint32:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = uint64(lv)
}
bv.ValueUintList = listArg
case []uint64:
bv.Type = pb.BindVariable_TYPE_UINT_LIST
listArg := make([]uint64, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueUintList = listArg
case float32:
bv.Type = pb.BindVariable_TYPE_FLOAT
bv.ValueFloat = float64(v)
case float64:
bv.Type = pb.BindVariable_TYPE_FLOAT
bv.ValueFloat = float64(v)
case []float32:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = float64(lv)
}
bv.ValueFloatList = listArg
case []float64:
bv.Type = pb.BindVariable_TYPE_FLOAT_LIST
listArg := make([]float64, len(v))
for i, lv := range v {
listArg[i] = lv
}
bv.ValueFloatList = listArg
}
result.BindVariables[k] = bv
}
}
return result
}