add tars-client

Author: ruanshudong

tars-client/CMakeLists.txt

@@ -0,0 +1,30 @@
+
+include_directories(${CMAKE_SOURCE_DIR}/tarscpp/util/include)
+include_directories(${CMAKE_SOURCE_DIR}/tarscpp/util/src/epoll_windows)
+include_directories(${CMAKE_SOURCE_DIR}/tarscpp/servant/protocol)
+
+set(MODULE tars-client)
+
+aux_source_directory(. DIR_SRC)
+
+add_executable(${MODULE} ${DIR_SRC})
+
+add_dependencies(${MODULE} tarsservant tarsutil)
+
+target_link_libraries(${MODULE} tarsservant tarsutil)
+
+if(TARS_SSL)
+    target_link_libraries(${MODULE} tarsservant tarsutil ${LIB_SSL} ${LIB_CRYPTO})
+
+    if(WIN32)
+        target_link_libraries(${MODULE} Crypt32)
+    endif()
+endif()
+
+if(TARS_HTTP2)
+    target_link_libraries(${MODULE} ${LIB_HTTP2})
+endif()
+
+target_link_libraries(tars-client ${LIB_MYSQL})
+
+install(TARGETS tars-client DESTINATION deploy)

tars-client/TarsClient.cpp

@@ -0,0 +1,41 @@
+
+#include "TarsClient.h"
+#include "util/tc_port.h"
+
+TarsClient::TarsClient(TC_Option &option) : _option(option)
+{
+	_comm = new Communicator();
+
+	string locator;
+
+	if(option.hasParam("TARS_REGISTRY_HOST"))
+	{
+		locator = option.getValue("TARS_REGISTRY_HOST");
+	}
+	else 
+	{
+		locator = TC_Port::getEnv("TARS_REGISTRY_HOST");
+	}
+
+	if(locator.empty())
+	{
+		locator = "tars.tarsregistry.QueryObj@tcp -h 127.0.0.1 -p 17890";
+	}
+	else
+	{
+		locator = "tars.tarsregistry.QueryObj@tcp -h " + locator + " -p 17890";
+	}
+	_comm->setProperty("locator", locator);
+}
+
+void TarsClient::call(const string &command)
+{
+	cout << "call:" << command << endl;
+	_commands[command](this, _option);
+}
+
+TarsClient::~TarsClient()
+{
+	delete _comm;
+	_comm = NULL;
+}

tars-client/TarsClient.h

@@ -0,0 +1,37 @@
+#ifndef TARS_CLIENT
+#define TARS_CLIENT
+
+#include "servant/Application.h"
+#include "framework/AdminReg.h"
+#include <functional>
+
+
+class TarsClient
+{
+public:
+	typedef std::function<void(TarsClient*, TC_Option&)> command_function;
+
+	TarsClient(TC_Option &option);
+
+    ~TarsClient();
+	
+	void add(const string &command, command_function func)
+    {
+        _commands[command] = func;
+    }
+
+    Communicator *getCommunicator() { return _comm; }
+
+    void call(const string &command);
+
+protected:
+
+    TC_Option _option;
+
+    map<string, command_function> _commands;
+
+	Communicator *_comm;
+};
+
+
+#endif 

tars-client/command/ping.h

@@ -0,0 +1,27 @@
+#ifndef CMD_PING
+#define CMD_PING
+
+#include "util/tc_option.h"
+#include "TarsClient.h"
+
+using namespace tars;
+
+void ping(TarsClient* tarsClient, TC_Option& option)
+{
+    string obj = option.getValue("obj");
+
+    cout << "ping:" << obj << endl;
+
+    try
+    {
+        ServantPrx prx = tarsClient->getCommunicator()->stringToProxy<ServantPrx>(obj);
+        prx->tars_ping();
+        cout << "ping succ" << endl;
+    }
+    catch(exception &ex)
+    {
+        cout << "ping err:" << ex.what() << endl;
+    }
+}
+
+#endif 

tars-client/command/start.h

@@ -0,0 +1,36 @@
+#ifndef CMD_PING
+#define CMD_PING
+
+#include "util/tc_option.h"
+#include "TarsClient.h"
+
+using namespace tars;
+
+void start(TarsClient* tarsClient, TC_Option& option)
+{
+    string app = option.getValue("app");
+    string serverName = option.getValue("server-name");
+    string nodeName = option.getValue("node-name");
+
+    cout << "start:" << app << "." << serverName << endl;
+
+    AdminRegPrx adminPrx = tarsClient->getCommunicator()->stringToProxy<AdminRegPrx>("tars.tarsAdminRegistry.AdminRegObj");
+
+    try
+    {
+        // if(nodeName.empty())
+        // {
+        //     adminPrx->getAllServerIds();
+        // }
+
+        // ServantPrx prx = tarsClient->getCommunicator()->stringToProxy<ServantPrx>(obj);
+        // prx->tars_ping();
+        cout << "start succ" << endl;
+    }
+    catch(exception &ex)
+    {
+        cout << "start err:" << ex.what() << endl;
+    }
+}
+
+#endif 

tars-client/main.cpp

@@ -0,0 +1,57 @@
+/**
+ * Tencent is pleased to support the open source community by making Tars available.
+ *
+ * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
+ *
+ * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except 
+ * in compliance with the License. You may obtain a copy of the License at
+ *
+ * https://opensource.org/licenses/BSD-3-Clause
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed 
+ * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 
+ * CONDITIONS OF ANY KIND, either express or implied. See the License for the 
+ * specific language governing permissions and limitations under the License.
+ */
+
+#include "util/tc_mysql.h"
+#include "util/tc_option.h"
+#include "util/tc_file.h"
+#include "util/tc_common.h"
+#include "util/tc_config.h"
+#include <iostream>
+#include <string>
+#include "TarsClient.h"
+#include "command/ping.h"
+
+using namespace tars;
+using namespace std;
+
+int main(int argc, char *argv[])
+{
+	TC_Option option;
+
+    try
+    {
+		option.decode(argc, argv);
+
+		TarsClient tarsClient(option);
+
+		tarsClient.add("ping", ping);
+		tarsClient.add("start", TarsClient::command_function());
+		tarsClient.add("stop", TarsClient::command_function());
+		tarsClient.add("restart", TarsClient::command_function());
+		tarsClient.add("patch", TarsClient::command_function());
+
+		tarsClient.call("ping");
+    }
+    catch(exception &ex)
+    {
+		cout << "error: " << ex.what() << endl;
+        exit(-1);
+    }
+
+    return 0;
+}
+
+