beego中ORM的应用

墨蓝 2022-11-22 12:51 161阅读 0赞

*  安装ORM，和mysql的驱动

> go get github.com/astaxie/beego/orm  
> go get github.com/go-sql-driver/mysql

*  ORM数据库配置连接，在`models`文件夹下新建`models.go`文件

package models
    
    import (
    	"github.com/astaxie/beego"
    	"github.com/astaxie/beego/orm"
    )
    
    func Init() { 
    	dbhost := beego.AppConfig.String("dbhost")
    	dbport := beego.AppConfig.String("dbport")
    	dbuser := beego.AppConfig.String("dbuser")
    	dbpassword := beego.AppConfig.String("dbpassword")
    	dbname := beego.AppConfig.String("dbname")
    	if dbport == "" { 
    		dbport = "3306"
    	}
    	dsn := dbuser + ":" + dbpassword + "@tcp(" + dbhost + ":" + dbport + ")/" + dbname + "?charset=utf8&loc=Asia%2FShanghai"
    	orm.RegisterDataBase("default", "mysql", dsn)
    	orm.RegisterModel(new(User), new(Article))
    }
    
    //返回带前缀的表名
    func TableName(str string) string { 
    	return beego.AppConfig.String("dbprefix") + str
    }

*  `Init()`中读取`conf/app.conf`配置文件并连接数据库
 *  在models文件夹下新建模型`User`和`Article`并注册模型

package models
    
    type User struct { 
    	Id   int
    	Name string
    	Age  int
    }
    
    func (m *User) TableName() string { 
    	return TableName("user")
    }

package models
    
    type Article struct { 
    	Id      int
    	Title   string
    	Content string
    }
    
    func (m *Article) TableName() string { 
    	return TableName("article")
    }

*  在入口`main.go`中进行初始化

package main
    
    import (
    	_ "github.com/superkingM/beego_one/routers"
    	"github.com/astaxie/beego"
    	_ "github.com/go-sql-driver/mysql"
    	"github.com/superkingM/beego_one/models"
    )
    
    func init() { 
    	models.Init()
    }
    
    func main() { 
    	beego.Run()
    }

*  在控制器中进行CURD
 *  新增

package controllers
    
    import (
    	"github.com/astaxie/beego"
    	"github.com/astaxie/beego/orm"
    	"github.com/superkingM/beego_one/models"
    )
    
    type MainController struct { 
    	beego.Controller
    
    }
    
    func (c *MainController) Get() { 
    	
    	o :=orm.NewOrm()
    	var user models.User
    	user.Name="张三"
    	user.Age=38
    
    	id,err :=o.Insert(&user)
    	if err == nil { 
    		c.Data["json"]=id
    		c.ServeJSON()
    	}
    
    }

o :=orm.NewOrm()
    	users :=[]models.User{ 
    		{ Name:"小明",Age:26},
    		{ Name:"小红",Age:27},
    	}
    	o.InsertMulti(2,users)

*  更新

o := orm.NewOrm()
    user := User{ Id: 1}
    if o.Read(&user) == nil { 
        user.Name = "MyName"
        o.Update(&user)
    }

*  删除

o :=orm.NewOrm()
    	user := models.User{ Id:1}
    	o.Delete(&user)