Python Thrift 简单示例

本文基于Thrift-0.10，使用Python实现服务器端，使用Java实现客户端，演示了Thrift RPC调用示例。Java客户端提供两个字符串参数，Python服务器端计算这两个字符串的相似度，并返回相似度结果（double类型，范围[0, 1]，0表示不相似，1表示完全相同）

一，环境安装

开发环境：Windows10，PyCharm2016，Anaconda3，Python3.6

首先安装python 的thrift包：windows打开Anaconda prompt，输入：conda install -c anaconda thrift   安装thrift包。

输入：conda list 可查看系统中已经安装了哪些包，及包的版本，如下图所示：我们安装的是：thrift-0.10.0

在写代码之前，需要先定义一个 .thrift文件，然后使用Thrift Compiler生成相应的Thrift服务需要依赖的“文件”

①定义.thrift文件

namespace py similarityservice
namespace java similarityservice

service ChatSimilarityService{
    double similarity(1:string chat1, 2:string chat2),
}

namespace提供了一种组织代码的方式。其实就是，生成的文件放在：similarityservice这个文件夹下。

由于前面的Python安装的thrift-0.10，因此在官网上下载：thrift-0.10.exe，将它放在与 .thrift相同的目录下，cmd切换到该目录下，执行命令：

.	hrift-0.10.0.exe --gen py chat_similarity.thrift

生成的文件如下，将它们放在合适的python包下，即可供python 服务端程序 import 了。

二，Python服务端实现

pycharm thrift插件支持

可以去pycharm插件官网下载一个thrift插件，安装好之后，编写 .thrift 文件能够自动补全提示。

服务端的实现 主要有以下五方面：（个人理解，可能有错）

①Handler

服务端业务处理逻辑。这里就是业务代码，比如 计算两个字符串 相似度

②Processor

从Thrift框架 转移到 业务处理逻辑。因此是RPC调用，客户端要把 参数发送给服务端，而这一切由Thrift封装起来了，由Processor将收到的“数据”转交给业务逻辑去处理

③Protocol

数据的序列化与反序列化。客户端提供的是“字符串”，而数据传输是一个个的字节，因此会用到序列化与反序列化。

④Transport

传输层的数据传输。

⑤TServer

服务端的类型。服务器以何种方式来处理客户端请求，比如，一次Client请求创建一个新线程呢？还是使用线程池？……可参考：阻塞通信之Socket编程

TSimpleServer —— 单线程服务器端使用标准的阻塞式 I/O

TThreadPoolServer —— 多线程服务器端使用标准的阻塞式 I/O

TNonblockingServer —— 多线程服务器端使用非阻塞式 I/O

把上面生成的thrift文件复制到 thrift_service包下，如下图：

整个python 服务端的完整代码如下：

from thrift.protocol import TBinaryProtocol
from thrift.server import TServer
from thrift.transport import TSocket, TTransport

from text.thrift_service.similarityservice import ChatSimilarityService

from difflib import SequenceMatcher
from pypinyin import pinyin
import zhon
import pypinyin
from zhon.hanzi import punctuation
import re

__HOST = '127.0.0.1'
__PORT = 9090


def similar_num(list1, list2):
    return len(set(list1).intersection(list2))


def similar_ration(str1, str2):
    return SequenceMatcher(lambda x: x == ' ', str1, str2).ratio()

class SimilarityHandler(ChatSimilarityService.Iface):
    def __init__(self):
        self.log={}
    def ping(selfs):
        print('ping')

    def similarity(self, chat1, chat2):
        #去掉中文字符串中的特殊标点符号
        list1 = re.findall('[^{}]'.format(zhon.hanzi.punctuation), chat1)
        list2 = re.findall('[^{}]'.format(zhon.hanzi.punctuation), chat2)

        #将标点符号转换成拼音
        pinyin1 = pinyin(list1, style=pypinyin.STYLE_NORMAL)
        pinyin2 = pinyin(list2, style=pypinyin.STYLE_NORMAL)

        #将所有的拼音统一保存到 单个list 中
        pinyin_list1 = [word[0] for word in pinyin1]
        pinyin_list2 = [word[0] for word in pinyin2]

        #计算 list 中元素相同的个数
        result1 = similar_num(pinyin_list1, pinyin_list2)

        #list convert to string
        str1_pinyin = ''.join(pinyin_list1)
        str2_pinyin = ''.join(pinyin_list2)
        #计算字符串的相似度
        result2 = similar_ration(str1_pinyin, str2_pinyin)

        print('ratio:{}, nums:{}'.format(result2, result1))
        return result2


if __name__ == '__main__':
    handler = SimilarityHandler()
    processor = ChatSimilarityService.Processor(handler)
    transport = TSocket.TServerSocket(host=__HOST, port=__PORT)
    tfactory = TTransport.TBufferedTransportFactory()
    pfactory = TBinaryProtocol.TBinaryProtocolFactory()

    server = TServer.TThreadPoolServer(processor, transport, tfactory, pfactory)

    print('Starting the server')
    server.serve()
    print('done')

这里简单地介绍下实现思路：

①使用python 的 zhon 包过滤掉中文中出现的标点符号等特殊字符

②python的 pypinyin 包 将中文转换成字符串（其实也可以直接比较中文字符串的相似度，但我这里转换成了拼音，就相当于比较英文字符串了）

③使用python 的 difflib 包中的SequenceMatcher 类来计算两个字符串之间的相似度

三，Java客户端实现

①在maven工程的pom.xml中添加thrift依赖。这里的libthrift版本、windows10下载的thrift compiler版本（thrift-0.10.0.exe），还有 python的 thrift包的版本 最好保持一致。

        <!-- https://mvnrepository.com/artifact/org.apache.thrift/libthrift -->
        <dependency>
            <groupId>org.apache.thrift</groupId>
            <artifactId>libthrift</artifactId>
            <version>0.10.0</version>
        </dependency>

②cmd命令行执行：. hrift-0.10.0.exe --gen java chat_similarity.thrift  生成 ChatSimilarityService.java 文件，Java 客户端代码需要依赖它。

整个Java Client的代码如下：

import thrift.similarityservice.ChatSimilarityService;
import org.apache.thrift.TException;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;

/**
 * Created by Administrator on 2017/12/20.
 */
public class SimilarityThriftClient {

    public static void main(String[] args) {
        try {
            TTransport transport;
            transport = new TSocket("127.0.0.1", 9090);
            transport.open();

            TProtocol protocol = new TBinaryProtocol(transport);
            ChatSimilarityService.Client client = new ChatSimilarityService.Client(protocol);
            perform(client);
            transport.close();

        } catch (TException e) {
            e.printStackTrace();
        }
    }
    private static void perform(ChatSimilarityService.Client client)throws TException {
        String chat1 = "您好。";
        String chat2 = "你好";
        double ratio = client.similarity(chat1, chat2);
        System.out.println(ratio);
    }
}

四，总结

本文介绍了一个简单的 Python Server、Java Client的Thrift服务调用示例。关于Thrift可参考：

Thrift Tutorial

Apache Thrift - 可伸缩的跨语言服务开发框架

Thrift 安装及使用

原文：Python Thrift 简单示例