sqlc/internal/compiler/query.go at a502b2d2bed17609c7aad68c9cc65065f7d9aabd · sqlc-dev/sqlc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package compiler

import (
	"github.com/sqlc-dev/sqlc/internal/metadata"
	"github.com/sqlc-dev/sqlc/internal/sql/ast"
	"github.com/sqlc-dev/sqlc/internal/sql/catalog"
)

type Function struct {
	Rel        *ast.FuncName
	ReturnType *ast.TypeName
	Outs       []*catalog.Argument
}

type Table struct {
	Rel     *ast.TableName
	Columns []*Column
}

type Column struct {
	Name         string
	OriginalName string
	DataType     string
	NotNull      bool
	Unsigned     bool
	IsArray      bool
	ArrayDims    int
	Comment      string
	Length       *int
	IsNamedParam bool
	IsFuncCall   bool

	// XXX: Figure out what PostgreSQL calls `foo.id`
	Scope      string
	Table      *ast.TableName
	TableAlias string
	Type       *ast.TypeName
	EmbedTable *ast.TableName

	IsSqlcSlice bool // is this sqlc.slice()

	skipTableRequiredCheck bool
}

type Query struct {
	SQL      string
	Metadata metadata.Metadata
	Columns  []*Column
	Params   []Parameter

	// Needed for CopyFrom
	InsertIntoTable *ast.TableName

	// Target table for UPDATE queries
	UpdateTable *ast.TableName

	// Target table for DELETE queries
	DeleteFromTable *ast.TableName

	// Needed for vet
	RawStmt *ast.RawStmt
}

type ParameterContext int

const (
	ParameterContextUnspecified ParameterContext = iota
	ParameterContextSet
	ParameterContextValues
	ParameterContextWhere
	ParameterContextHaving
	ParameterContextFunctionArg
	ParameterContextLimit
	ParameterContextOffset
)

type Parameter struct {
	Number  int
	Column  *Column
	Context ParameterContext
}