From 7ba5854ad1bc770a03d620cea78bafdc62dfdf24 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期五, 19 六月 2020 18:16:31 +0800 Subject: [PATCH] update --- service/netdisk_service.c | 65 +++++++++++++++++++++----------- 1 files changed, 43 insertions(+), 22 deletions(-) diff --git a/service/netdisk_service.c b/service/netdisk_service.c index bed648c..1f55786 100644 --- a/service/netdisk_service.c +++ b/service/netdisk_service.c @@ -17,7 +17,6 @@ int WORKERS ; -std::string localUrl; std::string remoteUrl; PropertiesConfig config("../data/config.txt"); @@ -69,43 +68,57 @@ int work(Netdisk_DownloadRequest drequest) { Netdisk *netdisk = NULL; + std::vector<std::string> files; int rv; + char rmsg[MAXLINE]; + strcpy(rmsg, "success"); + Netdisk_LoginInfo loginInfo = loginStore.getLoginInfo(drequest.loginUUID); std::map<std::string, Netdisk *>::iterator userDeviceIter = userDeviceMap.find( drequest.loginUUID); if( userDeviceIter != userDeviceMap.end() ) { netdisk = userDeviceIter->second; } - Netdisk_LoginInfo loginInfo = loginStore.getLoginInfo(drequest.loginUUID); if (netdisk == NULL) { netdisk = NetdiskFacotory::create(loginInfo.deviceType); if(netdisk != NULL) { userDeviceMap.insert({loginInfo.loginUUID, netdisk}); } else { - err_msg(0, "鏃犳硶璇嗗埆鐨勮澶囩被鍨嬶細 %s", loginInfo.deviceType.c_str()); + snprintf(rmsg, MAXLINE, "鏃犳硶璇嗗埆鐨勮澶囩被鍨嬶細 %s", loginInfo.deviceType.c_str()); } } if ( (rv = netdisk->login(loginInfo)) != 0 ) { - printf("涓嬭浇鐧诲綍澶辫触\n"); + snprintf(rmsg, MAXLINE, "璇烽噸鏂扮櫥褰�"); + } else if ( (rv = netdisk->downloadByTime(drequest, &files) ) != 0) { + snprintf(rmsg, MAXLINE, "涓嬭浇澶辫触"); } + - std::vector<std::string> files; - if ( (rv = netdisk->downloadByTime(drequest, &files) ) != 0) { - printf("涓嬭浇澶辫触\n"); - } + // Json::Value response; + // Json::Value payload; + // response["code"] = rv; + // response["msg"] = rmsg; - Json::Value response; - Json::Value payload; - response["rv"] = rv; + // 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; Json::Value filelist; for(std::string f : files) { filelist.append(f); } - payload["filelist"] = filelist; - response["payload"] = payload; - std::string str = response.toStyledString(); + arguments["fileList"] = filelist; + arguments["loginUUID"] = drequest.loginUUID; + + request["arguments"] = arguments; + std::string str = request.toStyledString(); std::cout << "download finished, call back" << std::endl; std::cout << str << std::endl; @@ -121,6 +134,7 @@ { Netdisk_DownloadRequest request; task_queue.pop(request); + err_msg(0, "====take a task"); work(request); } } @@ -150,6 +164,7 @@ } jsonreader.parse(buf, request); + nng_free(buf, sz); std::string method = request["method"].asString(); std::map<std::string, RequestHandleFun>::iterator handleFunIter = requestHandleFunMap.find(method); @@ -160,15 +175,20 @@ std::cerr << "Don't support " << method << std::endl; } - nng_free(buf, sz); + } } - +/** + * 娉ㄥ唽璇锋眰澶勭悊鐨勬柟娉� + */ void registRequestHandleFun() { requestHandleFunMap.insert({"login", handleLogin}); + requestHandleFunMap.insert({"logout", handleLogout}); requestHandleFunMap.insert({"downloadByTime", handleDownloadByTimeAsync}); + requestHandleFunMap.insert({"getDeviceInfo", handleGetDeviceInfo}); } + void initThreadPool() { pthread_t tid; @@ -178,21 +198,22 @@ int main() { + //鐜鍙橀噺鍒濆鍖� + WORKERS = config.getInt("workers"); + remoteUrl = config.get("client_url"); + //娴峰悍璁惧鐜鍒濆鍖� Netdisk_EnvConfig hcEnvConfig; - hcEnvConfig.libpath = config.get("hclib"); + hcEnvConfig.libpath = "../lib/hc"; HCNetdisk::netdisk_init(&hcEnvConfig); - WORKERS = config.getInt("workers"); - localUrl = config.get("local_url"); - remoteUrl = config.get("remote_url"); - + registRequestHandleFun(); initThreadPool(); - startServer(localUrl.c_str()); + startServer(config.get("server_url").c_str()); HCNetdisk::netdisk_deinit(); -- Gitblit v1.8.0