Go语言使用xorm框架操作数据库

Xorm 是一个简单而强大的Go语言ORM(对象关系映射)库。它支持自动将结构体映射到数据库表,并提供了一系列便捷的API来执行CRUD(创建、读取、更新和删除)操作。

安装 Xorm

首先,你需要使用go get命令安装Xorm和数据库驱动(以MySQL为例):

go get xorm.io/xorm
go get github.com/go-sql-driver/mysql

初始化 Xorm 引擎

要使用Xorm,你需要初始化一个引擎(Engine)。引擎是Xorm操作数据库的主要入口。

import (
	"xorm.io/xorm"
	_ "github.com/go-sql-driver/mysql"
)

func initXORMEngine() (*xorm.Engine, error) {
	engine, err := xorm.NewEngine("mysql", "user:password@tcp(host:port)/dbname?charset=utf8")
	if err != nil {
		return nil, err
	}
	return engine, nil
}

在这里,我们通过xorm.NewEngine函数创建了一个新的引擎实例。mysql是数据库类型,后面的字符串是数据库的DSN(数据源名称),包括用户名、密码、地址、端口和数据库名称。

定义结构体映射

Xorm 使用结构体的字段来映射数据库表中的列。你需要定义与数据库表对应的结构体:

type User struct {
	Id   int64  `xorm:"'id' pk autoincr"`
	Name string `xorm:"'name'"`
	Age  int    `xorm:"'age'"`
}

// TableName 返回数据库中对应的表名
func (User) TableName() string {
	return "users"
}

在上述代码中,User结构体映射到users表。xorm标签用于指定列名,pk表示主键,autoincr表示自增。

CRUD 操作

以下是基本的CRUD操作示例:

创建(Create)
user := User{Name: "Alice", Age: 25}
affected, err := engine.Insert(&user)

使用Insert方法插入新记录。如果操作成功,affected会返回受影响的行数。

读取(Read)
var user User
has, err := engine.ID(1).Get(&user)

使用Get方法根据主键查询记录。如果找到记录,has将为true

更新(Update)
user.Name = "Bob"
affected, err := engine.ID(user.Id).Update(&user)

使用Update方法更新记录。你需要指定ID来更新特定的记录。

删除(Delete)
affected, err := engine.ID(1).Delete(&User{})

使用Delete方法删除记录。同样,你需要指定ID来删除特定的记录。

连接查询

如之前所述,可以使用Join方法进行连接查询。这里是一个查询用户和他们订单的示例:

type Order struct {
	OrderId     int64 `xorm:"'order_id'"`
	UserId      int64 `xorm:"'user_id'"`
	OrderNumber string
}

type UserWithOrders struct {
	User  `xorm:"extends"`
	Order `xorm:"extends"`
}

var userWithOrders []UserWithOrders
err := engine.Table("user").Join("INNER", "order", "user.id = order.user_id").Find(&userWithOrders)

在这个例子中,我们定义了Order结构体和UserWithOrders结构体,后者嵌套了UserOrder结构体,并使用了extends关键字。然后,我们使用Table方法选择了user表作为主表,并通过Join方法与order表进行内连接,最后使用Find方法获取所有匹配的记录。

示例二

// GetDigitalPersonsByUserID 函数,根据用户ID查询对应的数字人ID和名字
func GetDigitalPersonsByUserID(userID int64) ([]DigitalPerson, error) {
	var digitalPersons []DigitalPerson
	err := datasource.Engine.Table("digital_person").Join("INNER", "conversation", "digital_person.id = conversation.dp_id").
		Where("conversation.uid = ?", userID).
		Select("digital_person.id, digital_person.name,digital_person.gender").
		Find(&digitalPersons)
	if err != nil {
		return nil, err
	}
	return digitalPersons, nil
}

注意事项

  • 在实际开发中,你需要根据自己的数据库设计来定义结构体。
  • 确保在执行任何数据库操作之前,已经正确初始化并配置了Xorm引擎。
  • Xorm提供了丰富的标签系统来控制映射行为,例如自定义列名、索引、唯一约束等。
  • Xorm支持事务操作,你可以使用engine.NewSession()来开始一个新的会话,并使用session.Begin()session.Commit()session.Rollback()来控制事务。

以上就是Xorm的基本用法说明和示例。在实际应用中,你可能还需要根据具体需求使用Xorm的其他高级功能。

12-07 10:53