代码拉取完成,页面将自动刷新
package sharding
import (
"context"
"database/sql"
"gorm.io/gorm"
)
// ConnPool Implement a ConnPool for replace db.Statement.ConnPool in Gorm
type ConnPool struct {
// db, This is global db instance
sharding *Sharding
gorm.ConnPool
}
func (pool *ConnPool) String() string {
return "gorm:sharding:conn_pool"
}
func (pool ConnPool) PrepareContext(ctx context.Context, query string) (*sql.Stmt, error) {
return pool.ConnPool.PrepareContext(ctx, query)
}
func (pool ConnPool) ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error) {
ftQuery, stQuery, table, err := pool.sharding.resolve(query, args...)
if err != nil {
return nil, err
}
pool.sharding.querys.Store("last_query", stQuery)
if table != "" {
if r, ok := pool.sharding.configs[table]; ok {
if r.DoubleWrite {
pool.ConnPool.ExecContext(ctx, ftQuery, args...)
}
}
}
return pool.ConnPool.ExecContext(ctx, stQuery, args...)
}
// https://github.com/go-gorm/gorm/blob/v1.21.11/callbacks/query.go#L18
func (pool ConnPool) QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error) {
ftQuery, stQuery, table, err := pool.sharding.resolve(query, args...)
if err != nil {
return nil, err
}
pool.sharding.querys.Store("last_query", stQuery)
if table != "" {
if r, ok := pool.sharding.configs[table]; ok {
if r.DoubleWrite {
pool.ConnPool.ExecContext(ctx, ftQuery, args...)
}
}
}
return pool.ConnPool.QueryContext(ctx, stQuery, args...)
}
func (pool ConnPool) QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row {
_, query, _, _ = pool.sharding.resolve(query, args...)
pool.sharding.querys.Store("last_query", query)
return pool.ConnPool.QueryRowContext(ctx, query, args...)
}
// BeginTx Implement ConnPoolBeginner.BeginTx
func (pool *ConnPool) BeginTx(ctx context.Context, opt *sql.TxOptions) (gorm.ConnPool, error) {
if basePool, ok := pool.ConnPool.(gorm.ConnPoolBeginner); ok {
return basePool.BeginTx(ctx, opt)
}
return pool, nil
}
// Implement TxCommitter.Commit
func (pool *ConnPool) Commit() error {
if _, ok := pool.ConnPool.(*sql.Tx); ok {
return nil
}
if basePool, ok := pool.ConnPool.(gorm.TxCommitter); ok {
return basePool.Commit()
}
return nil
}
// Implement TxCommitter.Rollback
func (pool *ConnPool) Rollback() error {
if _, ok := pool.ConnPool.(*sql.Tx); ok {
return nil
}
if basePool, ok := pool.ConnPool.(gorm.TxCommitter); ok {
return basePool.Rollback()
}
return nil
}
func (pool *ConnPool) Ping() error {
return nil
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。