tnblog
首页
登录

.net core gRPC 客户端流式上传图片,文件

55人阅读 2020/2/9 14:02 总访问:134350 评论:0 手机 收藏
分类: .NET Core

html前台,完全一样:

<form method="post" action="/home/UpLoadImgStream" enctype="multipart/form-data">
    <input type="file" name="upfile" />
    <input type="submit" name="上传" />
</form>


gRPC服务端

proto

syntax = "proto3";

option csharp_namespace = "GrpcService1";

package UpImgStream;

// The greeting service definition.
service UpImgStream {
  // Sends a greeting
  rpc DoUpLoad (stream StreamRequest) returns (UpImgStreamReply);
}

// The request message containing the user's name.
message StreamRequest  {
  bytes content = 1;
  //文件后缀名(弃用,通过另外的方式在文件之前传递)
  string fileExt=2;
}

// The response message containing the greetings.
message UpImgStreamReply {
  string message = 1;
}

实现:

public override async Task<UpImgStreamReply> DoUpLoad(IAsyncStreamReader<StreamRequest> requestStream, ServerCallContext context)
{
    try
    {
        //获取调用方的id
        string host = context.Host;
        //if (context.UserState.ContainsKey("fileExt"))
        //{
        //    object fileExt = context.UserState["fileExt"];
        //}
        //注意获取的时候用小写
        var fileExtMetadata = context.RequestHeaders.FirstOrDefault(a => a.Key == "fileext");
        if (fileExtMetadata == null)
        {
            return new UpImgStreamReply
            {
                Message = "上传失败!"
            };
        }
        string fileExt = fileExtMetadata.Value;

        // 新建一个文件流用于存放我们获取到数据
        using (var fs = new FileStream("c://" + Guid.NewGuid().ToString().Replace("-", "") + "." + fileExt, FileMode.Create))
        {
            while (await requestStream.MoveNext()) // 循环从流中读取数据(监听gRPC客户端的流写入)
            {
                requestStream.Current.Content.WriteTo(fs); // 将数据写入到文件流中
            }
            //数据是先被读到了内存中,再把数据写到文件中,当你数据读完的时候不代表你的数据已经写完了,因为还有一部分有可能会留在内存这个缓冲区中。这时候如果你调用了close()方法关闭了读写流,那么这部分数据就会丢失,所以应该在关闭读写流之前先flush()。
            fs.Flush();
        }
        return new UpImgStreamReply
        {
            Message = "上传成功!"
        };
    }
    catch (Exception ex)
    {
        return new UpImgStreamReply
        {
            Message = "上传失败!"
        };
    }
}

gRPC客户端

方法一:文件分段传输

适用于文件有点点大的情况

/// <summary>
/// 调用gRPC 客户端流式图片上传(文件分段传输)
/// </summary>
public async void UpLoadImgStream(List<IFormFile> upfile)
{
    //获取页面传递的文件流
    foreach (var formFile in upfile)
    {
        if (formFile.Length > 0)
        {
            #region 获取gRPC接口
            AppContext.SetSwitch("System.Net.Http.SocketsHttpHandler.Http2UnencryptedSupport", true);
            var channel = GrpcChannel.ForAddress("http://localhost:8099/");
            var client = new UpImgStream.UpImgStreamClient(channel);
            #endregion

            //获取文件的后缀名
            string fileExt = formFile.FileName.Substring(formFile.FileName.LastIndexOf('.') + 1); //文件扩展名

            Metadata entries = new Metadata();
            entries.Add("fileExt", fileExt);
            AsyncClientStreamingCall<StreamRequest, UpImgStreamReply> call = client.DoUpLoad(entries);//先调用服务后面在提供数据

            //得到文件流
            Stream stream = formFile.OpenReadStream();

            try
            {
                var remainingLength = stream.Length; //文件总长度
                int sendlength = 100 * 1024;//1048576;

                //先把分段读取的数据存储在集合里边,因为直接调用gRPC传输,第二次读取流的时候会关闭,暂时还没有解决就使用这种方法
                List<byte[]> byteList = new List<byte[]>();

                byte[] buff = new byte[sendlength]; // 缓冲区,这里我们设置为 nkb
                while (remainingLength > 0) // 若未读完则继续读取(把文件分为单位nkb的传递,如果小于nkb则一次传递完)
                {
                    if (remainingLength < sendlength)//剩下的可以一次传递完
                    {
                        buff = new byte[remainingLength];
                    }

                    //注意读一次后调用WriteAsync,Stream流就关闭了,但是如果不执行WriteAsync是可以读完所有的
                    int len = stream.Read(buff, 0, buff.Length); // 异步从文件中读取数据到缓冲区中
                    remainingLength -= len; // 剩余长度=当前总长度-这次实际读取的长度

                    byteList.Add(buff);
                }

                //分段调用gRPC流进行文件传输
                foreach (byte[] item in byteList)
                {
                    //向gRPC流中写入我们刚刚读取的数据
                    await call.RequestStream.WriteAsync(new StreamRequest() { Content = ByteString.CopyFrom(item) });
                }

                await call.RequestStream.CompleteAsync();//完成关闭客户端流
            }
            catch (Exception ex)
            {

            }
        }
    }
}

方法二:一次性传输

/// <summary>
/// 调用gRPC 客户端流式图片上传(一次性传输)
/// </summary>
public async void UpLoadImgStream(List<IFormFile> upfile)
{
    //获取页面传递的文件流
    foreach (var formFile in upfile)
    {
        if (formFile.Length > 0)
        {
            #region 获取gRPC接口
            AppContext.SetSwitch("System.Net.Http.SocketsHttpHandler.Http2UnencryptedSupport", true);
            var channel = GrpcChannel.ForAddress("http://localhost:8099/");
            var client = new UpImgStream.UpImgStreamClient(channel);
            #endregion

            //获取文件的后缀名
            string fileExt = formFile.FileName.Substring(formFile.FileName.LastIndexOf('.') + 1); //文件扩展名

            Metadata entries = new Metadata();
            entries.Add("fileExt", fileExt);
            AsyncClientStreamingCall<StreamRequest, UpImgStreamReply> call = client.DoUpLoad(entries);//先调用服务后面在提供数据

            //得到文件流
            Stream stream = formFile.OpenReadStream();

            try
            {
                //方法1:一次传递完
                StreamRequest streamRequest = new StreamRequest() { Content = ByteString.FromStream(stream) };
                await call.RequestStream.WriteAsync(streamRequest);

                await call.RequestStream.CompleteAsync();//完成关闭客户端流
            }
            catch (Exception ex)
            {

            }
        }
    }
}

得到流式调用的返回值:

await call.RequestStream.CompleteAsync();//完成关闭客户端流(注意一定要调用不然整个调用过程完成不了)
//得到返回值
UpImgStreamReply upImgStreamReply = await call.ResponseAsync;

下载地址:http://www.tnblog.net/resource/show/aojiancc2/af3134fbbe174025ad34e7ccf7de465f




评价
很多事不是看到了希望才去努力,而是努力了才能看到希望
排名
6
文章
6
粉丝
16
评论
8
{{item.ArticleTitle}}
{{item.BlogName}} : {{item.Content}}
ICP备案 :渝ICP备18016597号-1
网站信息:2018-2020TNBLOG.NET