patch 4.0

unittest/observer/rpc/easy_server_basic.cpp

@@ -0,0 +1,166 @@
+/**
+ * Copyright (c) 2021 OceanBase
+ * OceanBase CE is licensed under Mulan PubL v2.
+ * You can use this software according to the terms and conditions of the Mulan PubL v2.
+ * You may obtain a copy of Mulan PubL v2 at:
+ *          http://license.coscl.org.cn/MulanPubL-2.0
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PubL v2 for more details.
+ */
+
+#define USING_LOG_PREFIX SERVER
+#include "rpc/frame/ob_net_easy.h"
+#include "observer/mysql/obsm_handler.h"
+
+
+using namespace oceanbase;
+using namespace oceanbase::observer;
+
+
+class MyReqHandler : public rpc::frame::ObReqHandler
+{
+public:
+  MyReqHandler() {
+    EZ_ADD_CB(decode);
+    EZ_ADD_CB(encode);
+    EZ_ADD_CB(process);
+    EZ_ADD_CB(on_connect);
+    EZ_ADD_CB(on_disconnect);
+    EZ_ADD_CB(cleanup);
+  }
+  void *decode(easy_message_t *m) override;
+  int process(easy_request_t *r) override;
+};
+
+class MyReq
+{
+public:
+  int len_ = 0;
+  char *data_ = nullptr;
+  easy_request_t *ez_r_ = nullptr;
+};
+
+void *MyReqHandler::decode(easy_message_t *m)
+{
+  // alloc packet
+  MyReq *req;
+  if ((req = (MyReq *)easy_pool_calloc(m->pool,
+                  sizeof(MyReq))) == NULL) {
+    m->status = EASY_ERROR;
+    return NULL;
+  }
+  int len = m->input->last - m->input->pos;
+  //std::string msg(m->input->pos, len);
+  LOG_INFO("decode", K(req), K(len), K(m->c->doing_request_count));
+  req->data_ = m->input->pos;
+  req->len_ = len;
+  m->input->last = m->input->pos;
+  return req;
+}
+
+void gen_http_resp(std::string &ret, std::string &output)
+{
+  ret.append("HTTP/1.1");
+  ret.append(" 200 OK\r\n",9);
+
+  ret.append("Content-Length: ",16);
+  ret.append(std::to_string(output.size()));
+  ret.append("\r\n",2);
+  //ret.append("Content-Type: text/plain; charset=utf-8\r\n",41);
+  ret.append("Connection: keep-alive\r\n\r\n",26);
+  ret.append(output);
+}
+
+int MyReqHandler::process(easy_request_t *r)
+{
+  int ret = OB_SUCCESS;
+  easy_buf_t *b;
+  b = easy_buf_create(r->ms->pool, 1024);
+
+  std::string resp, output;
+  gen_http_resp(resp, output);
+  LOG_INFO("process", K(resp.size()));
+  b->last = easy_memcpy(b->last, resp.c_str(), resp.size());
+  easy_request_addbuf(r, b);
+  return ret;
+}
+
+void thread_start(void *args)
+{
+  UNUSED(args);
+  lib::set_thread_name("MysqlIO");
+}
+
+class EasyServer
+{
+public:
+  EasyServer() : eio_(nullptr){}
+
+  int init(int port = 8080, int io_cnt = 1) {
+    int ret = OB_SUCCESS;
+    if (OB_ISNULL(eio_ = easy_eio_create(eio_, io_cnt))) {
+      ret = OB_LIBEASY_ERROR;
+    } else if (OB_ISNULL(easy_connection_add_listen(eio_, NULL, port, req_handler_.ez_handler()))) {
+      ret = OB_SERVER_LISTEN_ERROR;
+      LOG_ERROR("easy_connection_add_listen error", K(port), KERRMSG, K(ret));
+    } else {
+      easy_eio_set_uthread_start(eio_, thread_start, nullptr);
+      int eret = easy_eio_start(eio_);
+      if (EASY_OK == eret) {
+        LOG_INFO("start mysql easy io");
+      } else {
+        ret = OB_LIBEASY_ERROR;
+        LOG_ERROR("start mysql easy io fail", K(ret));
+      }
+    }
+    return ret;
+  }
+
+private:
+  easy_io_t *eio_;
+  MyReqHandler req_handler_;
+};
+
+int main(int argc, char **argv)
+{
+  int c = 0;
+  int time_sec = 3600;
+  char *log_level = (char*)"DEBUG";
+  int port = 8080;
+  int worker = 1;
+  while(EOF != (c = getopt(argc,argv,"p:c:t:l:"))) {
+    switch(c) {
+    case 'p':
+      port = atoi(optarg);
+      break;
+    case 'c':
+      worker = atoi(optarg);
+      break;
+    case 't':
+      time_sec = atoi(optarg);
+      break;
+    case 'l':
+     log_level = optarg;
+     break;
+    default:
+      break;
+    }
+  }
+
+  system("rm -rf easy_server_basic.log*");
+  OB_LOGGER.set_file_name("easy_server_basic.log", true);
+  OB_LOGGER.set_log_level(log_level);
+  OB_LOGGER.set_enable_async_log(true);
+
+  int ret = OB_SUCCESS;
+  EasyServer server;
+  if (OB_FAIL(server.init(port, worker))) {
+    LOG_ERROR("server init fail", K(ret));
+    return -1;
+  }
+
+  ::sleep(time_sec);
+  return 0;
+}