139 lines
3.4 KiB
C++
139 lines
3.4 KiB
C++
/**
|
|
* 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.
|
|
*/
|
|
|
|
#include <cstdlib>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "lib/allocator/ob_malloc.h"
|
|
#include "lib/net/ob_addr.h"
|
|
|
|
#include "ob_log_mysql_connector.h"
|
|
|
|
using namespace oceanbase;
|
|
using namespace common;
|
|
using namespace liboblog;
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace unittest
|
|
{
|
|
|
|
class PrintRow : public MySQLQueryBase
|
|
{
|
|
public:
|
|
PrintRow()
|
|
{
|
|
sql_ = "select table_id, partition_id, ip, port, role "
|
|
"from __all_root_table";
|
|
sql_len_ = strlen(sql_);
|
|
}
|
|
int print_row()
|
|
{
|
|
int ret = common::OB_SUCCESS;
|
|
while (common::OB_SUCCESS == (ret = next_row())) {
|
|
uint64_t table_id = 0;
|
|
int32_t partition_id = 0;
|
|
ObAddr addr;
|
|
if (OB_SUCC(ret)) {
|
|
if (common::OB_SUCCESS != (ret = get_uint(0, table_id))) {
|
|
OBLOG_LOG(WARN, "err get uint", K(ret));
|
|
}
|
|
}
|
|
if (OB_SUCC(ret)) {
|
|
int64_t val = 0;
|
|
if (common::OB_SUCCESS != (ret = get_int(1, val))) {
|
|
OBLOG_LOG(WARN, "err get int", K(ret));
|
|
} else {
|
|
partition_id = static_cast<int32_t>(val);
|
|
}
|
|
}
|
|
if (OB_SUCC(ret)) {
|
|
ObString ip_str;
|
|
int64_t port = 0;
|
|
if (common::OB_SUCCESS != (ret = get_varchar(2, ip_str))) {
|
|
OBLOG_LOG(WARN, "err get var char", K(ret));
|
|
} else if (common::OB_SUCCESS != (ret = get_int(3, port))) {
|
|
OBLOG_LOG(WARN, "err get int", K(ret));
|
|
} else {
|
|
addr.set_ip_addr(ip_str, static_cast<int32_t>(port));
|
|
}
|
|
}
|
|
// Print values.
|
|
if (OB_SUCC(ret)) {
|
|
OBLOG_LOG(INFO, "\n>>>", K(table_id),
|
|
K(partition_id),
|
|
K(addr));
|
|
}
|
|
}
|
|
ret = (common::OB_ITER_END == ret) ? common::OB_SUCCESS : ret;
|
|
return ret;
|
|
}
|
|
};
|
|
|
|
class CreateTable : public MySQLQueryBase
|
|
{
|
|
public:
|
|
CreateTable(const char *tname)
|
|
{
|
|
snprintf(buf_, 512, "create table %s(c1 int primary key)", tname);
|
|
sql_ = buf_;
|
|
sql_len_ = strlen(sql_);
|
|
}
|
|
private:
|
|
char buf_[512];
|
|
};
|
|
|
|
TEST(MySQLConnector, run)
|
|
{
|
|
ConnectorConfig cfg;
|
|
cfg.mysql_addr_ = "10.210.177.162";
|
|
cfg.mysql_port_ = 26556;
|
|
cfg.mysql_user_ = "root";
|
|
cfg.mysql_password_ = "";
|
|
cfg.mysql_db_ = "oceanbase";
|
|
cfg.mysql_timeout_ = 100;
|
|
|
|
ObLogMySQLConnector conn;
|
|
|
|
int ret = conn.init(cfg);
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
|
|
// Print rows.
|
|
PrintRow pr;
|
|
ret = conn.query(pr);
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
ret = pr.print_row();
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
|
|
// Create dup tables.
|
|
CreateTable ct("table_1");
|
|
ret = conn.exec(ct);
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
ret = conn.exec(ct);
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
|
|
ret = conn.destroy();
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
oceanbase::common::ObLogger::get_logger().set_log_level("debug");
|
|
testing::InitGoogleTest(&argc,argv);
|
|
testing::FLAGS_gtest_filter = "DO_NOT_RUN";
|
|
return RUN_ALL_TESTS();
|
|
}
|