db

Documentation

Overview

*

• @author huqiuyun
• 数据Elasticsearch

*

• @author huqiuyun
• 数据服务

Index

• Variables
• func AddDriver(driverName string, dial DialFunc)
• func UseServerPubKey(cnf *config.Database, opts ...commonoption.Option) error
• func UseTls(cnf *config.Database, opts ...commonoption.Option) error
• func WithDnsOption(f func(cnf *config.Database, tls, pubKey bool) string) commonoption.Option
• func WithServerPubKeyOption(f func(cnf *config.Database) (*rsa.PublicKey, error)) commonoption.Option
• func WithTlsOption(f func(cnf *config.Database) (*tls.Config, error)) commonoption.Option
• type Adapter
  • func NewAdapter(cfg *config.Sql, opts ...commonoption.Option) (*Adapter, error)
  • func (a *Adapter) Close()
  • func (a *Adapter) Db() *gorm.DB
  • func (a *Adapter) Name() string
• type ConditionCallback
• type DialFunc
• type EsDatabase
  • func NewES(conf []config.Elastic) (EsDatabase, error)
• type Logger
  • func (l *Logger) Error(ctx context.Context, msg string, data ...interface{})
  • func (l *Logger) Info(ctx context.Context, msg string, data ...interface{})
  • func (l *Logger) LogMode(level gormLogger.LogLevel) gormLogger.Interface
  • func (l *Logger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error)
  • func (l *Logger) Warn(ctx context.Context, msg string, data ...interface{})
• type ScopesFunc
• type ServerPubKeyFunc
  • func (t ServerPubKeyFunc) String() string
  • func (t ServerPubKeyFunc) Type() int
  • func (t ServerPubKeyFunc) Value() interface{}
• type Sql
• type SqlDatabase
  • func NewDB(cfgs []config.Sql, opts ...commonoption.Option) (SqlDatabase, error)

Variables

var DefaultDnsFunc = func(cnf *config.Database, tls, serverPubKey bool) string {
	if cnf == nil {
		return ""
	}
	if cnf.DriverType == "mysql" {
		if cnf.Charset == "" {
			cnf.Charset = "utf8mb4"
		}
		dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local", cnf.User, cnf.Pwd, cnf.Host, cnf.Port, cnf.Name, cnf.Charset)
		if tls {
			dsn += "&tls=" + url.QueryEscape(cnf.Tls.Key)
		}
		if serverPubKey {
			dsn += "&serverPubKey=" + url.QueryEscape(cnf.ServerPubKey.Key)
		}
		return dsn
	} else if cnf.DriverType == "postgres" {
		return fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s", cnf.Host, cnf.Port, cnf.User, cnf.Pwd, cnf.Name)
	} else if cnf.DriverType == "sqlite3" {
		return cnf.Name
	} else if cnf.DriverType == "sql" {
		return fmt.Sprintf("sqlserver://%s:%s@%s:%d?database=%s", cnf.User, cnf.Pwd, cnf.Host, cnf.Port, cnf.Name)
	}
	return ""
}

var DefaultTlsFunc = func(cnf *config.Database) (*tls.Config, error) {
	if strings.EqualFold(cnf.Tls.Key, "true") ||
		strings.EqualFold(cnf.Tls.Key, "false") ||
		strings.EqualFold(cnf.Tls.Key, "skip-verify") ||
		strings.EqualFold(cnf.Tls.Key, "preferred") {
		return nil, nil
	}
	cert, rootCertPool, err := commontls.GetCertificate(&cnf.Tls)
	if err != nil {
		return nil, err
	}
	return &tls.Config{
		RootCAs:      rootCertPool,
		Certificates: []tls.Certificate{cert},
	}, nil
}

Functions

func AddDriver

func AddDriver(driverName string, dial DialFunc)

func UseServerPubKey

func UseServerPubKey(cnf *config.Database, opts ...commonoption.Option) error

func UseTls

func UseTls(cnf *config.Database, opts ...commonoption.Option) error

func WithDnsOption

func WithDnsOption(f func(cnf *config.Database, tls, pubKey bool) string) commonoption.Option

func WithServerPubKeyOption

func WithServerPubKeyOption(f func(cnf *config.Database) (*rsa.PublicKey, error)) commonoption.Option

func WithTlsOption

func WithTlsOption(f func(cnf *config.Database) (*tls.Config, error)) commonoption.Option

Types

type Adapter

type Adapter struct {
	// contains filtered or unexported fields
}

Adapter represents the Gorm adapter for policy storage.

func NewAdapter

func NewAdapter(cfg *config.Sql, opts ...commonoption.Option) (*Adapter, error)

NewAdapter is the constructor for Adapter.

func (*Adapter) Close

func (a *Adapter) Close()

func (*Adapter) Db

func (a *Adapter) Db() *gorm.DB

func (*Adapter) Name

func (a *Adapter) Name() string

type ConditionCallback

type ConditionCallback func(ctx context.Context, data interface{}, db *gorm.DB) *gorm.DB

type DialFunc

type DialFunc func(string) gorm.Dialector

type EsDatabase

type EsDatabase interface {
	Get(name string) (*elastic.Client, string, error)
	Do(name string, call func(es *elastic.Client) error) error
	Close(name string) error
	CloseAll()
	Select(name string)
	GetSelector() string
}

func NewES

func NewES(conf []config.Elastic) (EsDatabase, error)

type Logger

type Logger struct {
	config.SqlLogger
}

func (*Logger) Error

func (l *Logger) Error(ctx context.Context, msg string, data ...interface{})

Error print error messages

func (*Logger) Info

func (l *Logger) Info(ctx context.Context, msg string, data ...interface{})

Info print info

func (*Logger) LogMode

func (l *Logger) LogMode(level gormLogger.LogLevel) gormLogger.Interface

LogMode log mode

func (*Logger) Trace

func (l *Logger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error)

Trace print sql message

func (*Logger) Warn

func (l *Logger) Warn(ctx context.Context, msg string, data ...interface{})

Warn print warn messages

type ScopesFunc

type ScopesFunc func(db *gorm.DB) *gorm.DB

type ServerPubKeyFunc

type ServerPubKeyFunc func(cnf *config.Database) (*rsa.PublicKey, error)

func (ServerPubKeyFunc) String

func (t ServerPubKeyFunc) String() string

func (ServerPubKeyFunc) Type

func (t ServerPubKeyFunc) Type() int

func (ServerPubKeyFunc) Value

func (t ServerPubKeyFunc) Value() interface{}

type Sql

type Sql interface {
	Db() *gorm.DB
	Name() string
}

type SqlDatabase

type SqlDatabase interface {
	Get(name string) (Sql, error)
	Do(name string, c func(db Sql) error) error
	Close(name string)
	CloseAll()
	Select(name string)
	GetSelector() string
}

func NewDB

func NewDB(cfgs []config.Sql, opts ...commonoption.Option) (SqlDatabase, error)