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）