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
|
// Package sqlserver defines and registers usql's Microsoft SQL Server driver.
//
// See: https://github.com/microsoft/go-mssqldb
// Group: base
package sqlserver
import (
"context"
"database/sql"
"fmt"
"io"
"strconv"
"strings"
mssql "github.com/microsoft/go-mssqldb"
sqlserver "github.com/microsoft/go-mssqldb" // DRIVER
"github.com/xo/usql/drivers"
"github.com/xo/usql/drivers/metadata"
// needed for azuresql authentication, named pipes, and shared memory transport protocols
_ "github.com/microsoft/go-mssqldb/azuread"
_ "github.com/microsoft/go-mssqldb/namedpipe"
_ "github.com/microsoft/go-mssqldb/sharedmemory"
)
func init() {
drivers.Register("sqlserver", drivers.Driver{
AllowMultilineComments: true,
RequirePreviousPassword: true,
LexerName: "tsql",
/*
// NOTE: this has been commented out, as it is not necessary. if
// NOTE: the azuread.DriverName is changed from `azuresql`, then
// NOTE: this func will be necessary as dburl will never import non
// NOTE: stdlib package. as is, dburl.Open will handle the call
// NOTE: to sql.Open and will pass the `azuresql` driver name
Open: func(_ context.Context, u *dburl.URL, _, _ func() io.Writer) (func(string, string) (*sql.DB, error), error) {
return func(_ string, params string) (*sql.DB, error) {
driver := "sqlserver"
switch {
case u.Query().Has("fedauth"),
strings.Contains(strings.ToLower(u.OriginalScheme), "azuresql"):
driver = azuread.DriverName
}
return sql.Open(driver, params)
}, nil
},
*/
Version: func(ctx context.Context, db drivers.DB) (string, error) {
var ver, level, edition string
err := db.QueryRowContext(
ctx,
`SELECT SERVERPROPERTY('productversion'), SERVERPROPERTY ('productlevel'), SERVERPROPERTY ('edition')`,
).Scan(&ver, &level, &edition)
if err != nil {
return "", err
}
return "Microsoft SQL Server " + ver + ", " + level + ", " + edition, nil
},
ChangePassword: func(db drivers.DB, user, newpw, oldpw string) error {
_, err := db.Exec(`ALTER LOGIN ` + user + ` WITH password = '` + newpw + `' old_password = '` + oldpw + `'`)
return err
},
ColumnTypes: func(col *sql.ColumnType) (interface{}, error) {
switch col.DatabaseTypeName() {
case "UNIQUEIDENTIFIER":
if nullable, ok := col.Nullable(); ok && nullable {
return new(NullUniqueIdentifier), nil
}
return new(mssql.UniqueIdentifier), nil
}
return new(interface{}), nil
},
Err: func(err error) (string, string) {
if e, ok := err.(sqlserver.Error); ok {
return strconv.Itoa(int(e.Number)), e.Message
}
msg := err.Error()
if i := strings.LastIndex(msg, "sqlserver:"); i != -1 {
msg = msg[i:]
}
return "", msg
},
IsPasswordErr: func(err error) bool {
return strings.Contains(err.Error(), "Login failed for")
},
NewMetadataReader: NewReader,
NewMetadataWriter: func(db drivers.DB, w io.Writer, opts ...metadata.ReaderOption) metadata.Writer {
return metadata.NewDefaultWriter(NewReader(db, opts...))(db, w)
},
Copy: drivers.CopyWithInsert(placeholder),
})
}
func placeholder(n int) string {
return fmt.Sprintf("@p%d", n)
}
type NullUniqueIdentifier struct {
ID mssql.UniqueIdentifier
Valid bool
}
func (nui *NullUniqueIdentifier) Scan(v interface{}) error {
nui.Valid = false
if v == nil {
return nil
}
if err := nui.ID.Scan(v); err != nil {
return err
}
nui.Valid = true
return nil
}
func (nui NullUniqueIdentifier) String() string {
if nui.Valid {
return nui.ID.String()
}
return ""
}
|
