使用 Thrift 进行文件传输

这里简单展现一下 Thrift 的文件传输功能

一 . 设计 Thrift 的接口定义

新建 Thrift 接口定义文件 Interface.thrift , 作为客户端和服务端进行通信说明，里面内容如下 :

// -------------------------------------------------------------------- File
struct FileData
{
    1:required string   name,                   // 文件名字
    2:required binary   buff,                   // 文件数据
}

// -------------------------------------------------------------------- Service

service FileInfoExtractService
{
    bool uploadFile(1:FileData filedata);       // 文件解析函数
}

命令行切换到 Interface.thrift 文件所在目录，执行> thrift.exe -gen java Interface.thrift , 生成文件 FileData.java 和 FileService.java ，这两个文件在 Thrift 的 Server 端和 Client 端都会被用上。

二 . 建立 Thrift 的 Client 端项目

1. Client 项目包含 3 个文件：除了步骤一中生成的 FileData.java 和 FileService.java 之外，新建一个 App.java 文件作为程序入口

2. Client 项目中 FileData.java 和 FileService.java 的具体代码由 Thrift 生成，这里略

3. Client 项目中 App.java 的代码如下：

import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.transport.TFramedTransport;
import org.apache.thrift.transport.TSocket;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.nio.ByteBuffer;

public class App
{
    /**
     * 类内测试 main 函数
     * */
    public static void main(String[] args)
    {
        // 测试文件路径
        String filePath = "C:\\Users\\jiangmengya\\Desktop\\1.jpg";

        // 构造文件数据
        byte[] bytes = toByteArray(filePath);
        FileData fileData = new FileData();
        fileData.name = filePath;
        fileData.buff = ByteBuffer.wrap(bytes);

        // 构造Thrift客户端，发起请求
        try
        {
            TSocket socket = new TSocket("localhost", 12345);
            socket.setSocketTimeout(60 * 1000);
            TFramedTransport framedTransport = new TFramedTransport(socket);
            framedTransport.open();
            TBinaryProtocol binaryProtocol = new TBinaryProtocol(framedTransport);
            FileService.Client client = new FileService.Client(binaryProtocol);
            client.uploadFile(fileData);
        }
        catch (Exception x)
        {
            x.printStackTrace();
        }
    }

    /**
     * 文件转化为字节数组
     * */
    private static byte[] toByteArray(String filePath){
        byte[] buffer = null;
        try {
            File file = new File(filePath);
            FileInputStream fis = new FileInputStream(file);
            ByteArrayOutputStream bos = new ByteArrayOutputStream(1000);
            byte[] b = new byte[1000];
            int n;
            while ((n = fis.read(b)) != -1) {
                bos.write(b, 0, n);
            }
            fis.close();
            bos.close();
            buffer = bos.toByteArray();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return buffer;
    }
}

三. 建立 Thrift 的 Server 端项目

1. Server 项目包含 4 个文件：除了步骤一中生成的 FileData.java 和 FileService.java 之外，新建一个 App.java 文件作为程序入口, 新建一个 FileServiceImpl.java 文件来对 Thrift 服务进行具体的实现

2. Server 项目中 FileServiceImpl.java 的代码如下：

import org.apache.thrift.TException;
import java.io.FileOutputStream;
import java.nio.channels.FileChannel;

public class FileServiceImpl implements FileService.Iface
{
    @Override
    public boolean uploadFile(FileData filedata) throws TException
    {
        System.out.println("uploadFile function has been called.");

        // 写到文件
        String filePath = "C:\\Users\\jiangmengya\\Desktop\\2.jpg";
        try
        {
            java.io.File file = new java.io.File(filePath);
            FileOutputStream fos = new FileOutputStream(file);
            FileChannel channel = fos.getChannel();
            channel.write(filedata.buff);
            channel.close();
        }
        catch (Exception x)
        {
            x.printStackTrace();
            return false;
        }
        return true;
    }
}

3. Server 项目中 App.java 的代码如下：

import org.apache.thrift.TProcessor;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TThreadedSelectorServer;
import org.apache.thrift.transport.TFramedTransport;
import org.apache.thrift.transport.TNonblockingServerSocket;
import org.apache.thrift.transport.TNonblockingServerTransport;
import org.apache.thrift.transport.TTransportFactory;

public class App
{
    public static void main(String[] args)
    {
        try
        {
            // 创建非阻塞的 Transport
            TNonblockingServerTransport serverSocket = new TNonblockingServerSocket(12345);

            // 创建 Processor
            TProcessor processor = new FileService.Processor(new FileServiceImpl());

            // 创建 transport factory , Nonblocking 使用 TFramedTransport
            TTransportFactory transportFactory = new TFramedTransport.Factory();

            // 创建 protocol factory
            TBinaryProtocol.Factory protocolFactory = new TBinaryProtocol.Factory();

            // 创建 arguments
            TThreadedSelectorServer.Args tArgs = new TThreadedSelectorServer.Args(serverSocket);
            tArgs.processor(processor);
            tArgs.transportFactory(transportFactory);
            tArgs.protocolFactory(protocolFactory);

            // 创建 server
            TServer server = new TThreadedSelectorServer(tArgs);

            // 启动 server
            server.serve();
        }
        catch (Exception x)
        {
            x.printStackTrace();
        }
    }
}

四. 运行项目

1. 运行 Server 项目 App

2. 运行 Client 项目 App

3. 输出：

uploadFile function has been called.

（同时，Client 指定的图片 1.jpg 也会被拷贝传输到 Server 指定的路径下成为 2.jpg）

使用 Thrift 进行文件传输

猜你喜欢