Go基础学习记录 - 编写Web应用程 - 数据库(四)

DurbanZhang · · 945 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

本次分享下 -- PostgreSQL数据库驱动程序

为了保持项目的可学习性,我这里将之前分享的代码积累了下,放在github上,想要尽快入手学习的,可以直接clone我的代码,写代码不上手,都等于白搭,光看的话,对于我来说,我是不行的,没办法学会。

项目地址

https://github.com/durban89/wiki_blog
tag: 1.0.10

有些同学可能看不懂,怎么就只给了这些,完全不懂呀。我把使用的命令打出来,照着操作,就应该可以解决了

git clone https://github.com/durban89/wiki_blog /local/path
cd /local/path
git fetch origin
git checkout 1.0.10

这些我觉得 够清晰了。OK!

继续分享"数据库"的分享。

数据库

对于Web开发人员而言,数据库是Web开发的核心。
您几乎可以将任何内容保存到数据库中,并查询或更新其中的数据,如用户信息,产品或新闻文章。
Go不提供任何数据库驱动程序,但它确实在database/sql包中定义了驱动程序接口。
我们可以基于该接口开发数据库驱动程序。首先先看下 “PostgreSQL数据库驱动程序”

PostgreSQL

PostgreSQL是一个对象关系数据库管理系统,可用于许多平台,包括Linux,FreeBSD,Solaris,Microsoft Windows和Mac OS X.它是在MIT风格的许可下发布的,因此是免费的开源软件。
它比MySQL大,因为它是为企业用作Oracle的替代品而设计的。
Postgresql是企业类项目的不错选择。

PostgreSQL drivers

PostgreSQL有许多可用的数据库驱动程序。以下是它们的三个例子

  1. https://github.com/lib/pq supports database/sql, written in pure Go.
  2. https://github.com/jbarham/go... supports database/sql, written in pure Go.
  3. https://github.com/lxn/go-pgsql supports database/sql, written in pure Go.

我将使用后面示例中的第一个。

Samples 实例演示

安装PostgreSQL驱动

如果本地没有安装的话,先去安装下这个包,安装命令如下

go get https://github.com/lib/pq
添加PostgreSQL驱动连接

项目目录下创建一个文件夹db,db中创建一个文件pg.go,文件内容如下

package db

import (
    "database/sql"
    "fmt"

    // Register PostgreSQL
    _ "github.com/lib/pq"
)

const (
    // DBUser 用户名
    DBUser = "root"
    // DBPassword 密码
    DBPassword = "123456"
    // DBName  库名
    DBName = "test"
)

// PostgreSQLDB Conn
var PostgreSQLDB *sql.DB

func init() {
    dbinfo := fmt.Sprintf("user=%s password=%s dbname=%s sslmode=disable",
        DBUser, DBPassword, DBName)
    db, err := sql.Open("postgres", dbinfo)

    PostgreSQLDB = db
    checkPostgreSQLErr(err)
}

func checkPostgreSQLErr(err error) {
    if err != nil {
        panic(err)
    }
}

由于我是初学者,一些大胆的尝试,可能存在规范上或者架构上的问题,还请多多指教。
我这里之所以这样建立,本着代码分离的原则,不希望在每个使用数据库的地方都重复写一边代码,这里会不会存在重复创建连接,导致内存崩溃还有待继续深入学习。
好了,废话不多说了,这里创建了一个PostgreSQLDBB,首先说下init这个函数的作用

init函数是用于程序执行前做包的初始化的函数,比如初始化包里的变量等

也就是所每次使用PostgreSQLDB的时候都会去连接下数据库

这就说明了,每次使用PostgreSQLDB的时候,PostgreSQLDB都是一个连接了的,方便后面的调用。

创建model

由于以前都是写PHP的,大多数使用的也都是MVC,所以我这里在项目目录下建立了一个models文件夹,然后创建一个blog.go文件,里面的代码如下

package models

import (
    "database/sql"

    "github.com/durban89/wiki/db"
)

// Conn 连接
var Conn *sql.DB

func init() {
    // MySQL
    // Conn = db.DB
    // SQLite
    // Conn = db.SQLiteDB
    // PostgreSQL
    Conn = db.PostgreSQLDB
}

// Query 获取一条数据
func Query() ([]string, error) {
    rows, err := Conn.Query("SELECT * FROM blog")

    if err != nil {
        return nil, err
    }

    var res = []string{}

    for rows.Next() {
        var autokid int
        var title string
        err = rows.Scan(&autokid, &title)

        if err != nil {
            return nil, err
        }
        res = append(res, title)
    }

    return res, nil
}
数据库创建

创建数据库blog

CREATE TABLE blog (
    autokid SERIAL PRIMARY KEY,
    title VARCHAR(100)
);

添加测试数据

INSERT INTO blog(title) VALUES('title1');
INSERT INTO blog(title) VALUES('title2');
INSERT INTO blog(title) VALUES('title3');
单元测试

下面创建测试单元,测试下我们的Query函数是否能获取到数据,创建blog_test.go文件(如果不存在),跟blog.go文件放在一起,代码内容如下

package models

import (
    "testing"
)

// TestQuery 测试获取数据
func TestQuery(t *testing.T) {
    row, err := Query()

    if err != nil {
        t.Error(err)
    } else {
        t.Log(row)
    }

    if len(row) > 0 {
        t.Log("正确")

    }

    for i, k := range row {
        t.Log(i)
        t.Log(k)
    }
}

当我们在models目录下进行测试的时候,会得到如下输出,说明我们的功能是正常的

$ go test
PASS
ok      github.com/durban89/wiki/models 0.023s

请注意,PostgreSQL使用$1,$2格式而不是MySQL使用的?,它在sql.Open中有不同的DSN格式。另一件事是PostgreSQL驱动程序不支持sql.Result.LastInsertId()。所以不是这样,

// Create 添加一条记录
func Create(title string) {
    var lastInsertID int
    err := PostgreSQLDB.QueryRow("INSERT INTO blog(title) VALUES($1) returning autokid;", title).Scan(&lastInsertID)
    checkPostgreSQLErr(err)
    fmt.Println("last inserted id =", lastInsertID)
}

使用db.QueryRow()和.Scan()来获取最后插入的id的值。

今天就分享到这里,如果你有其他疑问请在下方留言或者加群交流

项目更新地址

https://github.com/durban89/typescript_demo.git
tag: 1.0.11

有疑问加站长微信联系(非本文作者)

本文来自:Segmentfault

感谢作者:DurbanZhang

查看原文:Go基础学习记录 - 编写Web应用程 - 数据库(四)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

945 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传