Golang中的SQL操作：ORM框架和原生SQL语句

Golang是一门强大的编程语言，可以用于开发各种类型的应用程序。在很多场景下，需要使用SQL来操作关系型数据库。本文将介绍在Golang中使用ORM框架和原生SQL语句进行SQL操作的方法。

ORM框架

ORM（Object-Relational Mapping）是一种将对象和关系型数据库进行映射的技术，它可以将数据库操作转换为对象操作。在Golang中，有多个ORM框架可以使用，比如GORM、XORM和Beego ORM等。

在本文中，我们将以GORM为例来介绍ORM框架的使用。

安装GORM

GORM可以通过go get命令进行安装：

```go
go get -u gorm.io/gorm
```

连接数据库

在使用GORM之前，需要先连接数据库。通常可以在main函数中调用gorm.Open方法来连接数据库：

```go
import "gorm.io/gorm"

func main() {
  db, err := gorm.Open(mysql.Open("myuser:mypassword@tcp(127.0.0.1:3306)/mydb"), &gorm.Config{})
  if err != nil {
    panic("failed to connect database")
  }

  // ...
}
```

在上面的代码中，我们使用了mysql.Open方法连接MySQL数据库。其中，myuser是数据库用户名，mypassword是数据库密码，127.0.0.1:3306是数据库地址和端口号，mydb是数据库名。如果连接数据库失败，会抛出一个panic异常。

定义模型

定义模型是使用ORM框架进行数据库操作的关键。在GORM中，可以通过定义结构体来定义模型。在结构体中，可以定义属性和方法，属性可以与数据库中的表字段进行映射，方法可以用于实现业务逻辑。

下面是一个使用GORM定义模型的示例：

```go
import "gorm.io/gorm"

type User struct {
  gorm.Model
  Name string
  Age  int
}
```

在上面的代码中，我们定义了一个名为User的结构体，该结构体继承了gorm.Model结构体，这个结构体内置了一些字段，包括ID、CreatedAt、UpdatedAt和DeletedAt。Name和Age属性可以与数据库中的表字段进行映射。

创建表

定义模型后，需要使用AutoMigrate方法来创建表：

```go
db.AutoMigrate(&User{})
```

在上面的代码中，我们使用了AutoMigrate方法来创建User表。

插入数据

使用GORM插入数据很简单。只需要创建一个结构体对象，并使用Create方法即可：

```go
user := User{Name: "Tom", Age: 18}
db.Create(&user)
```

在上面的代码中，我们创建了一个名为Tom的User对象，并使用Create方法将其插入到数据库中。

查询数据

使用GORM查询数据也很容易。只需要使用Find方法即可：

```go
var users []User
db.Find(&users)
```

在上面的代码中，我们使用了Find方法来查询所有User对象，并将结果存储在一个名为users的切片中。

更新数据

使用GORM更新数据也很简单，只需要使用Updates或Update方法即可：

```go
// 方法1：使用Updates
user := User{Name: "Tom", Age: 18}
db.Create(&user)
db.Model(&user).Updates(User{Name: "Jerry"})

// 方法2：使用Update
db.Model(&User{}).Where("age = ?", 18).Update("name", "Jerry")
```

在上面的代码中，我们使用了Updates和Update方法来更新数据。Updates方法可以直接传递一个结构体对象，而Update方法需要传递一个SQL条件和要更新的属性。

删除数据

使用GORM删除数据也很容易，只需要使用Delete方法即可：

```go
db.Delete(&User{}, id)
```

在上面的代码中，我们使用了Delete方法来删除一个名为id的User对象。

原生SQL语句

虽然ORM框架可以帮助我们方便地操作数据库，但有些场景下我们需要使用原生SQL语句。在Golang中，可以使用database/sql包来执行原生SQL语句。

连接数据库

在使用database/sql包之前，需要先连接数据库。可以使用sql.Open方法连接数据库：

```go
import "database/sql"

func main() {
  db, err := sql.Open("mysql", "myuser:mypassword@tcp(127.0.0.1:3306)/mydb")
  if err != nil {
    panic("failed to connect database")
  }

  // ...
}
```

在上面的代码中，我们使用了sql.Open方法连接MySQL数据库。其中，myuser是数据库用户名，mypassword是数据库密码，127.0.0.1:3306是数据库地址和端口号，mydb是数据库名。如果连接数据库失败，会抛出一个panic异常。

查询数据

使用database/sql包查询数据的方法如下：

```go
rows, err := db.Query("SELECT id, name, age FROM users")
if err != nil {
  panic(err)
}
defer rows.Close()

for rows.Next() {
  var id int
  var name string
  var age int
  if err := rows.Scan(&id, &name, &age); err != nil {
    panic(err)
  }
  fmt.Println(id, name, age)
}
```

在上面的代码中，我们使用了Query方法执行SQL语句，并使用Scan方法将查询结果映射到变量中。需要注意的是，我们需要在查询完毕后调用rows.Close方法关闭查询结果集。

插入数据

使用database/sql包插入数据的方法如下：

```go
stmt, err := db.Prepare("INSERT INTO users(name, age) VALUES(?, ?)")
if err != nil {
  panic(err)
}
defer stmt.Close()

_, err = stmt.Exec("Tom", 18)
if err != nil {
  panic(err)
}
```

在上面的代码中，我们使用了Prepare方法创建了一个预处理语句，并使用Exec方法来执行插入操作。需要注意的是，我们需要在插入完毕后调用stmt.Close方法关闭预处理语句。

更新数据

使用database/sql包更新数据的方法如下：

```go
stmt, err := db.Prepare("UPDATE users SET name = ? WHERE id = ?")
if err != nil {
  panic(err)
}
defer stmt.Close()

_, err = stmt.Exec("Jerry", 1)
if err != nil {
  panic(err)
}
```

在上面的代码中，我们使用了Prepare方法创建了一个预处理语句，并使用Exec方法来执行更新操作。需要注意的是，我们需要在更新完毕后调用stmt.Close方法关闭预处理语句。

删除数据

使用database/sql包删除数据的方法如下：

```go
stmt, err := db.Prepare("DELETE FROM users WHERE id = ?")
if err != nil {
  panic(err)
}
defer stmt.Close()

_, err = stmt.Exec(1)
if err != nil {
  panic(err)
}
```

在上面的代码中，我们使用了Prepare方法创建了一个预处理语句，并使用Exec方法来执行删除操作。需要注意的是，我们需要在删除完毕后调用stmt.Close方法关闭预处理语句。

总结

在本文中，我们介绍了在Golang中使用ORM框架和原生SQL语句进行SQL操作的方法。使用ORM框架可以方便地进行数据库操作，而使用原生SQL语句可以更灵活地操作数据库。在实际开发中，需要根据场景选择合适的方式进行数据库操作。