76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
|
package base
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"github.com/gogf/gf/contrib/drivers/pgsql/v2"
|
||
|
|
"github.com/gogf/gf/v2/database/gdb"
|
||
|
|
)
|
||
|
|
|
||
|
|
type GDB struct {
|
||
|
|
*pgsql.Driver
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
// customDriverName is my driver name, which is used for registering.
|
||
|
|
customDriverName = "MyDriver"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
// It here registers my custom driver in package initialization function "init".
|
||
|
|
// You can later use this type in the database configuration.
|
||
|
|
if err := gdb.Register(customDriverName, &GDB{}); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// New creates and returns a database object for mysql.
|
||
|
|
// It implements the interface of gdb.Driver for extra database driver installation.
|
||
|
|
func (d *GDB) New(core *gdb.Core, node *gdb.ConfigNode) (gdb.DB, error) {
|
||
|
|
return &GDB{
|
||
|
|
&pgsql.Driver{
|
||
|
|
Core: core,
|
||
|
|
},
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// DoCommit commits current sql and arguments to underlying sql driver.
|
||
|
|
func (d *GDB) DoCommit(ctx context.Context, in gdb.DoCommitInput) (out gdb.DoCommitOutput, err error) {
|
||
|
|
out, err = d.Core.DoCommit(ctx, in)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// 其他接口方法
|
||
|
|
func (d *GDB) DoFilter(ctx context.Context, link gdb.Link, sql string, args []any) (string, []any, error) {
|
||
|
|
return d.Core.DoFilter(ctx, link, sql, args)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *GDB) DoPrepare(ctx context.Context, link gdb.Link, sql string) (*gdb.Stmt, error) {
|
||
|
|
return d.Core.DoPrepare(ctx, link, sql)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *GDB) ConvertValueForField(ctx context.Context, fieldType string, fieldValue any) (any, error) {
|
||
|
|
return fieldValue, nil
|
||
|
|
}
|
||
|
|
func (d *GDB) ConvertValueForLocal(ctx context.Context, fieldType string, fieldValue any) (any, error) {
|
||
|
|
return d.Core.ConvertValueForLocal(ctx, fieldType, fieldValue)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *GDB) Exec(ctx context.Context, sql string, args ...any) (sql.Result, error) {
|
||
|
|
in := gdb.DoCommitInput{Type: gdb.SqlTypeExecContext, Sql: sql, Args: args}
|
||
|
|
out, err := d.DoCommit(ctx, in)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return out.Result, nil
|
||
|
|
}
|
||
|
|
func (d *GDB) DoExec(ctx context.Context, link gdb.Link, sql string, args ...any) (sql.Result, error) {
|
||
|
|
in := gdb.DoCommitInput{Type: gdb.SqlTypeExecContext, Sql: sql, Args: args}
|
||
|
|
out, err := d.DoCommit(ctx, in)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return out.Result, nil
|
||
|
|
}
|