-
Notifications
You must be signed in to change notification settings - Fork 115
/
query.go
171 lines (149 loc) · 7.07 KB
/
query.go
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
// Copyright (c) 2017-2022 Snowflake Computing Inc. All rights reserved.
package gosnowflake
import (
"encoding/json"
"fmt"
"time"
)
type resultFormat string
const (
jsonFormat resultFormat = "json"
arrowFormat resultFormat = "arrow"
)
type execBindParameter struct {
Type string `json:"type"`
Value interface{} `json:"value"`
}
type execRequest struct {
SQLText string `json:"sqlText"`
AsyncExec bool `json:"asyncExec"`
SequenceID uint64 `json:"sequenceId"`
IsInternal bool `json:"isInternal"`
DescribeOnly bool `json:"describeOnly,omitempty"`
Parameters map[string]interface{} `json:"parameters,omitempty"`
Bindings map[string]execBindParameter `json:"bindings,omitempty"`
BindStage string `json:"bindStage,omitempty"`
QueryContext requestQueryContext `json:"queryContextDTO,omitempty"`
}
type requestQueryContext struct {
Entries []requestQueryContextEntry `json:"entries,omitempty"`
}
type requestQueryContextEntry struct {
Context contextData `json:"context,omitempty"`
ID int `json:"id"`
Priority int `json:"priority"`
Timestamp int64 `json:"timestamp,omitempty"`
}
type contextData struct {
Base64Data string `json:"base64Data,omitempty"`
}
type execResponseRowType struct {
Name string `json:"name"`
Fields []fieldMetadata `json:"fields"`
ByteLength int64 `json:"byteLength"`
Length int64 `json:"length"`
Type string `json:"type"`
Precision int64 `json:"precision"`
Scale int64 `json:"scale"`
Nullable bool `json:"nullable"`
}
func (ex *execResponseRowType) toFieldMetadata() fieldMetadata {
return fieldMetadata{
ex.Name,
ex.Type,
int(ex.Scale),
ex.Fields,
}
}
type fieldMetadata struct {
Name string `json:"name"`
Type string `json:"type"`
Scale int `json:"scale"`
Fields []fieldMetadata `json:"fields"`
}
func (fm *fieldMetadata) String() string {
return fmt.Sprintf("fieldMetadata: %v, %v, %v, %v", fm.Name, fm.Type, fm.Scale, fm.Fields)
}
type execResponseChunk struct {
URL string `json:"url"`
RowCount int `json:"rowCount"`
UncompressedSize int64 `json:"uncompressedSize"`
CompressedSize int64 `json:"compressedSize"`
}
type execResponseCredentials struct {
AwsKeyID string `json:"AWS_KEY_ID,omitempty"`
AwsSecretKey string `json:"AWS_SECRET_KEY,omitempty"`
AwsToken string `json:"AWS_TOKEN,omitempty"`
AwsID string `json:"AWS_ID,omitempty"`
AwsKey string `json:"AWS_KEY,omitempty"`
AzureSasToken string `json:"AZURE_SAS_TOKEN,omitempty"`
GcsAccessToken string `json:"GCS_ACCESS_TOKEN,omitempty"`
}
type execResponseStageInfo struct {
LocationType string `json:"locationType,omitempty"`
Location string `json:"location,omitempty"`
Path string `json:"path,omitempty"`
Region string `json:"region,omitempty"`
StorageAccount string `json:"storageAccount,omitempty"`
IsClientSideEncrypted bool `json:"isClientSideEncrypted,omitempty"`
Creds execResponseCredentials `json:"creds,omitempty"`
PresignedURL string `json:"presignedUrl,omitempty"`
EndPoint string `json:"endPoint,omitempty"`
}
// make all data field optional
type execResponseData struct {
// succeed query response data
Parameters []nameValueParameter `json:"parameters,omitempty"`
RowType []execResponseRowType `json:"rowtype,omitempty"`
RowSet [][]*string `json:"rowset,omitempty"`
RowSetBase64 string `json:"rowsetbase64,omitempty"`
Total int64 `json:"total,omitempty"` // java:long
Returned int64 `json:"returned,omitempty"` // java:long
QueryID string `json:"queryId,omitempty"`
SQLState string `json:"sqlState,omitempty"`
DatabaseProvider string `json:"databaseProvider,omitempty"`
FinalDatabaseName string `json:"finalDatabaseName,omitempty"`
FinalSchemaName string `json:"finalSchemaName,omitempty"`
FinalWarehouseName string `json:"finalWarehouseName,omitempty"`
FinalRoleName string `json:"finalRoleName,omitempty"`
NumberOfBinds int `json:"numberOfBinds,omitempty"` // java:int
StatementTypeID int64 `json:"statementTypeId,omitempty"` // java:long
Version int64 `json:"version,omitempty"` // java:long
Chunks []execResponseChunk `json:"chunks,omitempty"`
Qrmk string `json:"qrmk,omitempty"`
ChunkHeaders map[string]string `json:"chunkHeaders,omitempty"`
// ping pong response data
GetResultURL string `json:"getResultUrl,omitempty"`
ProgressDesc string `json:"progressDesc,omitempty"`
QueryAbortTimeout time.Duration `json:"queryAbortsAfterSecs,omitempty"`
ResultIDs string `json:"resultIds,omitempty"`
ResultTypes string `json:"resultTypes,omitempty"`
QueryResultFormat string `json:"queryResultFormat,omitempty"`
// async response placeholders
AsyncResult *snowflakeResult `json:"asyncResult,omitempty"`
AsyncRows *snowflakeRows `json:"asyncRows,omitempty"`
// file transfer response data
UploadInfo execResponseStageInfo `json:"uploadInfo,omitempty"`
LocalLocation string `json:"localLocation,omitempty"`
SrcLocations []string `json:"src_locations,omitempty"`
Parallel int64 `json:"parallel,omitempty"`
Threshold int64 `json:"threshold,omitempty"`
AutoCompress bool `json:"autoCompress,omitempty"`
Overwrite bool `json:"overwrite,omitempty"`
SourceCompression string `json:"sourceCompression,omitempty"`
ShowEncryptionParameter bool `json:"clientShowEncryptionParameter,omitempty"`
EncryptionMaterial encryptionWrapper `json:"encryptionMaterial,omitempty"`
PresignedURLs []string `json:"presignedUrls,omitempty"`
StageInfo execResponseStageInfo `json:"stageInfo,omitempty"`
Command string `json:"command,omitempty"`
Kind string `json:"kind,omitempty"`
Operation string `json:"operation,omitempty"`
// HTAP
QueryContext json.RawMessage `json:"queryContext,omitempty"`
}
type execResponse struct {
Data execResponseData `json:"Data"`
Message string `json:"message"`
Code string `json:"code"`
Success bool `json:"success"`
}