Golang数据库工具包与Atlas CLI使用指南

常用Golang数据库工具包

1. database/sql标准库

Go语言内置的标准数据库接口，支持多种数据库驱动。

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

func main() {
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/dbname")
    if err != nil {
        panic(err.Error())
    }
    defer db.Close()

    // 查询示例
    rows, err := db.Query("SELECT id, name FROM users")
    if err != nil {
        panic(err.Error())
    }
    defer rows.Close()
    
    for rows.Next() {
        var id int
        var name string
        err = rows.Scan(&id, &name)
        // 处理数据
    }
}

2. GORM (ORM框架)

流行的ORM库，支持多种数据库。

import (
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

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

func main() {
    dsn := "user:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }
    
    // 自动迁移
    db.AutoMigrate(&User{})
    
    // 创建记录
    db.Create(&User{Name: "Alice", Age: 25})
    
    // 查询
    var user User
    db.First(&user, 1) // 查找ID为1的用户
}

3. sqlx

扩展标准库，提供更方便的数据绑定功能。

import (
    "github.com/jmoiron/sqlx"
    _ "github.com/go-sql-driver/mysql"
)

type User struct {
    ID   int    `db:"id"`
    Name string `db:"name"`
}

func main() {
    db, err := sqlx.Connect("mysql", "user:password@tcp(127.0.0.1:3306)/dbname")
    if err != nil {
        panic(err)
    }
    
    // 查询到结构体切片
    users := []User{}
    err = db.Select(&users, "SELECT id, name FROM users")
    if err != nil {
        panic(err)
    }
}

Atlas CLI使用指南

Atlas是一个现代化的数据库迁移和Schema管理工具，支持多种数据库。

安装Atlas

# 使用Homebrew (macOS/Linux)
brew install ariga/tap/atlas

# 使用Go安装
go get -u ariga.io/atlas/cmd/atlas

基本使用

1. 初始化项目

mkdir myproject
cd myproject
atlas init --url mysql://user:pass[@localhost](/user/localhost):3306/dbname

1. 创建迁移文件

atlas migrate diff create_users_table \
  --dir "file://migrations" \
  --to "file://schema.hcl" \
  --dev-url "docker://mysql/8/dev"

1. 应用迁移

atlas migrate apply \
  --url mysql://user:pass[@localhost](/user/localhost):3306/dbname \
  --dir file://migrations

HCL Schema定义示例

schema "myapp" {
}

table "users" {
  schema = schema.myapp
  column "id" {
    type = int
    auto_increment = true
  }
  column "name" {
    type = varchar(100)
    null = false
  }
  column "email" {
    type = varchar(255)
    unique = true
  }
  primary_key {
    columns = [column.id]
  }
}

与Golang集成

Atlas提供了Go SDK，可以集成到应用中：

import (
    "context"
    "log"
    
    "ariga.io/atlas/sql/migrate"
    "ariga.io/atlas/sql/mysql"
    "ariga.io/atlas/sql/sqlclient"
)

func main() {
    // 创建客户端
    client, err := sqlclient.Open(context.Background(), "mysql://user:pass[@localhost](/user/localhost):3306/dbname")
    if err != nil {
        log.Fatal(err)
    }
    defer client.Close()
    
    // 创建迁移目录
    dir, err := migrate.NewLocalDir("migrations")
    if err != nil {
        log.Fatal(err)
    }
    
    // 应用迁移
    err = client.ApplyChanges(context.Background(), nil, dir)
    if err != nil {
        log.Fatal(err)
    }
}

高级功能

1. 版本控制集成

   atlas migrate lint --dir file://migrations --dev-url docker://mysql/8/dev
   

2. Schema检查

   atlas schema inspect -u mysql://user:pass[@localhost](/user/localhost):3306/dbname > schema.hcl
   

3. CI/CD集成

   atlas migrate validate --dir file://migrations --dev-url docker://mysql/8/dev
   

Atlas提供了强大的数据库管理功能，特别适合需要严格Schema管理和团队协作的项目。