11package database
22
33import (
4+ "os"
45 "context"
56 "database/sql"
67 "fmt"
@@ -10,6 +11,8 @@ import (
1011
1112 _ "github.com/denisenkom/go-mssqldb"
1213 "github.com/sqls-server/sqls/dialect"
14+ "github.com/jfcote87/sshdb"
15+ "github.com/jfcote87/sshdb/mssql"
1316 "golang.org/x/crypto/ssh"
1417)
1518
@@ -21,21 +24,51 @@ func init() {
2124func mssqlOpen (dbConnCfg * DBConfig ) (* DBConnection , error ) {
2225 var (
2326 conn * sql.DB
24- sshConn * ssh.Client
2527 )
2628 dsn , err := genMssqlConfig (dbConnCfg )
2729 if err != nil {
2830 return nil , err
2931 }
3032
3133 if dbConnCfg .SSHCfg != nil {
32- return nil , fmt .Errorf ("connect via SSH is not supported" )
33- }
34- dbConn , err := sql .Open ("sqlserver" , dsn )
35- if err != nil {
36- return nil , err
34+ key , err := os .ReadFile (dbConnCfg .SSHCfg .PrivateKey )
35+ if err != nil {
36+ return nil , fmt .Errorf ("unable to open private key" )
37+ }
38+
39+ signer , err := ssh .ParsePrivateKeyWithPassphrase (key , []byte (dbConnCfg .SSHCfg .PassPhrase ))
40+ if err != nil {
41+ return nil , fmt .Errorf ("unable to decrypt private key" )
42+ }
43+
44+ cfg := & ssh.ClientConfig {
45+ User : dbConnCfg .SSHCfg .User ,
46+ Auth : []ssh.AuthMethod {
47+ ssh .PublicKeys (signer ),
48+ },
49+ HostKeyCallback : ssh .InsecureIgnoreHostKey (),
50+ }
51+
52+ remoteAddr := fmt .Sprintf ("%s:%d" , dbConnCfg .SSHCfg .Host , dbConnCfg .SSHCfg .Port )
53+
54+ tunnel , err := sshdb .New (cfg , remoteAddr )
55+ if err != nil {
56+ return nil , fmt .Errorf ("%w" , err )
57+ }
58+
59+ connector , err := tunnel .OpenConnector (mssql .TunnelDriver , dsn )
60+ if err != nil {
61+ return nil , err
62+ }
63+
64+ conn = sql .OpenDB (connector )
65+ } else {
66+ conn , err = sql .Open ("mssql" , dsn )
67+ if err != nil {
68+ return nil , err
69+ }
3770 }
38- conn = dbConn
71+
3972 if err = conn .Ping (); err != nil {
4073 return nil , err
4174 }
@@ -45,7 +78,6 @@ func mssqlOpen(dbConnCfg *DBConfig) (*DBConnection, error) {
4578
4679 return & DBConnection {
4780 Conn : conn ,
48- SSHConn : sshConn ,
4981 }, nil
5082}
5183
0 commit comments