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)


  • 作者:合十
  • 发表时间:2022年3月19日 21:05
  • 更新时间:2022年12月10日 02:22
  • 所属分类:我用Python

Comments

该文章还未收到评论,点击下方评论框开始评论吧~