mirror of
https://github.com/grafana/grafana.git
synced 2024-11-25 10:20:29 -06:00
9d92818cae
* update with sdk * do sql * fix core plugins * fix proxy settings * bump SDK version * tidy * enable pdc for test * add codeowners * bump dep * go mod tidy * bump SDK
105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
package mssql
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"database/sql/driver"
|
|
"errors"
|
|
"net"
|
|
|
|
sdkproxy "github.com/grafana/grafana-plugin-sdk-go/backend/proxy"
|
|
"github.com/grafana/grafana/pkg/tsdb/sqleng"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
mssql "github.com/microsoft/go-mssqldb"
|
|
"golang.org/x/net/proxy"
|
|
"xorm.io/core"
|
|
)
|
|
|
|
// createMSSQLProxyDriver creates and registers a new sql driver that uses a mssql connector and updates the dialer to
|
|
// route connections through the secure socks proxy
|
|
func createMSSQLProxyDriver(cnnstr string, hostName string, opts *sdkproxy.Options) (string, error) {
|
|
sqleng.XormDriverMu.Lock()
|
|
defer sqleng.XormDriverMu.Unlock()
|
|
|
|
// create a unique driver per connection string
|
|
hash, err := util.Md5SumString(cnnstr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
driverName := "mssql-proxy-" + hash
|
|
|
|
// only register the driver once
|
|
if core.QueryDriver(driverName) == nil {
|
|
connector, err := mssql.NewConnector(cnnstr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
driver, err := newMSSQLProxyDriver(connector, hostName, opts)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
sql.Register(driverName, driver)
|
|
core.RegisterDriver(driverName, driver)
|
|
}
|
|
|
|
return driverName, nil
|
|
}
|
|
|
|
type HostTransportDialer struct {
|
|
Dialer proxy.ContextDialer
|
|
Host string
|
|
}
|
|
|
|
func (m HostTransportDialer) DialContext(ctx context.Context, network string, addr string) (conn net.Conn, err error) {
|
|
return m.Dialer.DialContext(ctx, network, addr)
|
|
}
|
|
|
|
func (m HostTransportDialer) HostName() string {
|
|
return m.Host
|
|
}
|
|
|
|
// mssqlProxyDriver is a regular mssql driver with an updated dialer.
|
|
// This is needed because there is no way to save a dialer to the mssql driver in xorm
|
|
type mssqlProxyDriver struct {
|
|
c *mssql.Connector
|
|
}
|
|
|
|
var _ driver.DriverContext = (*mssqlProxyDriver)(nil)
|
|
var _ core.Driver = (*mssqlProxyDriver)(nil)
|
|
|
|
// newMSSQLProxyDriver updates the dialer for a mssql connector with a dialer that proxys connections through the secure socks proxy
|
|
// and returns a new mssql driver to register
|
|
func newMSSQLProxyDriver(connector *mssql.Connector, hostName string, opts *sdkproxy.Options) (*mssqlProxyDriver, error) {
|
|
dialer, err := sdkproxy.New(opts).NewSecureSocksProxyContextDialer()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
contextDialer, ok := dialer.(proxy.ContextDialer)
|
|
if !ok {
|
|
return nil, errors.New("unable to cast socks proxy dialer to context proxy dialer")
|
|
}
|
|
|
|
connector.Dialer = HostTransportDialer{contextDialer, hostName}
|
|
return &mssqlProxyDriver{c: connector}, nil
|
|
}
|
|
|
|
// Parse uses the xorm mssql dialect for the driver (this has to be implemented to register the driver with xorm)
|
|
func (d *mssqlProxyDriver) Parse(a string, b string) (*core.Uri, error) {
|
|
sqleng.XormDriverMu.RLock()
|
|
defer sqleng.XormDriverMu.RUnlock()
|
|
|
|
return core.QueryDriver("mssql").Parse(a, b)
|
|
}
|
|
|
|
// OpenConnector returns the normal mssql connector that has the updated dialer context
|
|
func (d *mssqlProxyDriver) OpenConnector(name string) (driver.Connector, error) {
|
|
return d.c, nil
|
|
}
|
|
|
|
// Open uses the connector with the updated dialer context to open a new connection
|
|
func (d *mssqlProxyDriver) Open(dsn string) (driver.Conn, error) {
|
|
return d.c.Connect(context.Background())
|
|
}
|