update db default conn

db/db.go

@@ -19,6 +19,8 @@ var (
 	db     *sqlx.DB
 	err    error
 	once   sync.Once
+
+	defaultDB *DB
 )
 
 // DB define
@@ -35,6 +37,8 @@ func SetConfig(cfg Config) {
 	config.MaxOpenConns = cfg.MaxOpenConns
 	config.MaxIdle = cfg.MaxIdle
 	config.MaxLifetime = cfg.MaxLifetime * time.Second
+
+	defaultDB = &DB{Driver: config.Driver}
 }
 
 // New new DB object
@@ -330,3 +334,25 @@ func (d *DB) Limit(page, pagesize int) string {
 	// // PostgreSQL limit size offset 0
 	return fmt.Sprintf(" limit %d offset %d", pagesize, (page-1)*pagesize)
 }
+
+// Select select
+func Select(dest interface{}, query string, args ...interface{}) (err error) {
+	defaultDB.conn, err = connect()
+	if err != nil {
+		return err
+	}
+
+	err = defaultDB.conn.Select(dest, query, args...)
+	return
+}
+
+// Get get
+func Get(dest interface{}, query string, args ...interface{}) (err error) {
+	defaultDB.conn, err = connect()
+	if err != nil {
+		return
+	}
+
+	err = defaultDB.conn.Get(dest, query, args...)
+	return
+}

db/init.go

@@ -0,0 +1,5 @@
+package db
+
+func init() {
+
+}