forked from crawlab-team/crawlab-core
/
mssql.go
60 lines (52 loc) · 1.23 KB
/
mssql.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
package utils
import (
"context"
"fmt"
"github.com/xulei324/crawlab-core/constants"
"github.com/xulei324/crawlab-core/models/models"
"github.com/upper/db/v4"
"github.com/upper/db/v4/adapter/mssql"
"time"
)
func GetMssqlSession(ds *models.DataSource) (s db.Session, err error) {
return getMssqlSession(context.Background(), ds)
}
func GetMssqlSessionWithTimeout(ds *models.DataSource, timeout time.Duration) (s db.Session, err error) {
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
return getMssqlSession(ctx, ds)
}
func getMssqlSession(ctx context.Context, ds *models.DataSource) (s db.Session, err error) {
// normalize settings
host := ds.Host
port := ds.Port
if ds.Host == "" {
host = constants.DefaultHost
}
if ds.Port == "" {
port = constants.DefaultMssqlPort
}
// connect settings
settings := mssql.ConnectionURL{
User: ds.Username,
Password: ds.Password,
Database: ds.Database,
Host: fmt.Sprintf("%s:%s", host, port),
Options: nil,
}
// session
done := make(chan struct{})
go func() {
s, err = mssql.Open(settings)
close(done)
}()
// wait for done
select {
case <-ctx.Done():
if ctx.Err() != nil {
err = ctx.Err()
}
case <-done:
}
return s, err
}