mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-02 23:40:24 +08:00
110 lines
2.0 KiB
Go
110 lines
2.0 KiB
Go
package mssql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"mayfly-go/internal/db/dbm/dbi"
|
|
"mayfly-go/pkg/utils/collx"
|
|
"net/url"
|
|
"strings"
|
|
|
|
_ "github.com/microsoft/go-mssqldb"
|
|
)
|
|
|
|
func init() {
|
|
meta := new(Meta)
|
|
dbi.Register(DbTypeMssql, meta)
|
|
}
|
|
|
|
const (
|
|
DbTypeMssql dbi.DbType = "mssql"
|
|
)
|
|
|
|
type Meta struct {
|
|
}
|
|
|
|
func (mm *Meta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
|
|
err := d.IfUseSshTunnelChangeIpPort()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
query := url.Values{}
|
|
// The application name (default is go-mssqldb)
|
|
query.Add("app name", "mayfly")
|
|
// 指定与服务器协商加密的最低TLS版本
|
|
query.Add("tlsmin", "1.0")
|
|
// 连接超时时间10秒
|
|
query.Add("connection timeout", "10")
|
|
if d.Database != "" {
|
|
ss := strings.Split(d.Database, "/")
|
|
if len(ss) > 1 {
|
|
query.Add("database", ss[0])
|
|
query.Add("schema", ss[1])
|
|
} else {
|
|
query.Add("database", d.Database)
|
|
}
|
|
}
|
|
params := query.Encode()
|
|
if d.Params != "" {
|
|
if !strings.HasPrefix(d.Params, "&") {
|
|
params = params + "&" + d.Params
|
|
} else {
|
|
params = params + d.Params
|
|
}
|
|
}
|
|
|
|
const driverName = "mssql"
|
|
dsn := fmt.Sprintf("sqlserver://%s:%s@%s:%d?%s", url.PathEscape(d.Username), url.PathEscape(d.Password), d.Host, d.Port, params)
|
|
return sql.Open(driverName, dsn)
|
|
}
|
|
|
|
func (mm *Meta) GetDialect(conn *dbi.DbConn) dbi.Dialect {
|
|
return &MssqlDialect{dc: conn}
|
|
}
|
|
|
|
func (mm *Meta) GetMetadata(conn *dbi.DbConn) dbi.Metadata {
|
|
return &MssqlMetadata{dc: conn}
|
|
}
|
|
|
|
func (sm *Meta) GetDbDataTypes() []*dbi.DbDataType {
|
|
return collx.AsArray[*dbi.DbDataType](Bigint,
|
|
Numeric,
|
|
Bit,
|
|
Smallint,
|
|
Decimal,
|
|
Smallmoney,
|
|
Int,
|
|
Tinyint,
|
|
Money,
|
|
Float,
|
|
Real,
|
|
Date,
|
|
Datetimeoffset,
|
|
Datetime2,
|
|
Smalldatetime,
|
|
Datetime,
|
|
Time,
|
|
Char,
|
|
Varchar,
|
|
Text,
|
|
Nchar,
|
|
Nvarchar,
|
|
Ntext,
|
|
Binary,
|
|
Varbinary,
|
|
Cursor,
|
|
Rowversion,
|
|
Hierarchyid,
|
|
Uniqueidentifier,
|
|
Sql_variant,
|
|
Xml,
|
|
Table,
|
|
Geometry,
|
|
Geography,
|
|
)
|
|
}
|
|
|
|
func (sm *Meta) GetCommonTypeConverter() dbi.CommonTypeConverter {
|
|
return &commonTypeConverter{}
|
|
}
|