golang之gorm使用mysql实例

我不是女神ヾ 2023-01-24 02:09 154阅读 0赞

一、数据库链接  github.com/jinzhu/gorm
    package utils
    
    import (
       "fmt"
       _ "github.com/go-sql-driver/mysql"
       "github.com/jinzhu/gorm"
    )
    
    type mysqlModel struct {
       Host        string `yaml:"host"`
       Port        int    `yaml:"port"`
       User        string `yaml:"user"`
       Password    string `yaml:"password"`
       Database    string `yaml:"database"`
       MaxIdLeConn int    `yaml:"maxidleconn"`
       MaxOpenConn int    `yaml:"maxopenconn"`
       Debug       bool   `yaml:"debug"`
       IsPlural    bool   `yaml:"isplural"`
       TablePrefix string `yaml:"tableprefix"`
    }
    
    var DB *gorm.DB
    var mysqlConfig = mysqlModel{
       Host:        "127.0.0.1",
       Port:        3308,
       User:        "root",
       Password:    "root123456",
       Database:    "data_center",
       MaxIdLeConn: 10,
       MaxOpenConn: 100,
       Debug:       false,
       IsPlural:    true,
       TablePrefix: "data_center_",
    }
    
    func MysqlInit() error {
       var err error
       DbUrl := fmt.Sprintf("%s:%s@(%s:%d)/%s?charset=utf8",
          mysqlConfig.User,
          mysqlConfig.Password,
          mysqlConfig.Host,
          mysqlConfig.Port,
          mysqlConfig.Database)
       DB, err = gorm.Open("mysql", DbUrl)
       if err != nil {
          fmt.Printf("mysql connect error %v", err)
       }
       if DB.Error != nil {
          fmt.Printf("database error %v", DB.Error)
       }
       //最大打开的连接数
       DB.DB().SetMaxOpenConns(mysqlConfig.MaxOpenConn)
       //最大空闲连接数
       DB.DB().SetMaxIdleConns(mysqlConfig.MaxIdLeConn)
       //允许表名复数
       DB.SingularTable(mysqlConfig.IsPlural)
       DB.LogMode(mysqlConfig.Debug)
       return err
    }

//表结构体

type User struct {
       ID int
       Name string
       Age int
       AddTime int 
    }

//判断表不存在自动创建表
    func init() {
        if !utils.DB.HasTable(&User{}) {
           utils.DB.AutoMigrate(&User{})
       }
    }

/*查询所有数据*/
    user := []User{}
    utils.DB.Find(&user)
    /*
    查询所有数据 age=12
    user := []User{}
    utils.DB.Find(&user, "age = ?", 12)
    */
    /*
    查询一条数据 id=2
    user := User{ID:2}
    utils.DB.First(&user)
    */ 
    /*查询一条name=plum的数据
    user := User{}
    utils.DB.First(&user, "name = ?", "plum")
    */
    //新增
    user := User{
       Name:"张三",
       Age:22,
       AddTime:int(time.Now().Unix()),
    }
    utils.DB.Create(&user)
    //修改数据
    user := User{ID:1}
    utils.DB.First(&user)
    user.Name = username
    user.Age = age
    utils.DB.Save(&user)
    //删除数据
    user := User{ID:1}
    utils.DB.Delete(&user)
    //执行原生sql语句
    sql := `UPDATE user SET age = 66 WHERE id = 1;`
    err := utils.DB.Exec(sql).Error
    if err != nil {
        fmt.Print("修改消息失败：", err.Error())
    }
    //查询总条数  Count 返回int
    total :=0
    utils.DB.Table("user").Count(&total)

二、数据库链接  gorm.io/gorm  使用

package Database
    
    import (
       "fmt"
       "gorm.io/driver/mysql"
       "gorm.io/gorm"
       "gorm.io/gorm/schema"
       "time"
       "ruiyi/plum/gin/Common"
    )
    
    var DB *gorm.DB
    
    func MysqlInit() (err error) {
       mysqlConfig := Common.MysqlInfo
       DB, err = gorm.Open(mysql.New(mysql.Config{
          DSN: fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local", mysqlConfig.User,
             mysqlConfig.Password, mysqlConfig.Host, mysqlConfig.Port, mysqlConfig.Database),
          DefaultStringSize:         256,   // string 类型字段的默认长度
          DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
          DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
          DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
          SkipInitializeWithVersion: false, // 根据版本自动配置
       }), &gorm.Config{NamingStrategy: schema.NamingStrategy{
          SingularTable: mysqlConfig.IsPlural, //允许表名复数  `User` 的表名应该是 `t_users` 如果该值为true则表名为`t_user`
          //TablePrefix:   mysqlConfig.TablePrefix, // 表名前缀
       }})
       sqlDB, err := DB.DB()
       //设置空闲连接池中连接的最大数量
       sqlDB.SetMaxIdleConns(mysqlConfig.MaxIdLeConn)
       //设置打开数据库连接的最大数量
       sqlDB.SetMaxOpenConns(mysqlConfig.MaxOpenConn)
       //设置了连接可复用的最大时间
       sqlDB.SetConnMaxLifetime(time.Hour)
       //defer sqlDB.Close()
       return err
    }

//数据库操作基本与github.com/jinzhu/gorm 一致，区别：count返回的是int64