diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt index c3ebfca8bb1fa55f07fac80a431c56e2df7c27d7..5ca389d7538190f29bbb35652b07d9e6bb30a4f1 100644 --- a/src/CMakeLists.txt +++ b/src/CMakeLists.txt @@ -69,7 +69,6 @@ if (GRPC_CONNECTOR) ) add_subdirectory(cpputils) - add_subdirectory(websocket) list(APPEND SHARED_SRCS ${CPPUTILS_SRCS}) else() @@ -153,7 +152,6 @@ add_executable(isulad ${PLUGIN_SRCS} ${CONFIG_SRCS} ${CMAKE_CURRENT_SOURCE_DIR}/libisulad.c - ${WEBSOCKET_SERVICE_SRCS} ) target_include_directories(isulad PUBLIC @@ -167,7 +165,6 @@ target_include_directories(isulad PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}/plugin ${CMAKE_CURRENT_SOURCE_DIR}/config ${CMAKE_CURRENT_SOURCE_DIR}/http - ${WEBSOCKET_SERVICE_INCS} ) target_link_libraries(isulad ${LIBYAJL_LIBRARY} ${SYSTEMD_LIBRARY} ${SELINUX_LIBRARY} ${LIBARCHIVE_LIBRARY} ${LIBTAR_LIBRARY} ${WEBSOCKET_LIBRARY} ${CRYPTO_LIBRARY}) diff --git a/src/services/CMakeLists.txt b/src/services/CMakeLists.txt index d4b3be0af7ff8199f73738ee95d53e5fb5c915be..61779d5e5d6db48fc3c2c351f706da2340db632f 100644 --- a/src/services/CMakeLists.txt +++ b/src/services/CMakeLists.txt @@ -3,9 +3,9 @@ aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR}/ services_top_srcs) set(local_services_srcs ${services_top_srcs}) set(local_services_incs ${CMAKE_CURRENT_SOURCE_DIR}) if (GRPC_CONNECTOR) - add_subdirectory(cri) - list(APPEND local_services_srcs ${CRI_SRCS}) - list(APPEND local_services_incs ${CMAKE_CURRENT_SOURCE_DIR}/cri) + add_subdirectory(cri) + list(APPEND local_services_srcs ${CRI_SRCS}) + list(APPEND local_services_incs ${CRI_INCS}) endif() add_subdirectory(execution) list(APPEND local_services_srcs ${EXECUTION_SRCS}) diff --git a/src/services/cri/CMakeLists.txt b/src/services/cri/CMakeLists.txt index 5679e69581e1a1b7f0d8dfea67e913e22d3666ea..27e529bffc3b02464de8e5961f9dd1e6e4275918 100644 --- a/src/services/cri/CMakeLists.txt +++ b/src/services/cri/CMakeLists.txt @@ -1,7 +1,12 @@ # get current directory sources files -aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR} local_cri_srcs) +aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR} cri_top_srcs) +set(local_cri_srcs ${cri_top_srcs}) +set(local_cri_incs ${CMAKE_CURRENT_SOURCE_DIR}) + +add_subdirectory(websocket) +list(APPEND local_cri_srcs ${WEBSOCKET_SERVICE_SRCS}) +list(APPEND local_cri_incs ${WEBSOCKET_SERVICE_INCS}) + +set(CRI_SRCS ${local_cri_srcs} PARENT_SCOPE) +set(CRI_INCS ${local_cri_incs} PARENT_SCOPE) -set(CRI_SRCS - ${local_cri_srcs} - PARENT_SCOPE - ) diff --git a/src/websocket/CMakeLists.txt b/src/services/cri/websocket/CMakeLists.txt similarity index 100% rename from src/websocket/CMakeLists.txt rename to src/services/cri/websocket/CMakeLists.txt diff --git a/src/websocket/service/CMakeLists.txt b/src/services/cri/websocket/service/CMakeLists.txt similarity index 100% rename from src/websocket/service/CMakeLists.txt rename to src/services/cri/websocket/service/CMakeLists.txt diff --git a/src/websocket/service/attach_serve.cc b/src/services/cri/websocket/service/attach_serve.cc similarity index 100% rename from src/websocket/service/attach_serve.cc rename to src/services/cri/websocket/service/attach_serve.cc diff --git a/src/websocket/service/attach_serve.h b/src/services/cri/websocket/service/attach_serve.h similarity index 100% rename from src/websocket/service/attach_serve.h rename to src/services/cri/websocket/service/attach_serve.h diff --git a/src/websocket/service/exec_serve.cc b/src/services/cri/websocket/service/exec_serve.cc similarity index 100% rename from src/websocket/service/exec_serve.cc rename to src/services/cri/websocket/service/exec_serve.cc diff --git a/src/websocket/service/exec_serve.h b/src/services/cri/websocket/service/exec_serve.h similarity index 100% rename from src/websocket/service/exec_serve.h rename to src/services/cri/websocket/service/exec_serve.h diff --git a/src/websocket/service/route_callback_register.h b/src/services/cri/websocket/service/route_callback_register.h similarity index 100% rename from src/websocket/service/route_callback_register.h rename to src/services/cri/websocket/service/route_callback_register.h diff --git a/src/websocket/service/stream_server.cc b/src/services/cri/websocket/service/stream_server.cc similarity index 100% rename from src/websocket/service/stream_server.cc rename to src/services/cri/websocket/service/stream_server.cc diff --git a/src/websocket/service/stream_server.h b/src/services/cri/websocket/service/stream_server.h similarity index 100% rename from src/websocket/service/stream_server.h rename to src/services/cri/websocket/service/stream_server.h diff --git a/src/websocket/service/ws_server.cc b/src/services/cri/websocket/service/ws_server.cc similarity index 99% rename from src/websocket/service/ws_server.cc rename to src/services/cri/websocket/service/ws_server.cc index bd1f39bc87d8dc1797d4e77edb85cd3c30ddba68..86572fd3ad56c04eefd4409f4e3672e1caed064f 100644 --- a/src/websocket/service/ws_server.cc +++ b/src/services/cri/websocket/service/ws_server.cc @@ -29,7 +29,7 @@ struct lws_context *WebsocketServer::m_context = nullptr; std::atomic WebsocketServer::m_instance; std::mutex WebsocketServer::m_mutex; -std::map WebsocketServer::m_wsis; +std::unordered_map WebsocketServer::m_wsis; WebsocketServer *WebsocketServer::GetInstance() noexcept { WebsocketServer *server = m_instance.load(std::memory_order_relaxed); @@ -62,7 +62,7 @@ url::URLDatum WebsocketServer::GetWebsocketUrl() return m_url; } -std::map &WebsocketServer::GetWsisData() +std::unordered_map &WebsocketServer::GetWsisData() { return m_wsis; } diff --git a/src/websocket/service/ws_server.h b/src/services/cri/websocket/service/ws_server.h similarity index 95% rename from src/websocket/service/ws_server.h rename to src/services/cri/websocket/service/ws_server.h index 35f0c5ffdeba0265295e0eeebd66c1578a42e9de..08f633ff724e89daadd90575c1ec9f3d7c2350a2 100644 --- a/src/websocket/service/ws_server.h +++ b/src/services/cri/websocket/service/ws_server.h @@ -16,7 +16,7 @@ #ifndef __WEBSOCKET_SERVER_H_ #define __WEBSOCKET_SERVER_H_ #include -#include +#include #include #include #include @@ -71,7 +71,7 @@ public: void Shutdown(); void RegisterCallback(const std::string &path, std::shared_ptr callback); url::URLDatum GetWebsocketUrl(); - std::map &GetWsisData(); + std::unordered_map &GetWsisData(); void SetLwsSendedFlag(struct lws *wsi, bool sended); void LockAllWsSession(); void UnlockAllWsSession(); @@ -104,7 +104,7 @@ private: { NULL, NULL, 0, 0 } }; RouteCallbackRegister m_handler; - static std::map m_wsis; + static std::unordered_map m_wsis; url::URLDatum m_url; int m_listenPort; };