summaryrefslogtreecommitdiff
path: root/vendor/github.com/uptrace/bun/dialect/pgdialect/append.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/uptrace/bun/dialect/pgdialect/append.go')
-rw-r--r--vendor/github.com/uptrace/bun/dialect/pgdialect/append.go55
1 files changed, 55 insertions, 0 deletions
diff --git a/vendor/github.com/uptrace/bun/dialect/pgdialect/append.go b/vendor/github.com/uptrace/bun/dialect/pgdialect/append.go
index d5e0d0a57..a60bf5de2 100644
--- a/vendor/github.com/uptrace/bun/dialect/pgdialect/append.go
+++ b/vendor/github.com/uptrace/bun/dialect/pgdialect/append.go
@@ -307,3 +307,58 @@ func arrayAppendString(b []byte, s string) []byte {
b = append(b, '"')
return b
}
+
+//------------------------------------------------------------------------------
+
+var mapStringStringType = reflect.TypeOf(map[string]string(nil))
+
+func (d *Dialect) hstoreAppender(typ reflect.Type) schema.AppenderFunc {
+ kind := typ.Kind()
+
+ switch kind {
+ case reflect.Ptr:
+ if fn := d.hstoreAppender(typ.Elem()); fn != nil {
+ return schema.PtrAppender(fn)
+ }
+ case reflect.Map:
+ // ok:
+ default:
+ return nil
+ }
+
+ if typ.Key() == stringType && typ.Elem() == stringType {
+ return appendMapStringStringValue
+ }
+
+ return func(fmter schema.Formatter, b []byte, v reflect.Value) []byte {
+ err := fmt.Errorf("bun: Hstore(unsupported %s)", v.Type())
+ return dialect.AppendError(b, err)
+ }
+}
+
+func appendMapStringString(b []byte, m map[string]string) []byte {
+ if m == nil {
+ return dialect.AppendNull(b)
+ }
+
+ b = append(b, '\'')
+
+ for key, value := range m {
+ b = arrayAppendString(b, key)
+ b = append(b, '=', '>')
+ b = arrayAppendString(b, value)
+ b = append(b, ',')
+ }
+ if len(m) > 0 {
+ b = b[:len(b)-1] // Strip trailing comma.
+ }
+
+ b = append(b, '\'')
+
+ return b
+}
+
+func appendMapStringStringValue(fmter schema.Formatter, b []byte, v reflect.Value) []byte {
+ m := v.Convert(mapStringStringType).Interface().(map[string]string)
+ return appendMapStringString(b, m)
+}