grpc java 文件传输下载实现

要实现 gRPC Java 文件传输下载，可以使用 gRPC 提供的流（Stream）特性，这可以将文件分割成较小的块，并将其逐个传输。以下是一个简单的示例：

首先，在.proto 文件中定义一个新的服务，包含两个方法：一个用于上传文件，另一个用于下载文件。

service FileTransferService {
    rpc uploadFile(stream FileChunk) returns (UploadStatus);
    rpc downloadFile(FileRequest) returns (stream FileChunk);
}

message FileChunk {
    bytes data = 1;
}

message FileRequest {
    string filename = 1;
}

message UploadStatus {
    bool success = 1;
    string message = 2;
}

接下来，生成 Java 代码并实现服务接口：

public class FileTransferServiceImpl extends FileTransferServiceGrpc.FileTransferServiceImplBase {

    @Override
    public StreamObserver<FileChunk> uploadFile(StreamObserver<UploadStatus> responseObserver) {
        // 处理文件上传
    }

    @Override
    public void downloadFile(FileRequest request, StreamObserver<FileChunk> responseObserver) {
        // 处理文件下载
    }
}

对于上传文件，我们需要使用 StreamObserver<FileChunk> 来接收文件块，然后将它们合并起来，并给客户端返回上传状态。

public StreamObserver<FileChunk> uploadFile(StreamObserver<UploadStatus> responseObserver) {
    return new StreamObserver<FileChunk>() {
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();

        @Override
        public void onNext(FileChunk fileChunk) {
            try {
                outputStream.write(fileChunk.getData().toByteArray());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        @Override
        public void onError(Throwable t) {
            t.printStackTrace();
        }

        @Override
        public void onCompleted() {
            // 处理文件上传完成逻辑
            responseObserver.onNext(UploadStatus.newBuilder().setSuccess(true).build());
            responseObserver.onCompleted();
        }
    };
}

对于下载文件，我们需要根据客户端请求的文件名，读取文件并将其分成块，使用 StreamObserver<FileChunk> 发送文件块。

public void downloadFile(FileRequest request, StreamObserver<FileChunk> responseObserver) {
    try {
        FileInputStream fileInputStream = new FileInputStream(request.getFilename());
        byte[] buffer = new byte[1024];
        int bytesRead;

        while ((bytesRead = fileInputStream.read(buffer)) != -1) {
            FileChunk fileChunk = FileChunk.newBuilder().setData(ByteString.copyFrom(buffer, 0, bytesRead)).build();
            responseObserver.onNext(fileChunk);
        }

        responseObserver.onCompleted();
    } catch (IOException e) {
        responseObserver.onError(e);
    }
}

最后，在服务器端启动 gRPC 服务：

public static void main(String[] args) throws IOException, InterruptedException {
    Server server = ServerBuilder.forPort(8080)
            .addService(new FileTransferServiceImpl())
            .build();

    server.start();
    server.awaitTermination();
}

现在，您已经可以使用 gRPC Java 来实现文件传输下载了。