Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
@MaybeShewill-CV
MaybeShewill-CV committed Mar 19, 2024
1 parent 7c98719 commit e0f9c8c
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 15 deletions.
12 changes: 7 additions & 5 deletions src/server/scene_segmentation/bisenetv2_server.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
#include "rapidjson/writer.h"
#include "workflow/WFTaskFactory.h"
#include "workflow/WFHttpServer.h"
#include "workflow/Workflow.h"

#include "common/base64.h"
#include "common/status_code.h"
Expand Down Expand Up @@ -120,6 +119,7 @@ StatusCode BiseNetV2Server::Impl::init(const decltype(toml::parse("")) &config)
peer_resp_timeout = static_cast<int>(server_section.at("peer_resp_timeout").as_integer()) * 1000;
compute_threads = static_cast<int>(server_section.at("compute_threads").as_integer());
handler_threads = static_cast<int>(server_section.at("handler_threads").as_integer());
request_size_limit = static_cast<size_t>(server_section.at("request_size_limit").as_integer());

_m_successfully_initialized = true;
LOG(INFO) << "bisenetv2 segmentation server init successfully";
Expand Down Expand Up @@ -212,12 +212,14 @@ jinq::common::StatusCode BiseNetV2Server::init(const decltype(toml::parse("")) &
WFGlobalSettings settings = GLOBAL_SETTINGS_DEFAULT;
settings.compute_threads = _m_impl->compute_threads;
settings.handler_threads = _m_impl->handler_threads;
settings.endpoint_params.max_connections = _m_impl->max_connection_nums;
settings.endpoint_params.response_timeout = _m_impl->peer_resp_timeout;
WORKFLOW_library_init(&settings);

auto&& proc = std::bind(
&BiseNetV2Server::Impl::serve_process, std::cref(this->_m_impl), std::placeholders::_1);
WFServerParams server_params = SERVER_PARAMS_DEFAULT;
server_params.max_connections = _m_impl->max_connection_nums;
server_params.peer_response_timeout = _m_impl->peer_resp_timeout;
server_params.request_size_limit = _m_impl->request_size_limit * 1024 * 1024;

auto&& proc = [&](auto arg) { return this->_m_impl->serve_process(arg); };
_m_server = std::make_unique<WFHttpServer>(proc);

return StatusCode::OK;
Expand Down
12 changes: 7 additions & 5 deletions src/server/scene_segmentation/hrnet_server.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
#include "rapidjson/writer.h"
#include "workflow/WFTaskFactory.h"
#include "workflow/WFHttpServer.h"
#include "workflow/Workflow.h"

#include "common/base64.h"
#include "common/status_code.h"
Expand Down Expand Up @@ -120,6 +119,7 @@ StatusCode HRNetServer::Impl::init(const decltype(toml::parse("")) &config) {
peer_resp_timeout = static_cast<int>(server_section.at("peer_resp_timeout").as_integer()) * 1000;
compute_threads = static_cast<int>(server_section.at("compute_threads").as_integer());
handler_threads = static_cast<int>(server_section.at("handler_threads").as_integer());
request_size_limit = static_cast<size_t>(server_section.at("request_size_limit").as_integer());

_m_successfully_initialized = true;
LOG(INFO) << "hrnet segmentation server init successfully";
Expand Down Expand Up @@ -212,12 +212,14 @@ jinq::common::StatusCode HRNetServer::init(const decltype(toml::parse("")) &conf
WFGlobalSettings settings = GLOBAL_SETTINGS_DEFAULT;
settings.compute_threads = _m_impl->compute_threads;
settings.handler_threads = _m_impl->handler_threads;
settings.endpoint_params.max_connections = _m_impl->max_connection_nums;
settings.endpoint_params.response_timeout = _m_impl->peer_resp_timeout;
WORKFLOW_library_init(&settings);

auto&& proc = std::bind(
&HRNetServer::Impl::serve_process, std::cref(this->_m_impl), std::placeholders::_1);
WFServerParams server_params = SERVER_PARAMS_DEFAULT;
server_params.max_connections = _m_impl->max_connection_nums;
server_params.peer_response_timeout = _m_impl->peer_resp_timeout;
server_params.request_size_limit = _m_impl->request_size_limit * 1024 * 1024;

auto&& proc = [&](auto arg) { return this->_m_impl->serve_process(arg); };
_m_server = std::make_unique<WFHttpServer>(proc);

return StatusCode::OK;
Expand Down
12 changes: 7 additions & 5 deletions src/server/scene_segmentation/pphuman_seg_server.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
#include "rapidjson/writer.h"
#include "workflow/WFTaskFactory.h"
#include "workflow/WFHttpServer.h"
#include "workflow/Workflow.h"

#include "common/base64.h"
#include "common/status_code.h"
Expand Down Expand Up @@ -120,6 +119,7 @@ StatusCode PPHumanSegServer::Impl::init(const decltype(toml::parse("")) &config)
peer_resp_timeout = static_cast<int>(server_section.at("peer_resp_timeout").as_integer()) * 1000;
compute_threads = static_cast<int>(server_section.at("compute_threads").as_integer());
handler_threads = static_cast<int>(server_section.at("handler_threads").as_integer());
request_size_limit = static_cast<size_t>(server_section.at("request_size_limit").as_integer());

_m_successfully_initialized = true;
LOG(INFO) << "pphuman segmentation server init successfully";
Expand Down Expand Up @@ -212,12 +212,14 @@ jinq::common::StatusCode PPHumanSegServer::init(const decltype(toml::parse(""))
WFGlobalSettings settings = GLOBAL_SETTINGS_DEFAULT;
settings.compute_threads = _m_impl->compute_threads;
settings.handler_threads = _m_impl->handler_threads;
settings.endpoint_params.max_connections = _m_impl->max_connection_nums;
settings.endpoint_params.response_timeout = _m_impl->peer_resp_timeout;
WORKFLOW_library_init(&settings);

auto&& proc = std::bind(
&PPHumanSegServer::Impl::serve_process, std::cref(this->_m_impl), std::placeholders::_1);
WFServerParams server_params = SERVER_PARAMS_DEFAULT;
server_params.max_connections = _m_impl->max_connection_nums;
server_params.peer_response_timeout = _m_impl->peer_resp_timeout;
server_params.request_size_limit = _m_impl->request_size_limit * 1024 * 1024;

auto&& proc = [&](auto arg) { return this->_m_impl->serve_process(arg); };
_m_server = std::make_unique<WFHttpServer>(proc);

return StatusCode::OK;
Expand Down

0 comments on commit e0f9c8c

Please sign in to comment.