Add CustomModule for RPC and Http.

Author: holmes1412

src/http/http_client.cc

@@ -131,6 +131,9 @@ void HttpClient::add_filter(RPCFilter *filter)
 			case RPCModuleTypeMetrics:
 				module = new HttpMetricsModule();
 				break;
+			case RPCModuleTypeCustom:
+				module = new HttpCustomModule();
+				break;
 			default:
 				break;
 			}

src/http/http_module.h

@@ -41,6 +41,8 @@ class HttpMetricsModule : public MetricsModule
 	bool server_begin(SubTask *task, RPCModuleData& data) override;
 };
 
+using HttpCustomModule = MetricsModule;
+
 } // end namespace srpc
 
 #endif

src/http/http_server.cc

@@ -47,6 +47,9 @@ void HttpServer::add_filter(RPCFilter *filter)
 			case RPCModuleTypeMetrics:
 				module = new HttpMetricsModule();
 				break;
+			case RPCModuleTypeCustom:
+				module = new HttpCustomModule();
+				break;
 			default:
 				break;
 			}

src/module/rpc_module.cc

@@ -135,6 +135,26 @@ bool SnowFlake::get_id(long long group_id, long long machine_id,
 	return true;
 }
 
+bool CustomModule::client_begin(SubTask *task, RPCModuleData& data)
+{
+	return true;
+}
+
+bool CustomModule::client_end(SubTask *task, RPCModuleData& data)
+{
+	return true;
+}
+
+bool CustomModule::server_begin(SubTask *task, RPCModuleData& data)
+{
+	return true;
+}
+
+bool CustomModule::server_end(SubTask *task, RPCModuleData& data)
+{
+	return true;
+}
+
 bool http_set_header_module_data(const RPCModuleData& data,
 								 protocol::HttpMessage *msg)
 {

src/module/rpc_module.h

@@ -116,6 +116,23 @@ class SnowFlake
 	long long machine_shift;
 };
 
+class CustomModule : public RPCModule
+{
+public:
+	bool client_begin(SubTask *task, RPCModuleData& data) override;
+	bool client_end(SubTask *task, RPCModuleData& data) override;
+	bool server_begin(SubTask *task, RPCModuleData& data) override;
+	bool server_end(SubTask *task, RPCModuleData& data) override;
+
+public:
+	CustomModule() : RPCModule(RPCModuleTypeCustom) { }
+};
+
+template<class SERVER_TASK, class CLIENT_TASK>
+class RPCCustomModule : public CustomModule
+{
+};
+
 bool http_get_header_module_data(const protocol::HttpMessage *msg,
 								 RPCModuleData& data);
 

src/rpc_basic.h

@@ -143,6 +143,7 @@ enum RPCModuleType
 	RPCModuleTypeTrace		=	1,
 	RPCModuleTypeMetrics	=	2,
 	RPCModuleTypeLog		=	3,
+	RPCModuleTypeCustom		=	4,
 };
 
 class RPCCommon

src/rpc_client.h

@@ -154,6 +154,9 @@ void RPCClient<RPCTYPE>::add_filter(RPCFilter *filter)
 			case RPCModuleTypeMetrics:
 				module = new RPCMetricsModule<SERVER_TASK, CLIENT_TASK>();
 				break;
+			case RPCModuleTypeCustom:
+				module = new RPCCustomModule<SERVER_TASK, CLIENT_TASK>();
+				break;
 			default:
 				break;
 			}

src/rpc_server.h

@@ -165,6 +165,9 @@ void RPCServer<RPCTYPE>::add_filter(RPCFilter *filter)
 			case RPCModuleTypeMetrics:
 				module = new RPCMetricsModule<SERVER_TASK, CLIENT_TASK>();
 				break;
+			case RPCModuleTypeCustom:
+				module = new RPCCustomModule<SERVER_TASK, CLIENT_TASK>();
+				break;
 			default:
 				break;
 			}