1 files changed, 89 insertions, 0 deletions
diff --git a/vendor/github.com/go-pg/pg/v10/orm/composite_create.go b/vendor/github.com/go-pg/pg/v10/orm/composite_create.go
new file mode 100644
index 000000000..fd60a94e4
--- /dev/null
+++ b/vendor/github.com/go-pg/pg/v10/orm/composite_create.go
@@ -0,0 +1,89 @@
+package orm
+
+import (
+	"strconv"
+)
+
+type CreateCompositeOptions struct {
+	Varchar int // replaces PostgreSQL data type `text` with `varchar(n)`
+}
+
+type CreateCompositeQuery struct {
+	q   *Query
+	opt *CreateCompositeOptions
+}
+
+var (
+	_ QueryAppender = (*CreateCompositeQuery)(nil)
+	_ QueryCommand  = (*CreateCompositeQuery)(nil)
+)
+
+func NewCreateCompositeQuery(q *Query, opt *CreateCompositeOptions) *CreateCompositeQuery {
+	return &CreateCompositeQuery{
+		q:   q,
+		opt: opt,
+	}
+}
+
+func (q *CreateCompositeQuery) String() string {
+	b, err := q.AppendQuery(defaultFmter, nil)
+	if err != nil {
+		panic(err)
+	}
+	return string(b)
+}
+
+func (q *CreateCompositeQuery) Operation() QueryOp {
+	return CreateCompositeOp
+}
+
+func (q *CreateCompositeQuery) Clone() QueryCommand {
+	return &CreateCompositeQuery{
+		q:   q.q.Clone(),
+		opt: q.opt,
+	}
+}
+
+func (q *CreateCompositeQuery) Query() *Query {
+	return q.q
+}
+
+func (q *CreateCompositeQuery) AppendTemplate(b []byte) ([]byte, error) {
+	return q.AppendQuery(dummyFormatter{}, b)
+}
+
+func (q *CreateCompositeQuery) AppendQuery(fmter QueryFormatter, b []byte) ([]byte, error) {
+	if q.q.stickyErr != nil {
+		return nil, q.q.stickyErr
+	}
+	if q.q.tableModel == nil {
+		return nil, errModelNil
+	}
+
+	table := q.q.tableModel.Table()
+
+	b = append(b, "CREATE TYPE "...)
+	b = append(b, table.Alias...)
+	b = append(b, " AS ("...)
+
+	for i, field := range table.Fields {
+		if i > 0 {
+			b = append(b, ", "...)
+		}
+
+		b = append(b, field.Column...)
+		b = append(b, " "...)
+		if field.UserSQLType == "" && q.opt != nil && q.opt.Varchar > 0 &&
+			field.SQLType == "text" {
+			b = append(b, "varchar("...)
+			b = strconv.AppendInt(b, int64(q.opt.Varchar), 10)
+			b = append(b, ")"...)
+		} else {
+			b = append(b, field.SQLType...)
+		}
+	}
+
+	b = append(b, ")"...)
+
+	return b, q.q.stickyErr
+}