您现在的位置是：网站首页> 编程资料编程资料

Beego中ORM操作各类数据库连接方式详细示例_Golang_

2023-05-26 457人已围观

简介 Beego中ORM操作各类数据库连接方式详细示例_Golang_

beego中各类数据库连接方式

beego 框架是优秀得go REST API开发框架。下面针对beego中各类数据库连接操作做一个总结。

orm连接方式

beego中的orm操作支持三种数据库：mysql,sqlite3,postgersql。三种数据库的驱动如下：

import ( _ "github.com/go-sql-driver/mysql" _ "github.com/lib/pq" _ "github.com/mattn/go-sqlite3" )

对于以上三类数据库的使用，最好使用orm进行操作，因为简单的增删查改已经实现了通用的接口封装。无需再根据每一个模型，单独的实现一套增删查改。另外，orm也支持原生sql查询，复杂的sql操作可以直接使用orm.Raw()进行原生查询，简单快捷。

1.1 orm使用方式

a. 注册数据库驱动程序

// 参数1 driverName // 参数2 数据库类型 // 这个用来设置 driverName 对应的数据库类型 // mysql / sqlite3 / postgres 这三种是默认已经注册过的，所以可以无需设置 orm.RegisterDriver("mysql", orm.DRMySQL)

b.注册数据库

ORM 必须注册一个别名为 default 的数据库，作为默认使用。
ORM 使用 golang 自己的连接池

// 参数1 数据库的别名，用来在 ORM 中切换数据库使用 // 参数2 driverName // 参数3 对应的链接字符串 orm.RegisterDataBase("default", "mysql", "root:root@/orm_test?charset=utf8") // 参数4(可选) 设置最大空闲连接 // 参数5(可选) 设置最大数据库连接 (go >= 1.2) maxIdle := 30 maxConn := 30 orm.RegisterDataBase("default", "mysql", "root:root@/orm_test?charset=utf8", maxIdle, maxConn)

c. 注册模型

orm.RegisterModel(new(User), new(Profile), new(Post))

自此，就可以愉快的使用orm提供的接口进行数据库操作了。

type Ormer interface { Read(interface{}, …string) error ReadOrCreate(interface{}, string, …string) (bool, int64, error) Insert(interface{}) (int64, error) InsertMulti(int, interface{}) (int64, error) Update(interface{}, …string) (int64, error) Delete(interface{}) (int64, error) LoadRelated(interface{}, string, …interface{}) (int64, error) QueryM2M(interface{}, string) QueryM2Mer QueryTable(interface{}) QuerySeter Using(string) error Begin() error Commit() error Rollback() error Raw(string, …interface{}) RawSeter Driver() Driver }

1.2 操作示例

a. orm连接mysql

import ( "github.com/astaxie/beego/orm" _ "github.com/go-sql-driver/mysql" ) func init() { orm.RegisterDriver("mysql", orm.DRMySQL) orm.RegisterDataBase("default", "mysql", "root:root@tcp(192.168.1.1:3306)/ming?charset=utf8") orm.RegisterModel(new(User)) orm.SetMaxIdleConns("default", 10) orm.SetMaxOpenConns("default", 100) } func Create(param interface{}) (int64, error) { return orm.NewOrm().Insert(param) } func Update(param interface{}, fields ...string) (int64, error) { return orm.NewOrm().Update(param, fields...) } func Delete(param interface{}, cols ...string) (int64, error) { return orm.NewOrm().Delete(param, cols...) } func Read(md interface{}, cols ...string) error { return orm.NewOrm().Read(md, cols...) }

b. orm连接sqlite3

import ( "github.com/astaxie/beego/orm" _ "github.com/mattn/go-sqlite3" ) func init() { orm.RegisterDriver("sqlite3", orm.DRSqlite) orm.RegisterDataBase("default", "sqlite3", "conf/sqlite_test.db") orm.SetMaxIdleConns("default", 50) orm.SetMaxOpenConns("default", 200) //设置数据库时区 //orm.DefaultTimeLoc = time.UTC orm.RegisterModel(new(User)) } func Create(param interface{}) (int64, error) { return orm.NewOrm().Insert(param) } func Update(param interface{}, fields ...string) (int64, error) { return orm.NewOrm().Update(param, fields...) } func Delete(param interface{}, cols ...string) (int64, error) { return orm.NewOrm().Delete(param, cols...) } func Read(md interface{}, cols ...string) error { return orm.NewOrm().Read(md, cols...) }

c. orm连接 postgresql

golang orm可以连接postgres，但是好像不支持数据库中的schema.

import ( "github.com/astaxie/beego/orm" _ "github.com/lib/pq" ) func init() { orm.RegisterDriver("postgres", orm.DRPostgres) orm.RegisterDataBase("default", "postgres", "host=192.168.1.1 port=5432 user=root password=root dbname=test sslmode=disable") //orm.RegisterModelWithPrefix("schema_name", new(PmPark)) orm.RegisterModel(new(PmPark)) orm.SetMaxIdleConns("default", 10) orm.SetMaxOpenConns("default", 50) }

1.3非orm连接方式

使用非orm连接方式，除了mysql,sqlite3,postgresql外，其他的如sqlserver,mongodb,redis等等，都有自己的连接方式。

a. mysql

import ( "fmt" "database/sql" _ "github.com/go-sql-driver/mysql" ) func getMysqlDB() (*sql.DB, error) { connectString := "root:123456@tcp(localhost:3306)/test?charset=utf8" db, err := sql.Open("mysql", connectString) if err != nil { return nil, err } return db, nil }

b. sqlite3

import ( "database/sql" _ "github.com/mattn/go-sqlite3" ) func GetDBConn() (*sql.DB, error) { return sql.Open("sqlite3", "./data/location.db") }

c. postgresql

import ( "database/sql" "errors" "fmt" "time" "github.com/astaxie/beego" _ "github.com/lib/pq" ) var ( host string = "" port int user string = "" password string = "" dbname string = "" max_conn int = 40 idle_conn int = 10 postgreConn *sql.DB //全局sql连接,已实现连接池,所以可以只创建一个实例 DB_CONN_ERROR error ) func init() { host = beego.AppConfig.String("postgres_host") port, _ = beego.AppConfig.Int("postgres_port") user = beego.AppConfig.String("postgres_user") password = beego.AppConfig.String("postgres_password") dbname = beego.AppConfig.String("postgres_dbname") max_conn = beego.AppConfig.DefaultInt("postgres_max_conn", 50) idle_conn = beego.AppConfig.DefaultInt("postgres_idle_conn", 10) DB_CONN_ERROR = errors.New("数据库连接失败") } func GetPostgresSession() *sql.DB { if postgreConn == nil { psqlInfo := fmt.Sprintf(`host=%s port=%d user=%s password=%s dbname=%s sslmode=disable`, host, port, user, password, dbname) db, err := sql.Open("postgres", psqlInfo) if err != nil { return nil } db.SetConnMaxLifetime(30 * time.Minute) db.SetMaxOpenConns(max_conn) db.SetMaxIdleConns(idle_conn) // err = db.Ping() // if err != nil { // return nil // } postgreConn = db } return postgreConn }

d. mongodb

import ( "errors" "time" "github.com/astaxie/beego" "gopkg.in/mgo.v2" ) // 连接mongodb数据库 var ( MongodbAddr string = "" //mongodb数据库地址 MongodbName string = "" //mongodb数据名称 MongodbUser string = "" //mongodb用户名 MongodbPasswd string = "" //mongodb密码 ) var ( mongosession *mgo.Session ) func init() { MongodbAddr = beego.AppConfig.String("mongodb_addr") MongodbName = beego.AppConfig.String("mongodb_name") MongodbUser = beego.AppConfig.String("mongodb_username") MongodbPasswd = beego.AppConfig.String("mongodb_passwd") } func GetMongoSession() *mgo.Session { if mongosession == nil { var err error if MongodbUser == "" || MongodbPasswd == "" { mongosession, err = mgo.Dial(MongodbAddr) } else { dialInfo := &mgo.DialInfo{ Addrs: string{MongodbAddr}, Direct: false, Timeout: time.Second * 30, Database: MongodbName, Source: "admin", Username: MongodbUser, Password: MongodbPasswd, PoolLimit: 4096, // Session.SetPoolLimit } mongosession, err = mgo.DialWithInfo(dialInfo) } if err != nil { return nil } } return mongosession.Clone() } func WithMongoCollection(collectionName string, s func(*mgo.Collection) error) error { session := GetMongoSession() if session == nil { return errors.New("获取mongodb连接失败") } defer session.Close() c := session.DB(MongodbName).C(collectionName) return s(c) }

e.sqlserver

import ( "database/sql" "time" "fmt" "github.com/astaxie/beego" _ "github.com/denisenkom/go-mssqldb" ) const ( CONN_LIVE_TIME = 24 //连接使用时间 小时 ) var ( db *sql.DB = nil //全局数据库连接 ) func init() { host := beego.AppConfig.String("yr_host") port, err := beego.AppConfig.Int("yr_port") if err != nil { port = 1433 } user := beego.AppConfig.String("user") password := beego.AppConfig.String("password") dbName := beego.AppConfig.String("name") connString := fmt.Sprintf("server=%s;port%d;database=%s;user id=%s;password=%s", host, port, dbName, user, password) db, err = sql.Open("mssql", connString) if err != nil { return } db.SetMaxOpenConns(200) db.SetMaxIdleConns(50) db.SetConnMaxLifetime(time.Duration(CONN_LIVE_TIME) * time.Hour) }