From 58bd64afb5ce159b47f20165c64a7b100c08012d Mon Sep 17 00:00:00 2001 From: ftpgxm Date: Tue, 11 May 2021 12:34:41 +0800 Subject: [PATCH] init --- build.sh | 10 + client/client.go | 129 ++++++++++++ go.mod | 9 + go.sum | 91 ++++++++ proto/transport.pb.go | 477 ++++++++++++++++++++++++++++++++++++++++++ proto/transport.proto | 28 +++ server/server.go | 96 +++++++++ 7 files changed, 840 insertions(+) create mode 100755 build.sh create mode 100644 client/client.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 proto/transport.pb.go create mode 100644 proto/transport.proto create mode 100644 server/server.go diff --git a/build.sh b/build.sh new file mode 100755 index 0000000..87239f5 --- /dev/null +++ b/build.sh @@ -0,0 +1,10 @@ +#!/bin/bash + +CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build -o dist/client-darwin-amd64 client/client.go +CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build -o dist/server-darwin-amd64 server/server.go + +CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o dist/client-linux-amd64 client/client.go +CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o dist/server-linux-amd64 server/server.go + +CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o dist/client-windows-amd64.exe client/client.go +CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o dist/server-windows-amd64.exe server/server.go diff --git a/client/client.go b/client/client.go new file mode 100644 index 0000000..37e64a8 --- /dev/null +++ b/client/client.go @@ -0,0 +1,129 @@ +/** + * Created by IntelliJ IDEA. + * User: Ar.M + * Date: 2020-05-10 + * Time: 22:10 + */ +package main + +import ( + "bytes" + "context" + "flag" + "fmt" + "github.com/dustin/go-humanize" + "google.golang.org/grpc" + "io" + "log" + "os" + "strings" + transportpb "yunlian-file-transfer-demo/proto" +) + +var ( + fn *string + ip *string + port *int + savePath *string +) + +func init() { + fn = flag.String("fn", "", "文件名称") + ip = flag.String("ip", "127.0.0.1", "服务端 IP 地址") + port = flag.Int("port", 10086, "服务端端口号") + savePath = flag.String("savepath", "./", "存储路径") + + flag.Usage = func() { + fmt.Println("yunlian-file-transfer-demo version: 0.0.1") + flag.PrintDefaults() + } +} + +func main() { + flag.Parse() + + if len(*fn) < 1 { + flag.Usage() + return + } + + requestedFile := *fn + serverIp := fmt.Sprintf("%s:%d", *ip, *port) + + cc, err := grpc.Dial(serverIp, grpc.WithInsecure()) + if err != nil { + log.Fatalf("监听服务端异常 : %v\n", err) + } + defer cc.Close() + + c := transportpb.NewFileServiceClient(cc) + + if err := download(requestedFile, c); err != nil { + log.Fatalf("下载异常 %s : %v\n", requestedFile, err) + } + fmt.Printf("\n%s 下载成功\n", requestedFile) +} + +func writeFile(fileName string, content []byte) { + if len(content) != 0 { + fp, err := os.OpenFile(fileName, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0755) + if err != nil { + log.Fatalf("文件打开异常: %s\n", err) + } + defer fp.Close() + _, err = fp.Write(content) + if err != nil { + log.Fatalf("文件保存异常: %s\n", err) + } + } +} + +func getFileStat(fileName string) int64 { + fileinfo, err := os.Stat(fileName) + if err != nil { + if os.IsNotExist(err) { + return int64(0) + } + log.Fatalf("获取文件元数据失败: %s\n", err) + } + return fileinfo.Size() +} + +func download(fileName string, client transportpb.FileServiceClient) error { + fileRange := getFileStat(fileName) + req := &transportpb.FileRequest{ + FileName: fileName, + FileRange: fileRange, + } + + stream, err := client.Download(context.Background(), req) + if err != nil { + return err + } + + var recvSize int64 + var buffer bytes.Buffer + + recvSize = fileRange + for { + res, err := stream.Recv() + if err == io.EOF { + break + } + if err != nil { + buffer.Reset() + return err + } + + block := res.GetBlock() + fileSize := res.GetFileSize() + blockSize := len(block) + recvSize += int64(blockSize) + + writeFile(fileName, block) + + // fmt.Printf("\r%s", strings.Repeat(" ", 30)) + fmt.Printf("\r文件%s: %s / %s 已下载%s", fileName, humanize.IBytes(uint64(recvSize)), humanize.IBytes(uint64(fileSize)), strings.Repeat(" ", 60)) + } + return nil +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..bafe909 --- /dev/null +++ b/go.mod @@ -0,0 +1,9 @@ +module yunlian-file-transfer-demo + +go 1.16 + +require ( + github.com/dustin/go-humanize v1.0.0 + github.com/golang/protobuf v1.5.2 + google.golang.org/grpc v1.37.0 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..664e2a2 --- /dev/null +++ b/go.sum @@ -0,0 +1,91 @@ +cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= +github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= +github.com/cncf/udpa/go v0.0.0-20201120205902-5459f2c99403/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo= +github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk= +github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= +github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= +github.com/envoyproxy/go-control-plane v0.9.9-0.20210217033140-668b12f5399d/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk= +github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c= +github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= +github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8= +github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA= +github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs= +github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= +github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= +github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8= +github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= +github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/golang/protobuf v1.5.2 h1:ROPKBNFfQgOUMifHyP+KYbvpjbdoFNs+aK7DXlji0Tw= +github.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY= +github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M= +github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= +golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= +golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU= +golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190311183353-d8887717615a h1:oWX7TPOiFAMXLq8o0ikBYfCJVlRHBcsciT5bXOrH628= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a h1:1BGLXjeY4akVXGgbC9HugT3Jv3hCI0z56oJR5vAMgBU= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/text v0.3.0 h1:g61tztE5qeGQ89tm6NTjjM9VPIm088od1l6aSorWRWg= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY= +golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM= +google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= +google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc= +google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc= +google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013 h1:+kGHl1aib/qcwaRi1CbqBZ1rk19r85MNUf8HaBghugY= +google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo= +google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c= +google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg= +google.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY= +google.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk= +google.golang.org/grpc v1.37.0 h1:uSZWeQJX5j11bIQ4AJoj+McDBo29cY1MCoC1wO3ts+c= +google.golang.org/grpc v1.37.0/go.mod h1:NREThFqKR1f3iQ6oBuvc5LadQuXVGo9rkm5ZGrQdJfM= +google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8= +google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0= +google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM= +google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE= +google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo= +google.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c= +google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= +google.golang.org/protobuf v1.26.0 h1:bxAC2xTBsZGibn2RTntX0oH50xLsqy1OxA9tTL3p/lk= +google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= diff --git a/proto/transport.pb.go b/proto/transport.pb.go new file mode 100644 index 0000000..d1feb1e --- /dev/null +++ b/proto/transport.pb.go @@ -0,0 +1,477 @@ +// Code generated by protoc-gen-go. DO NOT EDIT. +// versions: +// protoc-gen-go v1.26.0 +// protoc v3.15.8 +// source: proto/transport.proto + +package proto + +import ( + context "context" + grpc "google.golang.org/grpc" + codes "google.golang.org/grpc/codes" + status "google.golang.org/grpc/status" + protoreflect "google.golang.org/protobuf/reflect/protoreflect" + protoimpl "google.golang.org/protobuf/runtime/protoimpl" + reflect "reflect" + sync "sync" +) + +const ( + // Verify that this generated code is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) + // Verify that runtime/protoimpl is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) +) + +type ListRequestType struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields +} + +func (x *ListRequestType) Reset() { + *x = ListRequestType{} + if protoimpl.UnsafeEnabled { + mi := &file_proto_transport_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *ListRequestType) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*ListRequestType) ProtoMessage() {} + +func (x *ListRequestType) ProtoReflect() protoreflect.Message { + mi := &file_proto_transport_proto_msgTypes[0] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use ListRequestType.ProtoReflect.Descriptor instead. +func (*ListRequestType) Descriptor() ([]byte, []int) { + return file_proto_transport_proto_rawDescGZIP(), []int{0} +} + +type ListResponseType struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + FileName string `protobuf:"bytes,1,opt,name=file_name,json=fileName,proto3" json:"file_name,omitempty"` + FileSize int64 `protobuf:"varint,2,opt,name=file_size,json=fileSize,proto3" json:"file_size,omitempty"` + FileMode uint32 `protobuf:"varint,3,opt,name=file_mode,json=fileMode,proto3" json:"file_mode,omitempty"` +} + +func (x *ListResponseType) Reset() { + *x = ListResponseType{} + if protoimpl.UnsafeEnabled { + mi := &file_proto_transport_proto_msgTypes[1] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *ListResponseType) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*ListResponseType) ProtoMessage() {} + +func (x *ListResponseType) ProtoReflect() protoreflect.Message { + mi := &file_proto_transport_proto_msgTypes[1] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use ListResponseType.ProtoReflect.Descriptor instead. +func (*ListResponseType) Descriptor() ([]byte, []int) { + return file_proto_transport_proto_rawDescGZIP(), []int{1} +} + +func (x *ListResponseType) GetFileName() string { + if x != nil { + return x.FileName + } + return "" +} + +func (x *ListResponseType) GetFileSize() int64 { + if x != nil { + return x.FileSize + } + return 0 +} + +func (x *ListResponseType) GetFileMode() uint32 { + if x != nil { + return x.FileMode + } + return 0 +} + +type FileRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + FileName string `protobuf:"bytes,1,opt,name=file_name,json=fileName,proto3" json:"file_name,omitempty"` + FileRange int64 `protobuf:"varint,2,opt,name=file_range,json=fileRange,proto3" json:"file_range,omitempty"` +} + +func (x *FileRequest) Reset() { + *x = FileRequest{} + if protoimpl.UnsafeEnabled { + mi := &file_proto_transport_proto_msgTypes[2] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *FileRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*FileRequest) ProtoMessage() {} + +func (x *FileRequest) ProtoReflect() protoreflect.Message { + mi := &file_proto_transport_proto_msgTypes[2] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use FileRequest.ProtoReflect.Descriptor instead. +func (*FileRequest) Descriptor() ([]byte, []int) { + return file_proto_transport_proto_rawDescGZIP(), []int{2} +} + +func (x *FileRequest) GetFileName() string { + if x != nil { + return x.FileName + } + return "" +} + +func (x *FileRequest) GetFileRange() int64 { + if x != nil { + return x.FileRange + } + return 0 +} + +type FileResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + Block []byte `protobuf:"bytes,1,opt,name=block,proto3" json:"block,omitempty"` + FileSize int64 `protobuf:"varint,2,opt,name=file_size,json=fileSize,proto3" json:"file_size,omitempty"` +} + +func (x *FileResponse) Reset() { + *x = FileResponse{} + if protoimpl.UnsafeEnabled { + mi := &file_proto_transport_proto_msgTypes[3] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *FileResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*FileResponse) ProtoMessage() {} + +func (x *FileResponse) ProtoReflect() protoreflect.Message { + mi := &file_proto_transport_proto_msgTypes[3] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use FileResponse.ProtoReflect.Descriptor instead. +func (*FileResponse) Descriptor() ([]byte, []int) { + return file_proto_transport_proto_rawDescGZIP(), []int{3} +} + +func (x *FileResponse) GetBlock() []byte { + if x != nil { + return x.Block + } + return nil +} + +func (x *FileResponse) GetFileSize() int64 { + if x != nil { + return x.FileSize + } + return 0 +} + +var File_proto_transport_proto protoreflect.FileDescriptor + +var file_proto_transport_proto_rawDesc = []byte{ + 0x0a, 0x15, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x70, 0x6f, 0x72, + 0x74, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x05, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x11, + 0x0a, 0x0f, 0x4c, 0x69, 0x73, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x54, 0x79, 0x70, + 0x65, 0x22, 0x69, 0x0a, 0x10, 0x4c, 0x69, 0x73, 0x74, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, + 0x65, 0x54, 0x79, 0x70, 0x65, 0x12, 0x1b, 0x0a, 0x09, 0x66, 0x69, 0x6c, 0x65, 0x5f, 0x6e, 0x61, + 0x6d, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x4e, 0x61, + 0x6d, 0x65, 0x12, 0x1b, 0x0a, 0x09, 0x66, 0x69, 0x6c, 0x65, 0x5f, 0x73, 0x69, 0x7a, 0x65, 0x18, + 0x02, 0x20, 0x01, 0x28, 0x03, 0x52, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x53, 0x69, 0x7a, 0x65, 0x12, + 0x1b, 0x0a, 0x09, 0x66, 0x69, 0x6c, 0x65, 0x5f, 0x6d, 0x6f, 0x64, 0x65, 0x18, 0x03, 0x20, 0x01, + 0x28, 0x0d, 0x52, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x4d, 0x6f, 0x64, 0x65, 0x22, 0x49, 0x0a, 0x0b, + 0x66, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x1b, 0x0a, 0x09, 0x66, + 0x69, 0x6c, 0x65, 0x5f, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, + 0x66, 0x69, 0x6c, 0x65, 0x4e, 0x61, 0x6d, 0x65, 0x12, 0x1d, 0x0a, 0x0a, 0x66, 0x69, 0x6c, 0x65, + 0x5f, 0x72, 0x61, 0x6e, 0x67, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x03, 0x52, 0x09, 0x66, 0x69, + 0x6c, 0x65, 0x52, 0x61, 0x6e, 0x67, 0x65, 0x22, 0x41, 0x0a, 0x0c, 0x66, 0x69, 0x6c, 0x65, 0x52, + 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x14, 0x0a, 0x05, 0x62, 0x6c, 0x6f, 0x63, 0x6b, + 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x05, 0x62, 0x6c, 0x6f, 0x63, 0x6b, 0x12, 0x1b, 0x0a, + 0x09, 0x66, 0x69, 0x6c, 0x65, 0x5f, 0x73, 0x69, 0x7a, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x03, + 0x52, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x53, 0x69, 0x7a, 0x65, 0x32, 0x46, 0x0a, 0x0b, 0x66, 0x69, + 0x6c, 0x65, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x37, 0x0a, 0x08, 0x44, 0x6f, 0x77, + 0x6e, 0x6c, 0x6f, 0x61, 0x64, 0x12, 0x12, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x66, 0x69, + 0x6c, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x13, 0x2e, 0x70, 0x72, 0x6f, 0x74, + 0x6f, 0x2e, 0x66, 0x69, 0x6c, 0x65, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, + 0x30, 0x01, 0x42, 0x09, 0x5a, 0x07, 0x2e, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x06, 0x70, + 0x72, 0x6f, 0x74, 0x6f, 0x33, +} + +var ( + file_proto_transport_proto_rawDescOnce sync.Once + file_proto_transport_proto_rawDescData = file_proto_transport_proto_rawDesc +) + +func file_proto_transport_proto_rawDescGZIP() []byte { + file_proto_transport_proto_rawDescOnce.Do(func() { + file_proto_transport_proto_rawDescData = protoimpl.X.CompressGZIP(file_proto_transport_proto_rawDescData) + }) + return file_proto_transport_proto_rawDescData +} + +var file_proto_transport_proto_msgTypes = make([]protoimpl.MessageInfo, 4) +var file_proto_transport_proto_goTypes = []interface{}{ + (*ListRequestType)(nil), // 0: proto.ListRequestType + (*ListResponseType)(nil), // 1: proto.ListResponseType + (*FileRequest)(nil), // 2: proto.fileRequest + (*FileResponse)(nil), // 3: proto.fileResponse +} +var file_proto_transport_proto_depIdxs = []int32{ + 2, // 0: proto.fileService.Download:input_type -> proto.fileRequest + 3, // 1: proto.fileService.Download:output_type -> proto.fileResponse + 1, // [1:2] is the sub-list for method output_type + 0, // [0:1] is the sub-list for method input_type + 0, // [0:0] is the sub-list for extension type_name + 0, // [0:0] is the sub-list for extension extendee + 0, // [0:0] is the sub-list for field type_name +} + +func init() { file_proto_transport_proto_init() } +func file_proto_transport_proto_init() { + if File_proto_transport_proto != nil { + return + } + if !protoimpl.UnsafeEnabled { + file_proto_transport_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*ListRequestType); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_proto_transport_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*ListResponseType); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_proto_transport_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*FileRequest); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_proto_transport_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} { + switch v := v.(*FileResponse); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + } + type x struct{} + out := protoimpl.TypeBuilder{ + File: protoimpl.DescBuilder{ + GoPackagePath: reflect.TypeOf(x{}).PkgPath(), + RawDescriptor: file_proto_transport_proto_rawDesc, + NumEnums: 0, + NumMessages: 4, + NumExtensions: 0, + NumServices: 1, + }, + GoTypes: file_proto_transport_proto_goTypes, + DependencyIndexes: file_proto_transport_proto_depIdxs, + MessageInfos: file_proto_transport_proto_msgTypes, + }.Build() + File_proto_transport_proto = out.File + file_proto_transport_proto_rawDesc = nil + file_proto_transport_proto_goTypes = nil + file_proto_transport_proto_depIdxs = nil +} + +// Reference imports to suppress errors if they are not otherwise used. +var _ context.Context +var _ grpc.ClientConnInterface + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the grpc package it is being compiled against. +const _ = grpc.SupportPackageIsVersion6 + +// FileServiceClient is the client API for FileService service. +// +// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://godoc.org/google.golang.org/grpc#ClientConn.NewStream. +type FileServiceClient interface { + Download(ctx context.Context, in *FileRequest, opts ...grpc.CallOption) (FileService_DownloadClient, error) +} + +type fileServiceClient struct { + cc grpc.ClientConnInterface +} + +func NewFileServiceClient(cc grpc.ClientConnInterface) FileServiceClient { + return &fileServiceClient{cc} +} + +func (c *fileServiceClient) Download(ctx context.Context, in *FileRequest, opts ...grpc.CallOption) (FileService_DownloadClient, error) { + stream, err := c.cc.NewStream(ctx, &_FileService_serviceDesc.Streams[0], "/proto.fileService/Download", opts...) + if err != nil { + return nil, err + } + x := &fileServiceDownloadClient{stream} + if err := x.ClientStream.SendMsg(in); err != nil { + return nil, err + } + if err := x.ClientStream.CloseSend(); err != nil { + return nil, err + } + return x, nil +} + +type FileService_DownloadClient interface { + Recv() (*FileResponse, error) + grpc.ClientStream +} + +type fileServiceDownloadClient struct { + grpc.ClientStream +} + +func (x *fileServiceDownloadClient) Recv() (*FileResponse, error) { + m := new(FileResponse) + if err := x.ClientStream.RecvMsg(m); err != nil { + return nil, err + } + return m, nil +} + +// FileServiceServer is the server API for FileService service. +type FileServiceServer interface { + Download(*FileRequest, FileService_DownloadServer) error +} + +// UnimplementedFileServiceServer can be embedded to have forward compatible implementations. +type UnimplementedFileServiceServer struct { +} + +func (*UnimplementedFileServiceServer) Download(*FileRequest, FileService_DownloadServer) error { + return status.Errorf(codes.Unimplemented, "method Download not implemented") +} + +func RegisterFileServiceServer(s *grpc.Server, srv FileServiceServer) { + s.RegisterService(&_FileService_serviceDesc, srv) +} + +func _FileService_Download_Handler(srv interface{}, stream grpc.ServerStream) error { + m := new(FileRequest) + if err := stream.RecvMsg(m); err != nil { + return err + } + return srv.(FileServiceServer).Download(m, &fileServiceDownloadServer{stream}) +} + +type FileService_DownloadServer interface { + Send(*FileResponse) error + grpc.ServerStream +} + +type fileServiceDownloadServer struct { + grpc.ServerStream +} + +func (x *fileServiceDownloadServer) Send(m *FileResponse) error { + return x.ServerStream.SendMsg(m) +} + +var _FileService_serviceDesc = grpc.ServiceDesc{ + ServiceName: "proto.fileService", + HandlerType: (*FileServiceServer)(nil), + Methods: []grpc.MethodDesc{}, + Streams: []grpc.StreamDesc{ + { + StreamName: "Download", + Handler: _FileService_Download_Handler, + ServerStreams: true, + }, + }, + Metadata: "proto/transport.proto", +} diff --git a/proto/transport.proto b/proto/transport.proto new file mode 100644 index 0000000..54ed814 --- /dev/null +++ b/proto/transport.proto @@ -0,0 +1,28 @@ +syntax = "proto3"; + +package proto; + +option go_package = "./proto"; + +message ListRequestType { +} + +message ListResponseType { + string file_name = 1; + int64 file_size = 2; + uint32 file_mode = 3; +} + +message fileRequest{ + string file_name = 1; + int64 file_range = 2; +} + +message fileResponse{ + bytes block = 1; + int64 file_size = 2; +} + +service fileService{ + rpc Download(fileRequest) returns (stream fileResponse){}; +} diff --git a/server/server.go b/server/server.go new file mode 100644 index 0000000..92087a1 --- /dev/null +++ b/server/server.go @@ -0,0 +1,96 @@ +/** + * Created by IntelliJ IDEA. + * User: Ar.M + * Date: 2020-05-10 + * Time: 22:13 + */ +package main + +import ( + "fmt" + "google.golang.org/grpc" + "io" + "log" + "net" + "os" + "path/filepath" + "runtime" + transportpb "yunlian-file-transfer-demo/proto" +) + +var filesDir string + +func init() { + _, fp, _, ok := runtime.Caller(0) + if !ok { + log.Fatalf("Get caller failed") + } + filesDir = filepath.Join(filepath.Dir(fp), "files") +} + +type server struct{} + +func (s *server) Download(req *transportpb.FileRequest, stream transportpb.FileService_DownloadServer) error { + fileName := req.GetFileName() + fileRange := req.GetFileRange() + + path := filepath.Join(filesDir, fileName) + + fileInfo, err := os.Stat(path) + if err != nil { + return err + } + fileSize := fileInfo.Size() + + f, err := os.Open(path) + if err != nil { + return err + } + defer f.Close() + + if fileRange > 0 { + _, err = f.Seek(fileRange, 0) + if err != nil { + return err + } + log.Printf("read file at seek: %d\n", fileRange) + } + + var totalBytesStreamed int64 + + for totalBytesStreamed < fileSize { + block := make([]byte, 1024) + bytesRead, err := f.Read(block) + if err == io.EOF { + break + } + + if err != nil { + return err + } + + if err := stream.Send(&transportpb.FileResponse{ + Block: block, + FileSize: fileSize, + }); err != nil { + return err + } + totalBytesStreamed += int64(bytesRead) + } + return nil +} + +func main() { + lis, err := net.Listen("tcp", "0.0.0.0:10086") + if err != nil { + log.Fatalf("Failed to listen on 10086 : %v\n", err) + } + + s := grpc.NewServer() + transportpb.RegisterFileServiceServer(s, &server{}) + + fmt.Println("Starting server on 10086") + if err := s.Serve(lis); err != nil { + log.Fatalf("failed to start server : %v\n", err) + } +}