跳到主要内容

Go操作数据库

零 Go数据库接口​

Go官方没有提供数据库驱动，而是为开发数据库驱动定义了一些标准接口，开发者可以根据定义的接口来开发相应的数据库驱动，这样做的好处是：框架迁移极其方便。

Go数据库标准包位于以下两个包中：

  • database/sql：提供了保证SQL或类SQL数据库的泛用接口
  • database/sql/driver：定义了应被数据库驱动实现的接口，这些接口会被sql包使用

关于包的详细使用位于07-标准库中。

一 Go操作MySQL​

Go中支持MySQL的驱动目前比较多，有如下几种，有些是支持database/sql标准，而有些是采用了自己的实现接口。

推荐使用：

代码示例：

package main

import (
	"database/sql"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
)

func main() {
	
	db, err := sql.Open("mysql", "root:123456@/mydata?charset=utf8")
	checkErr(err)

	//插入数据
	sql1 := "INSERT INTO user SET name=?,age=?"
	stmt, err := db.Prepare(sql1)
	checkErr(err)
	res, err := stmt.Exec("zs", "30",)
	checkErr(err)

	id, err := res.LastInsertId()
	checkErr(err)
	fmt.Println("插入id=", id)


	//更新数据
	sql2 := "UPDATE user SET name=? WHERE id=?";
	stmt, err = db.Prepare(sql2)
	checkErr(err)
	res, err = stmt.Exec("lisi", id)
	checkErr(err)
	affect, err := res.RowsAffected()
	checkErr(err)
	fmt.Println("更新行数=", affect)

	//查询数据
	rows, err := db.Query("SELECT * FROM user")
	checkErr(err)
	for rows.Next() {
		var id int
		var name string
		var age int
		err = rows.Scan(&id, &name, &age)
		checkErr(err)
		fmt.Println("id=", id)
		fmt.Println("name=", name)
		fmt.Println("age=", age)
	}

	//删除数据
	// stmt, err = db.Prepare("DELETE FROM user WHERE uid=?")
	// checkErr(err)
	// res, err = stmt.Exec(id)
	// checkErr(err)
	// affect, err = res.RowsAffected()
	// checkErr(err)
	// fmt.Println(affect)

	db.Close()

}

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

相关API：

  • sql.Open()：打开一个注册过的数据库驱动，第二个参数格式有：
    • user@unix(/path/to/socket)/dbname?charset=utf8
    • user:password@tcp(localhost:5555)/dbname?charset=utf8
    • user:password@/dbname
    • user:password@tcp([de:ad:be:ef::ca:fe]:80)/dbname
  • db.Prepare()：返回准备要执行的sql操作，然后返回准备完毕的执行状态
  • db.Query()：直接执行Sql返回Rows结果
  • stmt.Exec()：执行stmt准备好的SQL语句

二 Go操作redis​

2.1 Go操作Redis​

推荐驱动：

基本操作：

package main

import (
	"fmt"
	"github.com/gomodule/redigo/redis"
)


func main() {

	c, err := redis.Dial("tcp", "localhost:6379")

	if err != nil {
		fmt.Println("Redis connect err:", err)
		return
	}

	defer c.Close()

	_, err = c.Do("Set", "first", "hello")
	if err != nil {
		fmt.Println(err)
		return
	}

	r, err := redis.Int(c.Do("Get", "first"))
	if err != nil {
		fmt.Println(err)
		return
	}

	fmt.Println(r)
}

2.2 连接池​

package main

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/gomodule/redigo/redis"
)

var (
	Pool *redis.Pool
)

func init() {
	redisHost := ":6379"
	Pool = newPool(redisHost)
	listenForClose()
}

func newPool(server string) *redis.Pool {

	return &redis.Pool{
		MaxIdle:     3,
		IdleTimeout: 240 * time.Second,

		Dial: func() (redis.Conn, error) {
			c, err := redis.Dial("tcp", server)
			if err != nil {
				return nil, err
			}
			return c, err
		},

		TestOnBorrow: func(c redis.Conn, t time.Time) error {
			_, err := c.Do("PING")
			return err
		},
	}
}

func listenForClose() {
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)
	signal.Notify(c, syscall.SIGTERM)
	signal.Notify(c, syscall.SIGKILL)
	go func() {
		<-c
		_ = Pool.Close()
		os.Exit(0)
	}()
}

func Get(key string) ([]byte, error) {

	conn := Pool.Get()
	defer conn.Close()
	var data []byte
	data, err := redis.Bytes(conn.Do("GET", key))
	if err != nil {
		return data, fmt.Errorf("error get key %s: %v", key, err)
	}
	return data, err
}

func main() {
	test, err := Get("test")
	fmt.Println(test, err)
	ExampleNewClient()
	// Output
	// [] error get key test: redigo: nil returned
	// PONG <nil>
	// 4396 <nil>
	// key: overnote2 not exist
	// 0 redigo: nil returned
}

func ExampleNewClient() {
	c, _ := redis.Dial("tcp", "localhost:6379",
		redis.DialDatabase(0),
	)
	pong, err := redis.String(c.Do("ping"))
	fmt.Println(pong, err)
	// Output: PONG <nil>
	key := "overnote"
	val := 4396
	_, err = c.Do("SET", key, val)
	if err != nil {
		panic(err)
	}
	res, err := redis.Int(c.Do("GET", key))
	if err == redis.ErrNil {
		fmt.Printf("key: %s not exist\n", key)
	} else {
		if err != nil {
			panic(err)
		}
	}
	fmt.Println(res, err)
	// 4396 nil
	notExistKey := "overnote2"
	res, err = redis.Int(c.Do("GET", notExistKey))
	if err == redis.ErrNil {
		fmt.Printf("key: %s not exist\n", notExistKey)
		// key: overnote2 not exist
	} else {
		if err != nil {
			panic(err)
		}
	}
	fmt.Println(res, err)
	// 0 redigo: nil returned
}

三 Go操作MongoDB​

MongoDB为Go提供了官方驱动：

https://github.com/mongodb/mongo-go-driver