Add 'querytools/' from commit '2838cd8abdeb99507da080df5dc8abe388b28308'
Repo: github.com/lbryio/query.go git-subtree-dir: querytools git-subtree-mainline:094289d4cfgit-subtree-split:2838cd8abd
This commit is contained in:
Alex Grintsvayg
commit
f233e3b525
1 changed files with 121 additions and 0 deletions
121
querytools/query.go
Normal file
121
querytools/query.go
Normal file
|
|
@ -0,0 +1,121 @@
|
|||
package query
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/lbryio/errors.go"
|
||||
"github.com/lbryio/null.go"
|
||||
)
|
||||
|
||||
func InterpolateParams(query string, args ...interface{}) (string, error) {
|
||||
for i := 0; i < len(args); i++ {
|
||||
field := reflect.ValueOf(args[i])
|
||||
|
||||
if value, ok := field.Interface().(time.Time); ok {
|
||||
query = strings.Replace(query, "?", `"`+value.Format("2006-01-02 15:04:05")+`"`, 1)
|
||||
} else if nullable, ok := field.Interface().(null.Nullable); ok {
|
||||
if nullable.IsNull() {
|
||||
query = strings.Replace(query, "?", "NULL", 1)
|
||||
} else {
|
||||
switch field.Type() {
|
||||
case reflect.TypeOf(null.Time{}):
|
||||
query = strings.Replace(query, "?", `"`+field.Interface().(null.Time).Time.Format("2006-01-02 15:04:05")+`"`, 1)
|
||||
case reflect.TypeOf(null.Int{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatInt(int64(field.Interface().(null.Int).Int), 10), 1)
|
||||
case reflect.TypeOf(null.Int8{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatInt(int64(field.Interface().(null.Int8).Int8), 10), 1)
|
||||
case reflect.TypeOf(null.Int16{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatInt(int64(field.Interface().(null.Int16).Int16), 10), 1)
|
||||
case reflect.TypeOf(null.Int32{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatInt(int64(field.Interface().(null.Int32).Int32), 10), 1)
|
||||
case reflect.TypeOf(null.Int64{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatInt(field.Interface().(null.Int64).Int64, 10), 1)
|
||||
case reflect.TypeOf(null.Uint{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatUint(uint64(field.Interface().(null.Uint).Uint), 10), 1)
|
||||
case reflect.TypeOf(null.Uint8{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatUint(uint64(field.Interface().(null.Uint8).Uint8), 10), 1)
|
||||
case reflect.TypeOf(null.Uint16{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatUint(uint64(field.Interface().(null.Uint16).Uint16), 10), 1)
|
||||
case reflect.TypeOf(null.Uint32{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatUint(uint64(field.Interface().(null.Uint32).Uint32), 10), 1)
|
||||
case reflect.TypeOf(null.Uint64{}):
|
||||
query = strings.Replace(query, "?", strconv.FormatUint(field.Interface().(null.Uint64).Uint64, 10), 1)
|
||||
case reflect.TypeOf(null.String{}):
|
||||
query = strings.Replace(query, "?", `"`+field.Interface().(null.String).String+`"`, 1)
|
||||
case reflect.TypeOf(null.Bool{}):
|
||||
if field.Interface().(null.Bool).Bool {
|
||||
query = strings.Replace(query, "?", "1", 1)
|
||||
} else {
|
||||
query = strings.Replace(query, "?", "0", 1)
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
switch field.Kind() {
|
||||
case reflect.Bool:
|
||||
boolString := "0"
|
||||
if field.Bool() {
|
||||
boolString = "1"
|
||||
}
|
||||
query = strings.Replace(query, "?", boolString, 1)
|
||||
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
||||
query = strings.Replace(query, "?", strconv.FormatInt(field.Int(), 10), 1)
|
||||
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
||||
query = strings.Replace(query, "?", strconv.FormatUint(field.Uint(), 10), 1)
|
||||
case reflect.Float32, reflect.Float64:
|
||||
query = strings.Replace(query, "?", strconv.FormatFloat(field.Float(), 'f', -1, 64), 1)
|
||||
case reflect.String:
|
||||
query = strings.Replace(query, "?", `"`+field.String()+`"`, 1)
|
||||
default:
|
||||
return "", errors.Err("dont know how to interpolate type " + field.Type().String())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// tabs to spaces, for easier copying into mysql prompt
|
||||
query = strings.Replace(query, "\t", " ", -1)
|
||||
|
||||
return query, nil
|
||||
}
|
||||
|
||||
// Qs is a shortcut for one group of positional placeholders
|
||||
func Qs(count int) string {
|
||||
return placeholders(false, count, 1, 1)
|
||||
}
|
||||
|
||||
// placeholders creates indexed or positional placeholders, in groups, with different starts
|
||||
func placeholders(indexPlaceholders bool, count int, start int, group int) string {
|
||||
buf := bytes.Buffer{}
|
||||
|
||||
if start == 0 || group == 0 {
|
||||
panic("invalid start or group numbers supplied.")
|
||||
}
|
||||
|
||||
if group > 1 {
|
||||
buf.WriteByte('(')
|
||||
}
|
||||
for i := 0; i < count; i++ {
|
||||
if i != 0 {
|
||||
if group > 1 && i%group == 0 {
|
||||
buf.WriteString("),(")
|
||||
} else {
|
||||
buf.WriteByte(',')
|
||||
}
|
||||
}
|
||||
if indexPlaceholders {
|
||||
buf.WriteString(fmt.Sprintf("$%d", start+i))
|
||||
} else {
|
||||
buf.WriteByte('?')
|
||||
}
|
||||
}
|
||||
if group > 1 {
|
||||
buf.WriteByte(')')
|
||||
}
|
||||
|
||||
return buf.String()
|
||||
}
|
||||
Loading…
Reference in a new issue