Go语言中构建轻量级ORM的策略与实践

package main

import (
    "database/sql"
    "fmt"
    "log"
    "time"

    _ "github.com/go-sql-driver/mysql" // 导入数据库驱动
)

// Person 结构体映射数据库中的 people 表
type Person struct {
    ID        int       `db:"pID"`        // 数据库列名为 pID
    FirstName string    `db:"fName"`      // 数据库列名为 fName
    LastName  string    `db:"lName"`      // 数据库列名为 lName
    Job       string    `db:"job"`
    Location  string    `db:"location"`
    CreatedAt time.Time `db:"created_at"` // 假设有一个 created_at 字段
}

// 假设的数据库连接函数
func connectDB() *sql.DB {
    // 实际应用中应从配置加载连接字符串
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/testdb?parseTime=true")
    if err != nil {
        log.Fatalf("无法连接到数据库: %v", err)
    }
    // 验证数据库连接
    if err = db.Ping(); err != nil {
        log.Fatalf("数据库连接失败: %v", err)
    }
    return db
}

// GetPersonByID 从数据库中获取指定ID的Person
func GetPersonByID(db *sql.DB, id int) (*Person, error) {
    person := &Person{}
    query := "SELECT pID, fName, lName, job, location, created_at FROM people WHERE pID = ?"
    row := db.QueryRow(query, id)

    err := row.Scan(&person.ID, &person.FirstName, &person.LastName, &person.Job, &person.Location, &person.CreatedAt)
    if err == sql.ErrNoRows {
        return nil, fmt.Errorf("未找到ID为 %d 的用户", id)
    } else if err != nil {
        return nil, fmt.Errorf("查询用户失败: %w", err)
    }
    return person, nil
}

// 示例调用
// db := connectDB()
// p, err := GetPersonByID(db, 1)
// if err != nil {
//     log.Println(err)
// } else {
//     fmt.Printf("获取到用户: %+v\n", p)
// }

// InsertPerson 将新的Person插入到数据库
func InsertPerson(db *sql.DB, person *Person) (int64, error) {
    query := "INSERT INTO people (fName, lName, job, location, created_at) VALUES (?, ?, ?, ?, ?)"
    result, err := db.Exec(query, person.FirstName, person.LastName, person.Job, person.Location, time.Now())
    if err != nil {
        return 0, fmt.Errorf("插入用户失败: %w", err)
    }
    lastID, err := result.LastInsertId()
    if err != nil {
        return 0, fmt.Errorf("获取最后插入ID失败: %w", err)
    }
    return lastID, nil
}

// 示例调用
// db := connectDB()
// newPerson := &Person{
//     FirstName: "Alice",
//     LastName:  "Smith",
//     Job:       "Engineer",
//     Location:  "New York",
// }
// id, err := InsertPerson(db, newPerson)
// if err != nil {
//     log.Println(err)
// } else {
//     fmt.Printf("插入新用户成功，ID: %d\n", id)
// }

// UpdatePerson 更新数据库中指定ID的Person
func UpdatePerson(db *sql.DB, person *Person) (int64, error) {
    query := "UPDATE people SET fName=?, lName=?, job=?, location=? WHERE pID=?"
    result, err := db.Exec(query, person.FirstName, person.LastName, person.Job, person.Location, person.ID)
    if err != nil {
        return 0, fmt.Errorf("更新用户失败: %w", err)
    }
    rowsAffected, err := result.RowsAffected()
    if err != nil {
        return 0, fmt.Errorf("获取受影响行数失败: %w", err)
    }
    return rowsAffected, nil
}

// 示例调用
// db := connectDB()
// existingPerson, err := GetPersonByID(db, 1) // 假设ID为1的用户存在
// if err == nil {
//     existingPerson.Job = "Senior Engineer"
//     rows, err := UpdatePerson(db, existingPerson)
//     if err != nil {
//         log.Println(err)
//     } else {
//         fmt.Printf("更新用户成功，影响行数: %d\n", rows)
//     }
// }

// DeletePerson 从数据库中删除指定ID的Person
func DeletePerson(db *sql.DB, id int) (int64, error) {
    query := "DELETE FROM people WHERE pID=?"
    result, err := db.Exec(query, id)
    if err != nil {
        return 0, fmt.Errorf("删除用户失败: %w", err)
    }
    rowsAffected, err := result.RowsAffected()
    if err != nil {
        return 0, fmt.Errorf("获取受影响行数失败: %w", err)
    }
    return rowsAffected, nil
}

// 示例调用
// db := connectDB()
// rows, err := DeletePerson(db, 2) // 假设ID为2的用户存在
// if err != nil {
//     log.Println(err)
// } else {
//     fmt.Printf("删除用户成功，影响行数: %d\n", rows)
// }

// TransferFunds 示例：一个简单的转账事务
func TransferFunds(db *sql.DB, fromAccountID, toAccountID int, amount float64) error {
    tx, err := db.Begin()
    if err != nil {
        return fmt.Errorf("开启事务失败: %w", err)
    }
    defer func() {
        if r := recover(); r != nil {
            tx.Rollback()
            panic(r) // re-throw panic after Rollback
        } else if err != nil {
            tx.Rollback() // error occurred, rollback
        } else {
            err = tx.Commit() // everything good, commit
        }
    }()

    // 1. 扣除转出方余额
    _, err = tx.Exec("UPDATE accounts SET balance = balance - ? WHERE id = ?", amount, fromAccountID)
    if err != nil {
        return fmt.Errorf("扣除转出方余额失败: %w", err)
    }

    // 2. 增加转入方余额
    _, err = tx.Exec("UPDATE accounts SET balance = balance + ? WHERE id = ?", amount, toAccountID)
    if err != nil {
        return fmt.Errorf("增加转入方余额失败: %w", err)
    }

    return err // 返回tx.Commit()的错误
}