liuxiaolong
2019-08-29 ba842438f497fb4985444de2173ca4e55dea3d64
add detect rpc client
1个文件已添加
35 ■■■■■ 已修改文件
service/FDetectClient.go 35 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
service/FDetectClient.go
New file
@@ -0,0 +1,35 @@
package service
import (
    "basic.com/pubsub/protomsg.git"
    "context"
    "errors"
    "time"
    "webserver/extend/logger"
    "google.golang.org/grpc"
)
const (
    address = "localhost:4009"
)
func DoFDetectWithRpc(fdr *protomsg.FaceDetectRequest,t time.Duration) ([]*protomsg.ResultFaceDetect,error) {
    conn,err := grpc.Dial(address, grpc.WithInsecure())
    if err !=nil {
        logger.Debug("grpc dial err:",err)
    }
    defer conn.Close()
    c := protomsg.NewFaceDetectClient(conn)
    ctx,cancel := context.WithTimeout(context.Background(), t)
    defer cancel()
    faces,err := c.FaceDetect(ctx,fdr)
    if err !=nil {
        logger.Debug("rpc call err:",err)
        return nil,err
    }
    if faces !=nil {
        return faces.Faces,nil
    }
    return nil,errors.New("no face detected")
}