我正在尝试将 go 代码编译并运行为 Postgresql 存储过程。 我的动机是因为 postgresql 可以有用 C 编写的扩展，而 golang 可以编译为 c-shared

所以我必须提交文件，pl.go:

package main

/*
#cgo CFLAGS: -Wall -Wpointer-arith -Wno-declaration-after-statement -Wendif-labels -Wmissing-format-attribute -Wformat-security -fno-strict-aliasing -fwrapv -fexcess-precision=standard -march=x86-64 -mtune=generic -O2 -pipe -fstack-protector-strong -fpic -I. -I./ -I/usr/include/postgresql/server -I/usr/include/postgresql/internal -D_FORTIFY_SOURCE=2 -D_GNU_SOURCE -I/usr/include/libxml2
#cgo LDFLAGS: -Wall -Wmissing-prototypes -Wpointer-arith -Wdeclaration-after-statement -Wendif-labels -Wmissing-format-attribute -Wformat-security -fno-strict-aliasing -fwrapv -fexcess-precision=standard -march=x86-64 -mtune=generic -O2 -pipe -fstack-protector-strong -fpic -L/usr/lib -Wl,-O1,--sort-common,--as-needed,-z,relro  -Wl,--as-needed -Wl,-rpath,'/usr/lib',--enable-new-dtags -shared

#include "postgres.h"
#include "fmgr.h"
#include "utils/builtins.h"

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

//the return value must be allocated trough palloc
void* ret(void *val, uint64 *size) {
    void *retDatum = palloc(*size);
    memcpy(retDatum, val, *size);
    return retDatum;
}

PG_FUNCTION_INFO_V1(plgo_func);
*/
import "C"
import "unsafe"

func main() {}

//PGVal returns the Postgresql C type from Golang type (currently implements just stringtotext)
func PGVal(val interface{}) (ret interface{}) {
    var size uintptr
    switch v := val.(type) {
    case string:
        ret = C.cstring_to_text(C.CString(v))
        size = unsafe.Sizeof(ret)
    default:
        ret = val
        size = unsafe.Sizeof(ret)
    }
    return C.ret(ret, (*C.uint64)(unsafe.Pointer(size)))
}

CFLAGS 和 LDFLAGS 我是从 pg_config

以及我在其中创建要调用的函数的文件 plgo.go:

package main

/*
#include "postgres.h"
#include "fmgr.h"
#include "utils/builtins.h"
*/
import "C"

//export plgo_func
func plgo_func(fcinfo *C.FunctionCallInfoData) interface{} {
    return PGVal("meh")
}

共享库是通过以下方式创建的:go build -buildmode=c-shared -o plgo.so plgo.go pl.go && sudo cp plgo.so/usr/lib/postgresql

postgresql 中的函数是用以下方法创建的:

CREATE OR REPLACE FUNCTION public.plgo_func(integer)
  RETURNS text AS
'$libdir/plgo', 'plgo_func'
  LANGUAGE c IMMUTABLE STRICT
  COST 1;

但是当我运行时:psql -U root -d meh -c "select plgo_func(0)"

服务器崩溃:

server closed the connection unexpectedly
    This probably means the server terminated abnormally
    before or while processing the request.
connection to server was lost

编辑:我已经成功创建了一个 golang“库”，用于在 golang 中创建存储过程和触发器 plgo :)

最佳答案

诀窍是使用版本 0 调用约定，因为这些约定允许调用简单的 C 函数，而无需使用他们添加到版本 1 调用中的所有花哨的宏。

您还需要一个 C 文件来调用 PG_MODULE_MAGIC 宏。

我有一个可以执行所有这些操作的工作示例项目，如果您只是将其复制为起点，这可能是最简单的。

不确定它是否能让您完全按照自己的意愿行事，但它确实适用于某些用例。

https://github.com/dbudworth/gopgfuncs

还将概述 repo 告诉您的相同内容...

第一步

用你的函数创建一个 .go 文件，CGO 包含

package main

//#include "postgres.h"
//#include "fmgr.h"
//#include <string.h>
import "C"
import (
    "log"
    "sync/atomic"
)

// Functions are scoped to the db session
var counter int64

//Inc atomically increments a session local counter by a given delta
//export Inc
func Inc(delta C.int64) C.int64 {
    log.Printf("Inc(%v) called", delta)
    return C.int64(atomic.AddInt64(&counter, int64(delta)))
}

//AddOne adds one to the given arg and retuns it
//export AddOne
func AddOne(i C.int) C.int {
    log.Printf("AddOne(%v) called", i)
    return i + 1
}

func main() {
}

第二步

在从 postgres.h 调用 PG_MODULE_MAGIC 宏的同一目录中创建一个 .c 文件，这是所有扩展库的要求。 Go 会在编译时自动包含 C 文件，不需要特殊说明。

#include "postgres.h"
#include "fmgr.h"

PG_MODULE_MAGIC;

第 3 步

构建你的so文件，这里的技巧是使用-buildmode=c-shared

go build -buildmode=c-shared -o libMyMod.so

第四步

执行 SQL 以注册您的函数。由于注册扩展需要绝对路径，我更喜欢在命令行上传递模块以避免使“install.sql”脚本在其中具有硬编码路径。

PGMOD=`pwd`/libMyMod.so
psql $(DBNAME) --set=MOD=\'$(PGMOD)\' -f install.sql

安装脚本包含您导出的每个函数的这些语句之一。

获得正确的输入/输出类型绝对至关重要，因为系统无法验证它们，您最终可能会占用内存、存储空间等。您可以看到 pg 类型到 c 类型的翻译 here

create or replace function add_one(integer)
  returns integer as :MOD,'AddOne'
  LANGUAGE C STRICT;

关于postgresql - Postgresql 的 Golang 过程语言，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/38590942/