gorm-数据库操作

目录

• 1.1. 连接数据库
  • 1.1.1. MySQL
  • 1.1.2. PostgreSQL
  • 1.1.3. Sqlite3
  • 1.1.4. 不支持的数据库
• 1.2. 迁移
  • 1.2.1. 自动迁移
  • 1.2.2. 检查表是否存在
  • 1.2.3. 创建表
  • 1.2.4. 删除表
  • 1.2.5. 修改列
  • 1.2.6. 删除列
  • 1.2.7. 添加外键
  • 1.2.8. 索引

1.1. 连接数据库

要连接到数据库首先要导入驱动程序。例如

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

为了方便记住导入路径，GORM包装了一些驱动。

import _ "github.com/jinzhu/gorm/dialects/mysql"
// import _ "github.com/jinzhu/gorm/dialects/postgres"
// import _ "github.com/jinzhu/gorm/dialects/sqlite"
// import _ "github.com/jinzhu/gorm/dialects/mssql"

1.1.1. MySQL

注：为了处理time.Time，您需要包括parseTime作为参数。 （更多支持的参数）

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
)

var (
	bsDb *gorm.DB
)

func main() {
	db, err := gorm.Open("mysql", "root:123456@tcp(127.0.0.1:3307)/bs?charset=utf8")
	defer db.Close()
	if err != nil {
		fmt.Println("===========")
		fmt.Println(err)
		return
	}
	sqlDb := db.DB()
	sqlDb.SetMaxOpenConns(10 / 2)
	sqlDb.SetMaxIdleConns(5 / 2)
	bsDb = db
	info := Info{}
	bsDb.Model(Info{}).Where("id = ?", 1).Save(&info)
	fmt.Println(info)
}

type Info struct {
	Id   int64  `db:"id"`
	Name string `db:"name"`
}

func (Info) TableName() string {
	return "user"
}

1.1.2. PostgreSQL

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/postgres"
)

func main() {
  db, err := gorm.Open("postgres", "host=myhost user=gorm dbname=gorm sslmode=disable password=mypassword")
  defer db.Close()
}

1.1.3. Sqlite3

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
)

func main() {
  db, err := gorm.Open("sqlite3", "/tmp/gorm.db")
  defer db.Close()
}

1.1.4. 不支持的数据库

GORM正式支持上述的数据库，如果您使用的是不受支持的数据库请按照下面的连接编写对应数据库支持文件。 https://github.com/jinzhu/gorm/blob/master/dialect.go

1.2. 迁移

1.2.1. 自动迁移

自动迁移模式将保持更新到最新。

警告：自动迁移仅仅会创建表，缺少列和索引，并且不会改变现有列的类型或删除未使用的列以保护数据。

db.AutoMigrate(&User{})

db.AutoMigrate(&User{}, &Product{}, &Order{})

// 创建表时添加表后缀
db.Set("gorm:table_options", "ENGINE=InnoDB").AutoMigrate(&User{})

1.2.2. 检查表是否存在

// 检查模型`User`表是否存在
db.HasTable(&User{})

// 检查表`users`是否存在
db.HasTable("users")

1.2.3. 创建表

// 为模型`User`创建表
db.CreateTable(&User{})

// 创建表`users'时将“ENGINE = InnoDB”附加到SQL语句
db.Set("gorm:table_options", "ENGINE=InnoDB").CreateTable(&User{})

1.2.4. 删除表

// 删除模型`User`的表
db.DropTable(&User{})

// 删除表`users`
db.DropTable("users")

// 删除模型`User`的表和表`products`
db.DropTableIfExists(&User{}, "products")

1.2.5. 修改列

修改列的类型为给定值

// 修改模型`User`的description列的数据类型为`text`
db.Model(&User{}).ModifyColumn("description", "text")

1.2.6. 删除列

// 删除模型`User`的description列
db.Model(&User{}).DropColumn("description")

1.2.7. 添加外键

// 添加主键
// 1st param : 外键字段
// 2nd param : 外键表(字段)
// 3rd param : ONDELETE
// 4th param : ONUPDATE
db.Model(&User{}).AddForeignKey("city_id", "cities(id)", "RESTRICT", "RESTRICT")

1.2.8. 索引

// 为`name`列添加索引`idx_user_name`
db.Model(&User{}).AddIndex("idx_user_name", "name")

// 为`name`, `age`列添加索引`idx_user_name_age`
db.Model(&User{}).AddIndex("idx_user_name_age", "name", "age")

// 添加唯一索引
db.Model(&User{}).AddUniqueIndex("idx_user_name", "name")

// 为多列添加唯一索引
db.Model(&User{}).AddUniqueIndex("idx_user_name_age", "name", "age")

// 删除索引
db.Model(&User{}).RemoveIndex("idx_user_name")