在go中使用PostgreSQL

在go中使用PostgreSQL

1. PostgreSQL

进行后台开发的时候，不可避免的就是需要使用到数据库，在众多数据库中，PostgreSQL是一个很好的选择，本人一开始使用的是blotdb，因为在服务计算课上使用过，这是一个很容易使用的键值数据库，但是在使用过程中，本人就发现键值数据库有很大的弊端，blotdb其实更加使用查询比较多的环境，不适合需要频繁更新的环境，因此，后面就放弃了使用blotdb作为数据库；

之后改用了PostgreSQL，主要是因为在数据库上学习了该数据库，至少数据库操作部分可以很快上手，并且不需要重新配置环境；在另一方面，PostgreSQL和MySQL比较，它更加庞大一点，因为它是用来替代Oracle而设计的。所以在企业应用中采用PostgreSQL是一个明智的选择。

PostgreSQL是一个功能强大的开源对象关系数据库管理系统(ORDBMS)。 用于安全地存储数据; 支持最佳做法，并允许在处理请求时检索它们。
PostgreSQL(也称为Post-gress-Q-L)由PostgreSQL全球开发集团(全球志愿者团队)开发。 它不受任何公司或其他私人实体控制。 它是开源的，其源代码是免费提供的。
PostgreSQL是跨平台的，可以在许多操作系统上运行，如Linux，FreeBSD，OS X，Solaris和Microsoft Windows等。

2. PostgreSQL 相关操作

在使用go语言操作数据库之前，我们至少需要了解一下如何使用PostgreSQL语言：

• 新建数据库：create database testdb;

• 查看所有的数据库：\l

• 查看所有的表：\d

• 切换数据库：\c testdb;

• 删除数据库：drop database testdb;

• 创建表：

CREATE TABLE table_name( 
column1 datatype,  
column2 datatype,  
column3 datatype,   
PRIMARY KEY( one or more columns )  );

• 删除表：drop table table_name;

• 创建模式：create schema schema_name;

• 删除模式：drop schema schema_name;

• 递归删除模式：drop schema schema_name cascade;

• 插入数据：INSERT INTO TABLE_NAME (column1, column2, column3,...columnN) VALUES (value1, value2, value3,...valueN); 可以同时插入多个值

• 查询语句：SELECT ID, NAME, AGE, SALARY FROM EMPLOYEES WHERE [condition];

• 更新语句：UPDATE table_name SET column1 = value1, column2 = value2...., columnN = valueN WHERE [condition];

• 删除语句：DELETE FROM table_name WHERE [condition];

• Order By：

SELECT column-list
FROM table_name 
[WHERE condition]
ORDER BY column1, column2, .. columnN [ASC | DESC];

• 分组：

SELECT column-list  
FROM table_name 
WHERE [conditions ] 
GROUP BY column1, column2....columnN [ORDER BY column1, column2....columnN]
eg：select name,sum(id) from student group by name;

• Having：

SELECT column-list  
	   FROM table_name 
       WHERE [conditions ] 
       GROUP BY column1, column2....columnN 
	   Having  [ conditions ]  
       [ORDER BY column1, column2....columnN]

• 查询条件：And、Or、Not、Like、In、Not In、Between

• Like：以su开头：“%su”

• 内连接：

SELECT table1.columns, table2.columns  
       FROM table1  INNER JOIN table2  
       ON table1.common_filed = table2.common_field;

• 外连接：

SELECT table1.columns, table2.columns 
       FROM table1  LEFT OUTER JOIN table2 
       ON table1.common_filed = table2.common_field;

• 外连接分为左外连接，右外连接，全外连接

• 跨连接（叉乘）：SELECT coloums FROM table1 CROSS JOIN table2

• 创建视图：

CREATE [TEMP | TEMPORARY] VIEW view_name 
       AS  SELECT column1, column2.....  FROM table_name 
       WHERE [condition];

• 删除视图：DROP VIEW view_name;

• 函数：

CREATE [OR REPLACE] FUNCTION function_name (arguments)   
RETURNS return_datatype AS $variable_name$  
DECLARE  
  declaration;  
  [...]  
BEGIN  
  < function_body >  
  [...]  
  RETURN { variable_name | value }  
END; LANGUAGE plpgsql;

--example
CREATE OR REPLACE FUNCTION totalRecords ()  
RETURNS integer AS $total$  
declare  
  total integer;  
BEGIN  
 SELECT count(*) into total FROM EMPLOYEES;  
 RETURN total;  
END;  
$total$ LANGUAGE plpgsql;
--执行创建的函数
select totalRecords();

• 触发器

CREATE  TRIGGER trigger_name [BEFORE|AFTER|INSTEAD OF] event_name  
ON table_name  
[
  -- Trigger logic goes here....  
];

• 别名：As 可以列、表添加别名

• 索引：

  • 单列索引：CREATE INDEX index_name ON table_name (column_name);
  • 多列索引：CREATE INDEX index_name ON table_name (column1_name, column2_name);
  • 唯一索引：CREATE UNIQUE INDEX index_name on table_name (column_name);

• Union：两个结合的表必须有相同的列，去重

• Union all：两个结合的表必须有相同的列，不去重

• Alter

使用ALTER TABLE语句在现有表中添加新列的基本语法如下：
ALTER TABLE table_name ADD column_name datatype;

现有表中ALTER TABLE到DROP COLUMN(删除某个字段)的基本语法如下：
ALTER TABLE table_name DROP COLUMN column_name;

ALTER TABLE更改表中列的DATA TYPE(修改字段类型)的基本语法如下：
ALTER TABLE table_name ALTER COLUMN column_name TYPE datatype;

ALTER TABLE向表中的列添加NOT NULL约束的基本语法如下：
ALTER TABLE table_name MODIFY column_name datatype NOT NULL;

ALTER TABLE添加唯一约束ADD UNIQUE CONSTRAINT到表中的基本语法如下：
ALTER TABLE table_name
ADD CONSTRAINT MyUniqueConstraint UNIQUE(column1, column2...);

ALTER TABLE将“检查约束”添加到表中的基本语法如下所示：
ALTER TABLE table_name
ADD CONSTRAINT MyUniqueConstraint CHECK (CONDITION);

ALTER TABLE添加主键ADD PRIMARY KEY约束的基本语法如下：
ALTER TABLE table_name
ADD CONSTRAINT MyPrimaryKey PRIMARY KEY (column1, column2...);

使用ALTER TABLE从表中删除约束(DROP CONSTRAINT)的基本语法如下：
ALTER TABLE table_name
DROP CONSTRAINT MyUniqueConstraint;

使用ALTER TABLE从表中删除主键约束(DROP PRIMARY KEY)约束的基本语法如下：
ALTER TABLE table_name
DROP CONSTRAINT MyPrimaryKey;

• 截断表：TRUNCATE TABLE table_name;

• 事务：

• 锁：LOCK [ TABLE ]name IN lock_mode;

  • lock_mode包括：ACCESS SHARE，ROW SHARE，ROW EXCLUSIVE，SHARE UPDATE EXCLUSIVE，SHARE，SHARE ROW EXCLUSIVE，EXCLUSIVE，ACCESS EXCLUSIVE（未指定的情况下选择最严格的模式）。

• 自动递增：serial

CREATE TABLE tablename (
    colname SERIAL
);

• 权限：

--Grant
GRANT privilege [, ...]
ON object [, ...]
TO { PUBLIC | GROUP group | username }

--Revoke
REVOKE privilege [, ...]
ON object [, ...]
FROM { PUBLIC | GROUP groupname | username }

3. 数据库驱动

在使用之前，我们首先需要安装相关的驱动，Go实现的支持PostgreSQL的驱动也很多，本项目中使用的是"github.com/lib/pq"，在使用之前需要在终端运行下面的命令：

go get -u "github.com/lib/pq"
go get -u "database/sql"

4. go 操作数据库

go提供数据库很简单，下面是一些简单的例子：

在看例子之前，先看一下表的结构：

• t_msg（消息通知）

Field	Type	Key	Description
msgID	text	PRIMARY KEY	消息的ID
state	integer		消息的状态（已读和未读）
receiver	text	FOREIGN KEY	消息的接受者
time	text		消息的时间
title	text		消息的标题
content	text		消息的内容

（1）打开数据库

psqlInfo := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%ssslmode=disable",yourHost, yourPort, yourUser, yourPassword, yourDbname)

db, err := sql.Open("postgres", psqlInfo)

（2）查询

rows, err := db.Query("select * from t_msg where receiver=$1", userID)

for rows.Next() {
    messageObj := models.Message{}
    err = rows.Scan(&messageObj.MsgID, &messageObj.State, &messageObj.Receiver, &messageObj.Time, &messageObj.Title, &messageObj.Content)
    if err != nil {
        fmt.Printf("could not find message, %v", err)
        db.Close()
        return nil, false
    }
    messages = append(messages, messageObj)
}

db.Close()

（3）插入

stmt, err := db.Prepare("insert into t_msg (msgID, state, receiver, time, title, content) values($1, $2, $3, $4, $5, $6)")
if err != nil {
    fmt.Printf("could not Insert Mesage, %v", err)
    ok = false
}

_, err = stmt.Exec(message.MsgID, message.State, message.Receiver, message.Time, message.Title, message.Content)

if err != nil {
    fmt.Printf("could not insert mesage, %v", err)
    ok = false
}

db.Close()

（4）更新

stmt, err := db.Prepare("update t_msg set state=$1 WHERE msgID=$2 and receiver=$3")
if err != nil {
    fmt.Println("fail to update:%v", err)
    ok = false
}
_, err = stmt.Exec(state, messageID, userID)
if err != nil {
    fmt.Println(err)
    ok = false
}
db.Close()

（5）删除

stmt, err := db.Prepare("delete from t_msg where msgID=$1 and receiver=$2")
if err != nil {
    fmt.Println("delete error: %v", err)
    ok = false
}
_, err = stmt.Exec(messageID, userID)
if err != nil {
    ok = false
    fmt.Println("delete error: %v", err)
}

db.Close()

（6）count

var count int64
err = db.QueryRow("select count(*) from t_msg where msgID=$1", id).Scan(&count)

ok = true
if err != nil || count == 0 {
    fmt.Printf("could not query, %v", err)
    ok = false
}

db.Close()

5. 其他

需要注意的是，在每次操作完数据库之后需要调用db.Close()函数关闭数据库，否则可能出现无法写入的情况，这一点本人也不是很清楚。