Go DB4. 使用PostgreSQL数据库

PostgreSQL 是一个自由的对象-关系数据库服务器(数据库管理系统)，它在灵活的BSD-风格许可证下发行。它提供了相对其他开放源代码数据库系统(比如 MySQL 和 Firebird)，和对专有系统比如Oracle、Sybase、IBM 的 DB2 和 Microsoft SQL Server的一种选择。

PostgreSQL和MySQL比较，它更加庞大一点，因为它是用来替代Oracle而设计的。所以在企业应用中采用PostgreSQL是一个明智的选择。

MySQL被Oracle收购之后正在逐步的封闭（自MySQL 5.5.31以后的所有版本将不再遵循GPL协议），鉴于此，将来我们也许会选择PostgreSQL而不是MySQL作为项目的后端数据库。

PostgreSQL驱动

Go实现的支持PostgreSQL的驱动也很多，因为国外很多人在开发中使用了这个数据库。

* https://github.com/lib/pq 支持database/sql驱动，纯Go写的（推荐）

* https://github.com/jbarham/gopgsqldriver 支持database/sql驱动，纯Go写的

* https://github.com/lxn/go-pgsql 支持database/sql驱动，纯Go写的

推荐使用第一个驱动，因为它目前使用的人最多，在github上也比较活跃。

示例代码

1、示例的数据库表结构如下所示，相应的建表SQL：

CREATE TABLE userinfo

(

    uid serial NOT NULL,

    username character varying(100) NOT NULL,

    department character varying(500) NOT NULL,

    Created date,

    CONSTRAINT userinfo_pkey PRIMARY KEY (uid)

)

WITH (OIDS=FALSE);

CREATE TABLE userdetail

(

    uid integer,

    intro character varying(100),

    profile character varying(100)

)

WITH(OIDS=FALSE);

2、使用database/sql接口对数据库表进行增删改查操作：

package main

import (

    "database/sql"

    "fmt"

    _ "github.com/lib/pq"

)

func main() {

    // 连接数据库

    db, err := sql.Open("postgres", "user=root password=1234 dbname=test sslmode=disable")

    checkErr(err)

    defer db.Close()

    // 插入数据

    stmt, err := db.Prepare("INSERT INTO userinfo(username, department, created) values($1,$2,$3) RETURNING uid")

    checkErr(err)

    res, err := stmt.Exec("lai", "rm", "2018-01-01")

    checkErr(err)

    // pg不支持这个函数，因为他没有类似MySQL的自增ID

    // id, err := res.LastInsertId()

    // checkErr(err)

    // fmt.Println(id)

    var lastInsertId int

    err = db.QueryRow("INSERT INTO userinfo(username,department,created) VALUES($1,$2,$3) returning uid;", "lai", "rm", "2018-01-01").Scan(&lastInsertId)

    checkErr(err)

    fmt.Println("lastInsertId =", lastInsertId)

    // 更新数据

    stmt, err = db.Prepare("UPDATE userinfo SET username=$1 WHERE uid=$2")

    checkErr(err)

    res, err = stmt.Exec("laiupdate", 1)

    checkErr(err)

    affect, err := res.RowsAffected()

    checkErr(err)

    fmt.Println(affect)

    // 查询数据

    rows, err := db.Query("SELECT * FROM userinfo")

    checkErr(err)

    for rows.Next() {

        var uid int

        var username string

        var department string

        var created string

        err = rows.Scan(&uid, &username, &department, &created)

        checkErr(err)

        fmt.Println(uid)

        fmt.Println(username)

        fmt.Println(department)

        fmt.Println(created)

    }

    // 删除数据

    stmt, err = db.Prepare("DELETE FROM userinfo WHERE uid=$1")

    checkErr(err)

    res, err = stmt.Exec(1)

    checkErr(err)

    affect, err = res.RowsAffected()

    checkErr(err)

    fmt.Println(affect)

}

func checkErr(err error) {

    if err != nil {

        panic(err)

    }

}

从上面的代码中我们可以看到，PostgreSQL是通过$1, $2这种方式来指定要传递的参数，而不是MySQL中的？，另外在sql.Open中的DSN信息的格式也与MySQL的驱动中的dsn格式不一样，所以在使用时注意他们的差异。

还有pg不支持LastInsertId函数，因为PostgreSQL内部没有实现类似MySQL的自增ID返回，其他的代码几乎是一模一样。