第1步

Question

我尝试编译并运行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＆＃34;库＆＃34;用于在golang plgo创建存储过程和触发器：）

Answer 1

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

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

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

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

https://github.com/dbudworth/gopgfuncs

还会概述回购告诉你的相同内容......

第1步

使用您的函数创建.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() {
}

第2步

在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

第4步

执行SQL以注册您的功能。由于注册扩展需要绝对路径，我更喜欢在命令行上传递模块，以避免使用＆＃34; install.sql＆＃34;脚本中有硬编码路径。

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的Golang过程语言

1 个答案:

第1步

第2步

第3步

第4步