m

code/base-project/base-go/pkg/utils/db/db.go

@@ -0,0 +1,68 @@
+package db
+
+import (
+	"base-framework/pkg/config"
+	"base-framework/pkg/router"
+	"database/sql"
+	"errors"
+	"fmt"
+)
+
+type Client struct {
+	conn *sql.DB
+	tx   *sql.Tx
+}
+
+// JdbcTemplate 创建 JdbcTemplate
+func JdbcTemplate(c *router.Context) (*Client, error) {
+	val, ok := c.Get("orgId")
+	if !ok {
+		return nil, errors.New("missing orgId in context")
+	}
+	orgId, ok := val.(string)
+	if !ok || orgId == "" {
+		return nil, errors.New("invalid orgId in context")
+	}
+	conn, ok := config.GetDB(orgId)
+	if !ok || conn == nil {
+		return nil, fmt.Errorf("no db connection found for orgId=%s", orgId)
+	}
+	return &Client{conn: conn}, nil
+}
+
+// ------------------------ 内部方法 ------------------------
+
+// 执行查询，返回 *sql.Rows
+func (c *Client) query(query string, args ...any) (*sql.Rows, error) {
+	if c.tx != nil {
+		return c.tx.Query(query, args...)
+	}
+	return c.conn.Query(query, args...)
+}
+
+// 执行执行类语句（insert/update/delete）
+func (c *Client) exec(query string, args ...any) (sql.Result, error) {
+	if c.tx != nil {
+		return c.tx.Exec(query, args...)
+	}
+	return c.conn.Exec(query, args...)
+}
+
+// WithTransaction 自动处理事务提交或回滚
+func (c *Client) WithTransaction(fn func(txClient *Client) error) error {
+	if c.tx != nil {
+		// 已经在事务中，直接执行
+		return fn(c)
+	}
+
+	tx, err := c.conn.Begin()
+	if err != nil {
+		return err
+	}
+	txClient := &Client{conn: c.conn, tx: tx}
+	if err := fn(txClient); err != nil {
+		_ = tx.Rollback()
+		return err
+	}
+	return tx.Commit()
+}

code/base-project/base-go/pkg/utils/db/delete.go

@@ -0,0 +1,5 @@
+package db
+
+import "database/sql"
+
+func (c *Client) Delete(query string, args ...any) (sql.Result, error) { return c.exec(query, args...) }

code/base-project/base-go/pkg/utils/db/insert.go

@@ -0,0 +1,15 @@
+package db
+
+import "database/sql"
+
+func (c *Client) Insert(query string, args ...any) (sql.Result, error) { return c.exec(query, args...) }
+
+// BatchInsert 批量插入，传入多组参数
+func (c *Client) BatchInsert(query string, params [][]any) error {
+	for _, args := range params {
+		if _, err := c.Insert(query, args...); err != nil {
+			return err
+		}
+	}
+	return nil
+}

code/base-project/base-go/pkg/utils/db/select.go

@@ -0,0 +1,63 @@
+package db
+
+import "database/sql"
+
+func (c *Client) Select(query string, args ...any) ([]map[string]any, error) {
+	rows, err := c.query(query, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer func(rows *sql.Rows) {
+		_ = rows.Close()
+	}(rows)
+
+	cols, _ := rows.Columns()
+	var result []map[string]any
+
+	for rows.Next() {
+		columns := make([]any, len(cols))
+		columnPointers := make([]any, len(cols))
+		for i := range columns {
+			columnPointers[i] = &columns[i]
+		}
+		if err := rows.Scan(columnPointers...); err != nil {
+			return nil, err
+		}
+		rowMap := make(map[string]any)
+		for i, colName := range cols {
+			rowMap[colName] = columns[i]
+		}
+		result = append(result, rowMap)
+	}
+	return result, nil
+}
+
+func (c *Client) SelectOne(query string, args ...any) (map[string]any, error) {
+	rows, err := c.query(query, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer func(rows *sql.Rows) {
+		_ = rows.Close()
+	}(rows)
+
+	if !rows.Next() {
+		return nil, sql.ErrNoRows
+	}
+
+	cols, _ := rows.Columns()
+	columns := make([]any, len(cols))
+	columnPointers := make([]any, len(cols))
+	for i := range columns {
+		columnPointers[i] = &columns[i]
+	}
+	if err := rows.Scan(columnPointers...); err != nil {
+		return nil, err
+	}
+
+	rowMap := make(map[string]any)
+	for i, colName := range cols {
+		rowMap[colName] = columns[i]
+	}
+	return rowMap, nil
+}

code/base-project/base-go/pkg/utils/db/update.go

@@ -0,0 +1,5 @@
+package db
+
+import "database/sql"
+
+func (c *Client) Update(query string, args ...any) (sql.Result, error) { return c.exec(query, args...) }