Beego中ORM操作各类数据库连接方式详细示例
作者：Jeff的技术栈 发布时间：2024-01-20 08:07:07
标签：Beego,ORM,数据库,连接
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)
}
f.redis
import (
"time"
"github.com/astaxie/beego"
"github.com/gomodule/redigo/redis"
)
var (
db_addr string = ""
db_password string = ""
redisPool *redis.Pool //redis连接池
)
func init() {
db_addr = beego.AppConfig.String("redis_addr")
db_password = beego.AppConfig.String("redis_password")
}
//获取Redis连接池
func newRedisPool(server, password string) (*redis.Pool, error) {
var err error
return &redis.Pool{
MaxIdle: 32,
IdleTimeout: 180 * time.Second,
Dial: func() (redis.Conn, error) {
var c redis.Conn
c, err = redis.Dial("tcp", server)
if err != nil {
return nil, err
}
if password != "" {
if _, err = c.Do("AUTH", password); err != nil {
c.Close()
return nil, err
}
}
return c, err
},
TestOnBorrow: func(c redis.Conn, t time.Time) error {
_, err := c.Do("PING")
return err
},
}, err
}
/*
获取redis数据库连接
*/
func GetRedisConnection() (redis.Conn, error) {
if redisPool == nil {
var err error
redisPool, err = newRedisPool(db_addr, db_password)
if err != nil {
return nil, err
}
}
return redisPool.Get(), nil
}
来源：https://www.cnblogs.com/guyouyin123/p/14041582.html
0
投稿猜你喜欢
- 使用数据库的过程中，由于断电或其他原因,有可能导致数据库出现一些小错误，比如检索某些表特别慢,查询不到符合条件的数据等。出现这些情况的原因,
- 我就废话不多说了，大家还是直接看代码吧~func main() { fmt.Println(exponent (5,3))}//a的n次方/
- 在编程过程中，多了解语言周边的一些知识，以及一些技巧，可以让你加速成为一个优秀的程序员。对于Python程序员，你需要注意一下本文所提到的这
- 在日常的工作中,有时候会有这样的需求,需要一个常驻任务,持续的监听一个目录下文件的变化,对此作出回应.pyinotify就是这样的一个pyt
- 本文实例讲述了golang实现http服务器处理静态文件的方法。分享给大家供大家参考，具体如下：新版本更精简:package mainimp
- Dethe Elza (delza@livingcode.org), 高级技术架构师, Blast Radius &n
- <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN&
- 1，js取得IP地址的方法一<script src="http://pv.sohu.com/cityjson?ie=utf-
- 在使用Python做socket编程时，由于需要使用阻塞（默认）的方式来读取数据流，此时对于数据的结束每次都需要自己处理，太麻烦。并且网上也
- 如果你是我的长期读者，那么你应该知道我在寻找一个完美备份程序，最后我写了一个基于bup的我自己的加密层。在写encbup的时候，我对仅仅恢复
- 前言Python 这门语言最大的优点之一就是语法简洁，好的代码就像伪代码一样，干净、整洁、一目了然。但有时候我们写代码，特别是 Python
前言： 在项目开发中，一些业务表字段经常使用日期和时间类型，而且后续还会牵涉到这类字段的查询。关于日期及时间的查询等各类需求也很多，本篇文章- 本文实例讲述了Python3中正则模块re.compile、re.match及re.search函数用法。分享给大家供大家参考，具体如下：r
- 一、数据的概括性度量1、统计学概括：统计学是应用数学的一个分支，主要通过利用概率论建立数学模型，收集所观察系统的数据，进行量化的分析、总结，
- PyMongo是什么PyMongo是驱动程序，使python程序能够使用Mongodb数据库，使用python编写而成．安装环境
- 我们在很多网站上都可以看到用户注册使用电子邮件激活或启用的方式。也就是说，用户在注册后填写正确的电子邮件地址，接着网站会发送一封启用电子邮件
- MySQL根据配置文件会限制Server接受的数据包大小。有时候大的插入和更新会受 max_allowed_packet 参数限制，导致写入
- 本文实例讲述了Python协程 yield与协程greenlet简单用法。分享给大家供大家参考，具体如下：协程协程，又称微线程，纤程。英文名
- 需求说明：将单个或者多个Excel文件数据进行去重操作，去重的列可以通过自定义制定。开始源码说明之前，先说明一下工具的使用过程。1、准备需要
- 使用 pd.Series把dataframe转成Seriests = pd.Series(df['Value'].value
