beego中各类数据库连接方式

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

1. 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:[email protected]/orm_test?charset=utf8")

// 参数4(可选)  设置最大空闲连接
// 参数5(可选)  设置最大数据库连接 (go >= 1.2)
maxIdle := 30
maxConn := 30
orm.RegisterDataBase("default", "mysql", "root:[email protected]/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:[email protected](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)
}

2. 非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:[email protected](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/tianwenxue/p/10083509.html

时间: 12-07

beego中各类数据库连接方式的相关文章

Advanced Installer 中测试数据库连接提示“未发现数据源名称并且未指定默认驱动程序”的解决办法

原文:Advanced Installer 中测试数据库连接提示"未发现数据源名称并且未指定默认驱动程序"的解决办法 最近需要制作一个安装包,安装包的要求如下: 1.用户手工填写MySql的地址.用户名和密码 2.填写完成后提供测试数据库连接是否正常的按钮功能 3....... 项目要求打包的Mysql版本为5.5.27 x64版本,通过研究Advanced Installer 发现,此打包工具提供了数据库连接字符串,并且是通过ODBC的方式来连接的,既然是安装的64位的数据库,那么连

混合式开发框架模式中各种登陆方式的配置实现

在我们很多程序的登陆过程中,如果对于一些复杂的内容,可能都有一些配置管理的对话框,本文介绍的混合式开发框架,是指什么呢,我下面给了一个定义. <混合式开发框架>是指混合了传统<Winform开发框架>和<WCF开发框架>的特点,可以在两者之间自由切换,统一了系统界面层对业务服务的调用模式,所有组件模块均实现两种方式的调用,是一种弹性化非常好的框架应用,既可用于单机版软件或者基于局域网内的应用软件,也可以用于分布式技术的互联网环境应用. 对于这种混合式模式,上面说到了有两

Mac OS X中,有三种方式来实现启动项的配置

p.p1 { margin: 0.0px 0.0px 0.0px 0.0px; text-align: center; font: 22.0px "Times New Roman"; color: #000000 } p.p2 { margin: 0.0px 0.0px 0.0px 0.0px; font: 14.0px "Times New Roman"; color: #000000 } p.p3 { margin: 0.0px 0.0px 0.0px 0.0p

iOS中的桥接方式

iOS中的桥接方式 C ->OC (__bridge tupe)expression  : 将CoreFoundation框架的对象所有权交给Foundation框架来使用,但是Foundation框架中的对象并不能管理该对象内存 (__bridge_transfer Objective-C type)expression : 将CoreFoundation框架的对象所有权交给Foundation来管理,如果Foundation中对象销毁,那么我们之前的对象(CoreFoundation)会一起

[EF] 如何在 Entity Framework 中以手动方式设定 Code First 的 Migration 作业

Entity Framework (简称 EF) 发展到现在, 版本已经进入 6.1.0, 距离我写的「在 VS2013 以 Code First 方式建立 EF 资料库」这篇文章已有半年的时间.如果你和我一样从那时候开始使用 EF Code First, 那么你对 EF 应该已经有了基本的了解.依我个人的使用经验, EF 虽然好用, 但是如果一直使用 AutomaticMigrations 的方式维护你的资料库, 也许会遇到一些麻烦.因为在正常作业环境下, 资料库的格式不可能永远不变; 当我们

笨鸟先飞随笔:ADO.NET中有关C#从App.config中使用数据库连接字符串的方法

应用程序配置文件是标准的 XML 文件,XML 标记和属性是区分大小写的.它是可以按需要更改的,开发人员可以使用配置文件来更改设置,而不必重编译应用程序.在App.config文件中,根节点是configuration,在根节点下可以创建连接字符串的节点,也就是connectionStrings. 一.关于App.config的配置方法 1.在项目中添加一个应用程序配置文件(App.config),在根节点下添加连接字符串的节点<connectionStrings></connectio

C/C++浮点数在内存中的存储方式

任何数据在内存中都是以二进制的形式存储的,例如一个short型数据1156,其二进制表示形式为00000100 10000100.则在Intel CPU架构的系统中,存放方式为  10000100(低地址单元) 00000100(高地址单元),因为Intel CPU的架构是小端模式.但是对于浮点数在内存是如何存储的?目前所有的C/C++编译器都是采用IEEE所制定的标准浮点格式,即二进制科学表示法. 在二进制科学表示法中,S=M*2^N 主要由三部分构成:符号位+阶码(N)+尾数(M).对于fl

移码及浮点数在内存中的存储方式

首先说一下十进制的小数怎么转化为二进制的小数,计算机根本就不认识10进制的数据,他只认识0和1,所以,10进制的小数在计算机中是用二进制的小数表示的. 十进制的小数转化为二进制的小数的方法: 可以简单的概括为正序取整,将十进制的小数部分乘以2,然后取整数部分. 例如将0.2转化为二进制的小数,那么0.2*2=0.4,其整数部分是0,所以二进制小数的第一位为0,然后0.4*2=0.8,其整数部分是0,所以二进制小数的第二位为0,然后0.8*2=1.6,其整数部分是1,所以二进制小数的第三位是1,然

Delphi中编写OwnerDraw方式按钮的方法以及注意点

{在Win32标准控件库中(comctl32.dll)的控件绝大部分的控件都支持自画(OwnerDraw)功能的设置,这里以按钮为例子说说此类控件在delphi中的编写方式以及应该注意的问题!由于标准的按钮都有一套规定的外观标准(由windows的comctl32.dll中的代码绘制),为了给用户一个自己绘制控件的机会,多数的控件都支持在创建的时候设置一个标志来告诉系统这个控件需要自己绘制,例如按钮中有:BS_OWNERDRAW,菜单的有:MF_OWNERDRAW等等(还有诸如:ComboBox

jmeter中通过jdbc方式连接mysql数据库的配置参考

jmeter中通过jdbc方式连接mysql数据库的配置参考: Database URL=jdbc:mysql://ip:port/dbname?useUnicode=true&characterEncoding=UTF-8 JDBC Driver class=com.mysql.jdbc.Driver jmeter中配置截图: