filetext_grpc.go 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. package main
  2. import (
  3. "context"
  4. "google.golang.org/grpc"
  5. "proto"
  6. qu "qfw/util"
  7. )
  8. var FileTextClient proto.FileExtractClient
  9. //直接调用接口地址
  10. //func InitFileTextGrpcClient() {
  11. // defer qu.Catch()
  12. // qu.Debug("address:", qu.ObjToString(Config["ocrserveraddr"])) //192.168.3.13:22101
  13. // conn, err := grpc.Dial(qu.ObjToString(Config["ocrserveraddr"]), grpc.WithTransportCredentials(insecure.NewCredentials()))
  14. // if err != nil {
  15. // qu.Debug("Init Connect Error: ", err)
  16. // }
  17. // FileTextClient = proto.NewFileExtractClient(conn)
  18. //}
  19. func GetFileText_back(fileName, fileUrl, fileType string, fileBytes []byte) (*proto.FileResponse, error) {
  20. defer qu.Catch()
  21. req := &proto.Request{
  22. FileName: fileName,
  23. FileUrl: fileUrl,
  24. FileType: fileType,
  25. FileBytes: fileBytes,
  26. ExtractType: 1,
  27. ReturnType: 0, //正式环境改为0
  28. }
  29. fileReq := &proto.FileRequest{
  30. Message: []*proto.Request{
  31. req,
  32. },
  33. }
  34. resp, err := FileTextClient.FileExtract(context.Background(), fileReq)
  35. if err != nil {
  36. return nil, err
  37. }
  38. qu.Debug(resp.Result)
  39. return resp, nil
  40. }
  41. func GetFileText(conn *grpc.ClientConn, fileName, fileUrl, fileType string, fileBytes []byte) *proto.FileResponse {
  42. defer qu.Catch()
  43. FileTextClient = proto.NewFileExtractClient(conn)
  44. defer conn.Close()
  45. req := &proto.Request{
  46. FileName: fileName,
  47. FileUrl: fileUrl,
  48. FileType: fileType,
  49. FileBytes: fileBytes,
  50. ExtractType: 1,
  51. ReturnType: 0, //正式环境改为0
  52. }
  53. fileReq := &proto.FileRequest{
  54. Message: []*proto.Request{
  55. req,
  56. },
  57. }
  58. resp, err := FileTextClient.FileExtract(context.Background(), fileReq)
  59. if err != nil {
  60. return nil
  61. }
  62. //qu.Debug(resp.Result)
  63. return resp
  64. }