Thrift多接口支持例子

最新推荐文章于 2023-06-16 15:21:42 发布

LieBrother

最新推荐文章于 2023-06-16 15:21:42 发布

阅读量1.6k

点赞数

分类专栏： Thrift Thrift深入浅出文章标签： thrift

本文链接：https://blog.csdn.net/u011642663/article/details/56286831

版权

Thrift 同时被 2 个专栏收录

6 篇文章 0 订阅

订阅专栏

Thrift深入浅出

6 篇文章 2 订阅

订阅专栏

个人博客原文：Thrift多接口支持例子

摘要：本文简单写一个Thrift支持多接口的代码

Thrift通过TMultiplexedProcessor来设置多个服务接口
下面是代码：

IDL接口

HelloService.thrift

namespace java com.thrift.demo.service 
service HelloService {
    i32 sayInt(1:i32 param)
    string sayString(1:string param)
    bool sayBoolean(1:bool param)
    void sayVoid()
}

TestService.thrift

struct Test {
  1: required bool bool_; //required 必须填充也必须序列化
  2: optional i8 i8_; //optional 不填充则不序列化
  3: i16 i16_;
  4: i32 i32_;
  5: i64 i64_;
  6: double double_;
  7: string string_;
  8: binary binary_;
}

service TestService {
  void getVoid(1:Test test);
  bool getBool(1:Test test);
  i8 getI8(1:Test test);
  i16 getI16(1:Test test);
  i32 getI32(1:Test test);
  i64 getI64(1:Test test);
  double getDouble(1:Test test);
  string getString(1:Test test);
  binary getBinary(1:Test test);
}

接口实现类

HelloServiceImpl.java

public class HelloServiceImpl implements HelloService.Iface {

    public int sayInt(int param) throws TException {
        System.out.println("say int :" + param);
        return param;
    }

    public String sayString(String param) throws TException {
        System.out.println("say string :" + param);
        return param;
    }

    public boolean sayBoolean(boolean param) throws TException {
        System.out.println("say boolean :" + param);
        return param;
    }
    public void sayVoid() throws TException {
        System.out.println("say void ...");
    }
}

TestServiceImpl.java

public class TestServiceImpl implements TestService.Iface {

    @Override
    public void getVoid(Test test) throws TException {
        System.out.println("TestServiceImpl + " + test.toString());
    }

    @Override
    public boolean getBool(Test test) throws TException {
        System.out.println("TestServiceImpl.getBool() + " + test.bool_);
        return test.bool_;
    }

    @Override
    public byte getI8(Test test) throws TException {
        System.out.println("TestServiceImpl.getI8() + " + test.i8_);
        return test.i8_;
    }

    @Override
    public short getI16(Test test) throws TException {
        System.out.println("TestServiceImpl.getI16() + " + test.i16_);
        return test.i16_;
    }

    @Override
    public int getI32(Test test) throws TException {
        System.out.println("TestServiceImpl.getI32() + " + test.i32_);
        return test.i32_;
    }

    @Override
    public long getI64(Test test) throws TException {
        System.out.println("TestServiceImpl.getI64() + " + test.i64_);
        return test.i64_;
    }

    @Override
    public double getDouble(Test test) throws TException {
        System.out.println("TestServiceImpl.getDouble() + " + test.double_);
        return test.double_;
    }

    @Override
    public String getString(Test test) throws TException {
        System.out.println("TestServiceImpl.getString() + " + test.string_);
        return test.string_;
    }

    @Override
    public ByteBuffer getBinary(Test test) throws TException {
        System.out.println("TestServiceImpl.getBinary() + " + test.binary_);
        return test.binary_;
    }

}

Server端

public class ThriftServer {
    public static void main(String[] args) {
        try {
            TServerSocket serverTransport = new TServerSocket(9090);
            Factory protocolFactory = new TBinaryProtocol.Factory();
            //创建多个服务Processor
            Processor<TestService.Iface> processorTest = new TestService.Processor<TestService.Iface>(new TestServiceImpl());
            com.thrift.demo01.service.HelloService.Processor<Iface> processorHello = new HelloService.Processor<HelloService.Iface>(new HelloServiceImpl());

            //将服务注册到TMultiplexedProcessor中
            TMultiplexedProcessor processors = new TMultiplexedProcessor();
            processors.registerProcessor("testService", processorTest);
            processors.registerProcessor("helloService", processorHello);

            TThreadPoolServer.Args args = new TThreadPoolServer.Args(serverTransport)
                                .protocolFactory(protocolFactory)
                                .processor(processors)
                                .minWorkerThreads(1000)
                                .maxWorkerThreads(1000);
            TServer server = new TThreadPoolServer(args);
            System.out.println("开启thrift服务器，监听端口：9090");
            server.serve();
        } catch (TTransportException e) {
            e.printStackTrace();
        }
    }
}

Client端

public class ThriftClient {
    public static void testTMultiplexedProcessor() {
        try {
            TTransport transport = new TSocket("localhost", 9090);
            TProtocol protocol = new TBinaryProtocol(transport);
            //通过TMultiplexedProtocol获取对应的服务
            TMultiplexedProtocol protocolTest = new TMultiplexedProtocol(protocol, "testService");
            TMultiplexedProtocol protocolHello = new TMultiplexedProtocol(protocol, "helloService");

            //创建Client调用服务接口的方法
            TestService.Client client = new TestService.Client(protocolTest);
            HelloService.Client clientHello = new HelloService.Client(protocolHello);
            transport.open();

            //调用接口方法
            Test test = new Test();
            test.setBool_(true);
            test.setDouble_(5.0);
            System.out.println("Client getBool()" + client.getBool(test));
            System.out.println("Client getDouble()" + client.getDouble(test));

            System.out.println("HelloService sayString :" + clientHello.sayString("哈哈哈"));
            transport.close();
        } catch (TTransportException e) {
            e.printStackTrace();
        } catch (TException te) {
            te.printStackTrace();
        }
    }
}