update

wangzhengquan

2020-09-10 591aacee97f4a6486631c38a6b418e20b2c4109c

 service/netdisk_service.c

@@ -9,6 +9,8 @@
#include <nng/nng.h>
#include <nng/protocol/reqrep0/rep.h>
#include <nng/protocol/reqrep0/req.h>
#include <nng/protocol/survey0/survey.h>
#include <nng/protocol/survey0/respond.h>

using namespace std;

@@ -17,7 +19,6 @@


int  WORKERS ;
std::string localUrl;
std::string remoteUrl;

PropertiesConfig config("../data/config.txt");
@@ -96,18 +97,6 @@
  }
 

  // Json::Value response;
  // Json::Value payload;
  // response["code"] = rv;
  // response["msg"] = rmsg;

  // Json::Value filelist;
  // for(std::string f : files) {
  //   filelist.append(f);
  // }
  // payload["filelist"] = filelist;
  // response["payload"] = payload;

  Json::Value request;
  request["method"] = "downloadByTimeCallBack";
  Json::Value arguments;
@@ -121,8 +110,7 @@
  request["arguments"] = arguments;
  std::string str = request.toStyledString();

  std::cout << "download finished, call back" << std::endl;
  std::cout << str << std::endl;
  std::cout << "SENDING download finished\n" << str << std::endl;
  connectAndSend(remoteUrl.c_str(), strdup(str.c_str()) );
  
  return 0;
@@ -163,7 +151,7 @@
    if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0) {
      fatal("nng_recv", rv);
    }

printf("RECEIVED RPC REQUEST:\n %s", buf);
    jsonreader.parse(buf, request);
    nng_free(buf, sz);
    std::string method = request["method"].asString();
@@ -197,25 +185,64 @@
    pthread_create(&tid, NULL, worker, NULL);
}


void heartBeat(const char *url, const char *name)
{
  nng_socket sock;
  int rv;

  if ((rv = nng_respondent0_open(&sock)) != 0) {
    fatal("nng_respondent0_open", rv);
  }
  if ((rv = nng_dial(sock, url, NULL, NNG_FLAG_NONBLOCK)) != 0) {
    fatal("nng_dial", rv);
  }
  for (;;) {
    char *buf = NULL;
    size_t sz;
    if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) == 0) {
//printf("CLIENT (%s): RECEIVED \"%s\" SURVEY REQUEST\n", name, buf);
      nng_free(buf, sz);
      char response[1024];
      sprintf(response, "%s-%d", name, getpid());
//printf("CLIENT (%s): SENDING SURVEY RESPONSE:%s\n", name, response);
      if ((rv = nng_send(sock, response, strlen(response) + 1, 0)) != 0) {
        fatal("nng_send", rv);
      }
    }
  }
}

void *heart(void *vargp)
{
  pthread_detach(pthread_self());
  heartBeat(config.get("heart_server").c_str(), "netdisk");
  return NULL;
}

// 心跳发送进程
void initHeart() {
  pthread_t tid;
  pthread_create(&tid, NULL, heart, NULL);
}

int main()
{
  //环境变量初始化
  WORKERS = config.getInt("workers");
  localUrl = config.get("local_url");
  remoteUrl = config.get("remote_url");
  remoteUrl = config.get("client_url");

  //海康设备环境初始化
  Netdisk_EnvConfig hcEnvConfig;
  hcEnvConfig.libpath = "../lib/hc";
  HCNetdisk::netdisk_init(&hcEnvConfig);

 

  registRequestHandleFun();

  initThreadPool();
  initHeart();

  startServer(localUrl.c_str());
  startServer(config.get("server_url").c_str());


  HCNetdisk::netdisk_deinit();