modularization/src/driver/driver_cloud_grpc_server_stream/main.cpp

#include <QCoreApplication>
#include <QDateTime>
#include <iostream>
#include <vector>

#include "cumsgbuffer.h"
#include "pcmsgbuffer.h"

#include <iostream>
#include <memory>
#include <string>

#include <grpcpp/grpcpp.h>
#include <grpcpp/health_check_service_interface.h>
#include <grpcpp/ext/proto_server_reflection_plugin.h>

#include "uploadstreammsg.grpc.pb.h"

using grpc::Server;
using grpc::ServerBuilder;
using grpc::ServerContext;
using grpc::Status;



#include <QDateTime>



#include <thread>


static cumsgbuffer gcumsgbuf;
static pcmsgbuffer gpcmsgbuf;

char gstrport[255];

#include <getopt.h>

void print_useage()
{
    std::cout<<" -p --port $port : port . eq.  -p 50051"<<std::endl;
    std::cout<<" -h --help print help"<<std::endl;
}

int  GetOptLong(int argc, char *argv[]) {
    int nRtn = 0;
    int opt; // getopt_long() 的返回值
    int digit_optind = 0; // 设置短参数类型及是否需要参数

    // 如果option_index非空，它指向的变量将记录当前找到参数符合long_opts里的
    // 第几个元素的描述，即是long_opts的下标值
    int option_index = 0;
    // 设置短参数类型及是否需要参数
    const char *optstring = "m:r:x:y:o:p:s:h";


    static struct option long_options[] = {
        {"port", required_argument, NULL, 'p'},
        {"help",  no_argument,       NULL, 'h'},
 //       {"optarg", optional_argument, NULL, 'o'},
        {0, 0, 0, 0}  // 添加 {0, 0, 0, 0} 是为了防止输入空值
    };

    while ( (opt = getopt_long(argc,
                               argv,
                               optstring,
                               long_options,
                               &option_index)) != -1) {
        switch(opt)
        {

        case 'p':
            strncpy(gstrport,optarg,255);
            break;
        case 'h':
            print_useage();
            nRtn = 1; //because use -h
            break;
        default:
            break;
        }

    }

    return nRtn;
}



void uploadsend(::grpc::ServerReaderWriter<iv::UploadReplyStream, iv::UploadRequestStream>* stream,bool * pbrun,
                std::string * pstrvin,std::string * pstrmd5,bool *pbUpdatemd4orvin,QMutex * pmutex,int * preqid,qint64 * pnsendtime,qint64 * recvtime,QMutex * pmutexidtime)
{
    std::string strvin;
    std::string strmd5;
    pmutex->lock();
    strvin = *pstrmd5;
    strmd5 = *pstrvin;
    pmutex->unlock();

    QTime xTime;
    xTime.start();
    int nlastsend = xTime.elapsed();
    while(*pbrun)
    {
        if(*pbUpdatemd4orvin)
        {
            pmutex->lock();
            strvin = *pstrvin;
            strmd5 = *pstrmd5;
            *pbUpdatemd4orvin = false;
            pmutex->unlock();
        }

        int id;
        qint64 ntime;
        float framerate;

        std::vector<char > xvectorctrldata;

        int nres = gpcmsgbuf.getmsg(strvin,strmd5,id,ntime,&xvectorctrldata,&framerate);

        iv::UploadReplyStream reply;

        pmutexidtime->lock();
        reply.set_nreqsendtime(*pnsendtime);
        reply.set_nreqid(*preqid);
        reply.set_npausetime(QDateTime::currentMSecsSinceEpoch() - *recvtime);
        pmutexidtime->unlock();
        if(nres == 1)
        {
            reply.set_nres(nres);
            reply.set_xdata(xvectorctrldata.data(),xvectorctrldata.size());
            reply.set_framerate(framerate);
            stream->Write(reply);
            nlastsend = xTime.elapsed();
        }
        else
        {
            if(abs(xTime.elapsed() - nlastsend)>1000)
            {
                reply.set_nres(nres);
                reply.set_framerate(0);
                stream->Write(reply);

                nlastsend = xTime.elapsed();
            }
        }
        std::this_thread::sleep_for(std::chrono::milliseconds(1));


    }
}


qint64 calcserverlatency(std::vector<qint64> & xvectorserverlatency,qint64 nlatency)
{
    xvectorserverlatency.push_back(nlatency);
    while(xvectorserverlatency.size()>5)xvectorserverlatency.erase(xvectorserverlatency.begin());
    if(xvectorserverlatency.size()<1)return 0;
    int nsize = xvectorserverlatency.size();
    int i;
    qint64 nrtn = 0;
    for(i=0;i<nsize;i++)nrtn = nrtn +  xvectorserverlatency.at(i);
    return nrtn/nsize;
}

void queryctrlsend(::grpc::ServerReaderWriter<iv::queryReplyStream, iv::queryReqStream>* stream,bool * pbrun,
                std::string * pstrvin,std::string * pstrmd5,bool *pbUpdatemd4orvin,QMutex * pmutex,qint64 * pnserverlatency)
{
    std::string strvin;
    std::string strmd5;
    pmutex->lock();
    strvin = *pstrmd5;
    strmd5 = *pstrvin;
    pmutex->unlock();

    QTime xTime;
    xTime.start();
    int nlastsend = xTime.elapsed();
    qint64 nlastdatatime = 0;
    qint64 nculatency;



    while(*pbrun)
    {
        if(*pbUpdatemd4orvin)
        {
            pmutex->lock();
            strvin = *pstrvin;
            strmd5 = *pstrmd5;
            *pbUpdatemd4orvin = false;
            pmutex->unlock();
        }

        int id;
        qint64 ntime;
        std::vector<char > xvectorquerydata;

        int nres = gcumsgbuf.getmsg(strvin,strmd5,nlastdatatime,id,ntime,&xvectorquerydata,&nculatency);

        nlastdatatime = ntime;

        iv::queryReplyStream reply;
        reply.set_nres(nres);
        if(nres > 0)
        {

            reply.set_xdata(xvectorquerydata.data(),xvectorquerydata.size());
            reply.set_id(id);
            reply.set_ntime(ntime);
            reply.set_nculatency(nculatency);
            reply.set_nserverlatency(*pnserverlatency);
            reply.set_nsendtime(QDateTime::currentMSecsSinceEpoch());
            stream->Write(reply);
            nlastsend = xTime.elapsed();
        }
        else
        {
            if(abs(xTime.elapsed() - nlastsend)>1000)
            {
                stream->Write(reply);
                nlastsend = xTime.elapsed();
            }
        }


        std::this_thread::sleep_for(std::chrono::milliseconds(1));


    }
}


// Logic and data behind the server's behavior.
class UploadServiceImpl final : public iv::UploadStream::Service {
  Status upload(ServerContext* context, ::grpc::ServerReaderWriter<iv::UploadReplyStream, iv::UploadRequestStream>* stream) override {

      iv::UploadRequestStream request;
      bool brun = true;
      std::string strctrlmd5 = "md5";
      std::string strvin = "aaa";
      bool bUpdatemd4orvin = false;
      QMutex uploadmutex;
      int reqid = -1;
      qint64 pnsendtime = 0;
      qint64 nrecvtime = 0;
      QMutex mutexidtime;
      std::thread * pthread = new std::thread(uploadsend,stream,&brun,&strvin,&strctrlmd5,&bUpdatemd4orvin,&uploadmutex,
                                              &reqid,&pnsendtime,&nrecvtime,&mutexidtime);
      std::cout<<"new connect."<<std::endl;


      while (stream->Read(&request))
      {
          std::cout<<" rec req."<<std::endl;
          std::vector<char> xvectordata;
          qDebug("size is %d",request.xdata().size());
          if((strctrlmd5 != request.strctrlmd5())||(strvin != request.strvin()))
          {
              uploadmutex.lock();
              strctrlmd5 = request.strctrlmd5();
              strvin = request.strvin();
              bUpdatemd4orvin = true;
              uploadmutex.unlock();
          }
          if(request.xdata().size()>0)
          {

              xvectordata.resize(request.xdata().size());
              memcpy(xvectordata.data(),request.xdata().data(),request.xdata().size());
          }

          mutexidtime.lock();
          reqid = request.id();
          pnsendtime = request.nsendtime();
          nrecvtime = QDateTime::currentMSecsSinceEpoch();
          mutexidtime.unlock();

          gcumsgbuf.addmsg(request.id(),request.ntime(),request.strvin(),request.strquerymd5(),request.strctrlmd5(),
                           &xvectordata,request.bimportant(),request.kepptime(),request.nlatency());
//          std::cout << "收到请求，类型为" << request.askmsg() <<"\n"<<std::endl;
      }
      std::cout<<" no conn"<<std::endl;
      brun = false;
      pthread->join();
      std::cout<<"dis connect."<<std::endl;

    return Status::OK;
  }

  Status queryctrl(ServerContext* context, ::grpc::ServerReaderWriter<iv::queryReplyStream, iv::queryReqStream>* stream) override {

      iv::queryReqStream request;
      bool brun = true;
      std::string strctrlmd5 = "md5";
      std::string strquerymd5 = "md5";
      std::string strvin = "aaa";
      bool bUpdatemd4orvin = false;
      QMutex uploadmutex;
      qint64 nreplysendtime;
      qint64 npausetime;
      std::vector<qint64> xvectorserverlatency;
      qint64 nserverlatency = 0;
      std::thread * pthread = new std::thread(queryctrlsend,stream,&brun,&strvin,&strquerymd5,&bUpdatemd4orvin,&uploadmutex,&nserverlatency);
      std::cout<<"new connect."<<std::endl;
      while (stream->Read(&request))
      {
          std::cout<<" rec req."<<std::endl;
          std::vector<char> xvectordata;
          qDebug("size is %d",request.xdata().size());
          if((strquerymd5 != request.strquerymd5())||(strvin != request.strvin()))
          {
              uploadmutex.lock();
              strquerymd5 = request.strquerymd5();
              strvin = request.strvin();
              bUpdatemd4orvin = true;
              uploadmutex.unlock();
          }
          if(request.xdata().size()>0)
          {

              xvectordata.resize(request.xdata().size());
              memcpy(xvectordata.data(),request.xdata().data(),request.xdata().size());
          }
          nreplysendtime = request.nreplysendtime();
          npausetime = request.npausetime();
          if(nreplysendtime > 0)
          {
              nserverlatency =  calcserverlatency(xvectorserverlatency,QDateTime::currentMSecsSinceEpoch() - nreplysendtime - npausetime);
          }
          static int tempid = 0;
          tempid++;
          int nid = gpcmsgbuf.addmsg(tempid,request.ntime(),request.strvin(),request.strctrlmd5(),&xvectordata,
                                     request.bimportant(),request.kepptime(),request.nsuggestframerate());

          (void)&nid;

//          std::cout << "收到请求，类型为" << request.askmsg() <<"\n"<<std::endl;
      }
      std::cout<<" no conn"<<std::endl;
      brun = false;
      pthread->join();
      std::cout<<"dis connect."<<std::endl;
    return Status::OK;
  }

//  Status query(ServerContext* context, const iv::queryreq* request,
//                  iv::queryReply* reply) override {

//      int id;
//      qint64 ntime;
//      std::vector<char > xvectorquerydata;

//      int nres = gcumsgbuf.getmsg(request->strvin(),request->strquerymd5(),request->nlasttime(),id,ntime,&xvectorquerydata);

//      reply->set_nres(nres);
//      if(nres > 0)
//      {
//          reply->set_data(xvectorquerydata.data(),xvectorquerydata.size());
//          reply->set_id(id);
//          reply->set_ntime(ntime);
//      }

//      return Status::OK;


//  }

//  Status ctrl(ServerContext* context, const iv::ctrlreq* request,
//                  iv::ctrlReply * reply) override {

//      std::vector<char> xvectordata;
//      if(request->data().size()>0)
//      {
//          xvectordata.resize(request->data().size());
//          memcpy(xvectordata.data(),request->data().data(),request->data().size());
//      }

//      int nid = gpcmsgbuf.addmsg(request->id(),request->ntime(),request->strvin(),request->strctrlmd5(),&xvectordata,
//                                 request->bimportant(),request->kepptime());

//      reply->set_nsendid(nid);


//      return Status::OK;


//  }


};

void RunServer() {
  std::string server_address("0.0.0.0:");
  server_address = server_address.append(gstrport);
  UploadServiceImpl service;

  grpc::EnableDefaultHealthCheckService(true);
//  grpc::reflection::InitProtoReflectionServerBuilderPlugin();
  ServerBuilder builder;
  // Listen on the given address without any authentication mechanism.
  builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());

  builder.SetMaxReceiveMessageSize(300000000);
 // builder.SetMaxMessageSize(100000000);


//  builder.SetMaxSendMessageSize(100000000);
  // Register "service" as the instance through which we'll communicate with
  // clients. In this case it corresponds to an *synchronous* service.
  builder.RegisterService(&service);
  // Finally assemble the server.
  std::unique_ptr<Server> server(builder.BuildAndStart());
  std::cout << "Server listening on " << server_address << std::endl;

  // Wait for the server to shutdown. Note that some other thread must be
  // responsible for shutting down the server for this call to ever return.
  server->Wait();
}





int main(int argc, char *argv[])
{
    QCoreApplication a(argc, argv);

    snprintf(gstrport,255,"50051");

    int nRtn = GetOptLong(argc,argv);
    if(nRtn == 1)  //show help,so exit.
    {
        return 0;
    }

    gpcmsgbuf.start();

    RunServer();




    return a.exec();
}