From 98263a7de64269898a2f81207e38943b5c8e8653 Mon Sep 17 00:00:00 2001
From: Tobi Smethurst <31960611+tsmethurst@users.noreply.github.com>
Date: Thu, 12 Aug 2021 21:03:24 +0200
Subject: Grand test fixup (#138)

* start fixing up tests

* fix up tests + automate with drone

* fiddle with linting

* messing about with drone.yml

* some more fiddling

* hmmm

* add cache

* add vendor directory

* verbose

* ci updates

* update some little things

* update sig
---
 .../go-pg/pg/v10/orm/composite_create.go           | 89 ++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 vendor/github.com/go-pg/pg/v10/orm/composite_create.go

(limited to 'vendor/github.com/go-pg/pg/v10/orm/composite_create.go')

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
+}
-- 
cgit v1.2.3