/
snowflake.go
70 lines (63 loc) · 1.88 KB
/
snowflake.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
// Package snowflake defines and registers usql's Snowflake driver.
//
// See: https://github.com/snowflakedb/gosnowflake
package snowflake
import (
"io"
"io/ioutil"
"strconv"
"github.com/sirupsen/logrus"
"github.com/snowflakedb/gosnowflake" // DRIVER: snowflake
"github.com/xo/tblfmt"
"github.com/xo/usql/drivers"
"github.com/xo/usql/drivers/metadata"
infos "github.com/xo/usql/drivers/metadata/informationschema"
"github.com/xo/usql/env"
)
func init() {
r := logrus.New()
r.Out, r.Level = ioutil.Discard, logrus.PanicLevel
var l gosnowflake.SFLogger = &logger{r}
gosnowflake.SetLogger(&l)
newReader := infos.New(
infos.WithPlaceholder(func(int) string { return "?" }),
infos.WithCustomClauses(map[infos.ClauseName]string{
infos.SequenceColumnsIncrement: "''",
}),
infos.WithFunctions(false),
infos.WithIndexes(false),
infos.WithConstraints(false),
)
drivers.Register("snowflake", drivers.Driver{
Err: func(err error) (string, string) {
if e, ok := err.(*gosnowflake.SnowflakeError); ok {
return strconv.Itoa(e.Number), e.Message
}
return "", err.Error()
},
NewMetadataReader: newReader,
NewMetadataWriter: func(db drivers.DB, w io.Writer, opts ...metadata.ReaderOption) metadata.Writer {
writerOpts := []metadata.WriterOption{
metadata.WithListAllDbs(func(pattern string, verbose bool) error {
return listAllDbs(db, w, pattern, verbose)
}),
}
return metadata.NewDefaultWriter(newReader(db, opts...), writerOpts...)(db, w)
},
})
}
// logger is an empty logger.
type logger struct {
*logrus.Logger
}
func (*logger) SetLogLevel(string) error { return nil }
func listAllDbs(db drivers.DB, w io.Writer, pattern string, verbose bool) error {
rows, err := db.Query("SHOW databases")
if err != nil {
return err
}
defer rows.Close()
params := env.Pall()
params["title"] = "List of databases"
return tblfmt.EncodeAll(w, rows, params)
}