Golang使用gopg访问MemFireDB数据库

本篇介绍如何使用Golang访问MemFireDB数据库。

如果大家有个人项目或者公司内部测试项目，推荐大家尝试MemFire Cloud，不用自己搭建数据库，在家或者公司随时可以访问。

gopg是Golang中最常用的访问数据库的ORM库，MemFireDB兼容Postgres接口，所以可以直接把MemFireDB当Postgres使用。

在 memfiredb.com 上创建数据库时，可以选择创建密码或者证书两种认证模式，在公网上部署的数据库，如果存储的是比较重要的数据，建议使用证书认证模式，该模式安全系数要高很多，撞库、彩虹表、暴力破解等传统的密码破解方式对证书认证是无能为力的。如果只是简单测试，可以使用密码认证方式，这种方式配置简单些。

下面是示例代码：

1
2import (
3 "context"
4 "flag"
5 "fmt"
6 "github.com/go-pg/pg/v10"
7 "github.com/go-pg/pg/v10/orm"
8 "strings"
9 "sync"
10)
11
12var (
13 addr   = flag.String("addr", "192.168.80.161:5433", "memfire address to connect")
14 user   = flag.String("user", "test", "memfire user")
15 passwd = flag.String("passwd", "test", "memfire password")
16 dbname = flag.String("db", "dbname", "memfire database name to connect")
17)
18
19func panicIf(err error) {
20 if err != nil {
21  panic(err)
22 }
23}
24
25func test_transaction_try_again(db *pg.DB) {
26 incrInTx := func(db *pg.DB) error {
27  // Transaction is automatically rolled back on error.
28  return db.RunInTransaction(db.Context(), func(tx *pg.Tx) error {
29   var counter int
30   _, err := tx.QueryOne(
31    pg.Scan(&counter), `SELECT counter FROM counters FOR UPDATE`)
32   if err != nil {
33    return err
34   }
35
36   counter++
37
38   _, err = tx.Exec(`UPDATE counters SET counter = ?`, counter)
39   return err
40  })
41 }
42
43 var wg sync.WaitGroup
44 for i := 0; i < 10; i++ {
45  wg.Add(1)
46  go func() {
47   defer wg.Done()
48   for {
49    err := incrInTx(db)
50    if err != nil {
51     if strings.Contains(err.Error(), "40001") ||
52      strings.Contains(err.Error(), "Try again") ||
53      strings.Contains(err.Error(), "Restart read required") {
54      fmt.Println("Try again")
55      continue
56     }
57     panic(err)
58    }
59    break
60   }
61
62  }()
63 }
64 wg.Wait()
65}
66
67type Counter struct {
68 Counter int64
69}
70
71type User struct {
72 Id     int64
73 Name   string
74 Emails []string
75}
76
77// createSchema creates database schema for Counter/ User
78func createSchema(db *pg.DB) error {
79 models := []interface{}{
80  (*Counter)(nil),
81  (*User)(nil),
82 }
83
84 for _, model := range models {
85  err := db.Model(model).CreateTable(&orm.CreateTableOptions{
86   Temp:        false,
87   IfNotExists: true,
88  })
89  if err != nil {
90   return err
91  }
92 }
93 return nil
94}
95
96func main() {
97 flag.Parse()
98
99 opt := pg.Options{
100  Addr:     *addr,
101  User:     *user,
102  Password: *passwd,
103  Database: *dbname,
104  OnConnect: func(ctx context.Context, cn *pg.Conn) error {
105   println("new connection created")
106   return nil
107  },
108 }
109
110 db := pg.Connect(&opt)
111 defer db.Close()
112
113 err := createSchema(db)
114 panicIf(err)
115
116 _, err = db.Exec("delete from counters")
117 panicIf(err)
118
119 cnt := &Counter{
120  Counter: 1,
121 }
122 _, err = db.Model(cnt).Insert()
123 panicIf(err)
124
125 test_transaction_try_again(db)
126}