pgbase

简介

成功实现了多写, 通过自定义 DBConnect 将 PocketBase 的存储后端由 sqlite 切换到 postgres

原理

使用 Polyglot 将 sqlite 语句翻译成 postgres 语句

测试

docker-compose up -d
cd ./example
# download libpolyglot_sql_ffi.so in https://github.com/tobilg/polyglot/releases
go run -v . serve

示例

package main

import (
	"strings"

	"github.com/pocketbase/dbx"
	"github.com/pocketbase/pocketbase"
	"github.com/pocketbase/pocketbase/core"
	"github.com/shynome/err0/try"
	"github.com/shynome/pgbase"
	polyglot "github.com/tobilg/polyglot/packages/go"
)

func main() {
	tc := try.To1(polyglot.OpenDefault())
	polyglot.SetDefaultClient(tc)
	cfg := pocketbase.Config{
		DBConnect: func(dbPath string) (*dbx.DB, error) {
			if strings.HasSuffix(dbPath, "data.db") {
				return pgbase.Open("postgres://user:pass@127.0.0.1:5432/postgres?sslmode=disable")
			}
			return core.DefaultDBConnect(dbPath)
		},
		DataMaxOpenConns: core.DefaultDataMaxOpenConns,
		DataMaxIdleConns: core.DefaultDataMaxIdleConns,
	}
	app := pocketbase.NewWithConfig(cfg)
	try.To(app.Bootstrap())

	// required!
	// required!
	// required!
	// postgres support concurrent, reset max conns
	if db, ok := app.NonconcurrentDB().(*dbx.DB); ok {
		db.DB().SetMaxOpenConns(cfg.DataMaxOpenConns)
		db.DB().SetMaxIdleConns(cfg.DataMaxIdleConns)
	}
	try.To(app.Start())
}