python调用golang代码

一、调用步骤：

　　将go代码编译成so库 -> python中通过ctypes引用so库并指定需要调用的函数（同时可指定传入参数类型和返回值类型） -> 指定后按python使用函数方式调用。

　　需要注意的是：python和go之间参数传递是需要经过C的数据类型转换的，因此需要了解python中ctypes数据类型和python数据类型以及C的数据类型对应关系

　　三种数据类型使用场景：1. ctypes数据类型为指定调用函数时的传入参数和返回值的数据类型

　　　　　　　　　　　　　2. python数据类型为调用函数时传入的参数的数据类型

　　　　　　　　　　　　3. C的数据类型为go代码中定义的函数所需的参数和返回值数据类型

　　类型对应如下：文档传送地址

　　由此举例：当python传入的参数需是string时，ctypes中指定的传参参数类型需为c_wchar_p，go中需要指定接收的参数数据类型为 *C.wchar_t。

　　由于不知道go中如何将字符串和*C.wchar_t互相转化，因此我这里将python传入的参数指定为bytes，即ctypes中指定的传参参数类型需为c_char_p，go中需要指定接收的参数数据类型为*C.char。

二、下面开始实践：

1. 编写go代码

写一个rocketmq的producer函数（main.go），封装成Send函数如下：

 1 package main

 2

 3 import (

 4     "C"

 5     "context"

 6     "github.com/apache/rocketmq-client-go/v2"

 7     "github.com/apache/rocketmq-client-go/v2/primitive"

 8     "github.com/apache/rocketmq-client-go/v2/producer"

 9     "os"

10 )

11

12 var (

13     nameservs = []string{"192.168.2.1:9876"}

14     group     = "demo.xy"

15     topic     = "test"

16 )

17

18 //export Send

19 func Send(cid, message *C.char) *C.char {

20     p, err := rocketmq.NewProducer(

21         producer.WithNsResolver(primitive.NewPassthroughResolver(nameservs)),

22         producer.WithRetry(2),

23         producer.WithGroupName(group),

24     )

25     if err != nil {

26         return C.CString("create producer failed")

27         os.Exit(-1)

28     }

29

30     err = p.Start()

31     if err != nil {

32         return C.CString("start producer failed")

33         os.Exit(-1)

34     } else {

35         defer p.Shutdown()

36     }

37

38     msg := &primitive.Message{

39         Topic: topic,

40         Body:  []byte(C.GoString(message)),

41     }

42     msg.WithTag(C.GoString(cid))

43

44     _, err = p.SendSync(context.Background(), msg)

45     if err != nil {

46         return C.CString("producer send message failed")

47     } else {

48         return C.CString("producer send message success")

49     }

50 }

51

52 func main() {}

需要注意：1). Go里面将C的char指针类型数据通过C.GoString()转化成go中的字符串；反之通过C.CString()将go中的字符串转化为C的char指针类型数据。具体类型转化方式可查阅相关文档（待补充）

　　　　 2). python需要调用的函数必须在函数上方用 //export [函数名称] 加上说明，不然python中会报AttrbuteError错误(symbol not found)

2. 将go代码编译成so动态链接库

go build --buildmode=c-shared -o producer.so main.go

编译好后会生成两个文件：producer.so和producer.h

3. python中导入so文件并调用对应的函数

 1 import ctypes

 2 import json

 3

 4 # 指定go中的Send函数

 5 SendSync = ctypes.CDLL("./producer.so").Send

 6 # 指定调用函数时传入的参数数据类型

 7 SendSync.argtypes = [ctypes.c_char_p, ctypes.c_char_p]

 8 # 指定调用函数返回值的数据类型

 9 SendSync.restype = ctypes.c_char_p

10

11 cid = "123454321"

12 message = {

13     "cid": "123454321",

14     "cname": "指南朝北枪",

15     "age": 18,

16     "height": 1.88

17 }

18

19 result = SendSync(cid.encode("utf-8"), json.dumps(message).encode("utf-8"))

20 print(result, type(result))

需要注意的是：1). python2中的字节串是str；字符串是unicode。因此如果是python2调用SendSync函数时不需要使用encode，直接传入str即可

　　　　　　　2). python3中字节串是bytes；字符串是str。因此调用SendSync函数时需要将字符串str通过encode转换成bytes

　　　　　　 3). 指定的参数类型和传入参数类型一定要一致，否者报ctypes.ArgumentError错误（wrong type）

巴特西

python调用golang代码

最新文章

热门文章