hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[FEAT MERGE] [v4.2] add table generator and several random utility functions

Browse Source
This commit is contained in:
raywill
2023-04-13 08:22:22 +00:00
committed by ob-robot
parent 94159e6675
commit 27488211d2
34 changed files with 1659 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

194
src/sql/engine/expr/ob_expr_rand.cpp Normal file
View File

@ -0,0 +1,194 @@
/**
* 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 SQL_ENG
#include "lib/time/ob_time_utility.h" /* time */
#include "sql/engine/expr/ob_expr_rand.h"
#include "sql/engine/ob_exec_context.h"
namespace oceanbase
{
using namespace common;
namespace sql
{
OB_SERIALIZE_MEMBER((ObExprRand, ObFuncExprOperator), is_seed_const_);
const uint64_t ObExprRand::ObExprRandCtx::max_value_ = 0x3FFFFFFFL;
ObExprRand::ObExprRandCtx::ObExprRandCtx()
: seed1_(1), seed2_(1)
{
}
ObExprRand::ObExprRandCtx::~ObExprRandCtx()
{
}
void ObExprRand::ObExprRandCtx::set_seed(uint32_t seed)
{
seed1_ = static_cast<uint32_t>(seed * 0x10001L + 55555555L) % max_value_;
seed2_ = static_cast<uint32_t>(seed * 0x10000001L) % max_value_;
}
void ObExprRand::ObExprRandCtx::get_next_random(double &res)
{
seed1_ = (seed1_ * 3 + seed2_) % max_value_;
seed2_ = (seed1_ + seed2_ + 33) % max_value_;
res = static_cast<double> (seed1_) / static_cast<double> (max_value_);
}
ObExprRand::ObExprRand(common::ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_RAND, "rand", ZERO_OR_ONE, NOT_ROW_DIMENSION),
is_seed_const_(true)
{
}
ObExprRand::~ObExprRand()
{
}
int ObExprRand::calc_result_typeN(ObExprResType &type,
ObExprResType *types,
int64_t param_num,
common::ObExprTypeCtx &type_ctx) const
{
UNUSED(types);
UNUSED(type_ctx);
int ret = OB_SUCCESS;
if (param_num > 1) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid number of arguments", K(param_num));
} else {
if(param_num == 1) {
types[0].set_calc_type(ObIntType);
}
type.set_double();
}
return ret;
}
int ObExprRand::assign(const ObExprOperator &other)
{
int ret = OB_SUCCESS;
const ObExprRand *tmp_other = dynamic_cast<const ObExprRand*>(&other);
if (OB_UNLIKELY(NULL == tmp_other)) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument. wrong type for other", K(ret), K(other));
} else if (OB_LIKELY(this != tmp_other)) {
if (OB_FAIL(ObExprOperator::assign(other))) {
LOG_WARN("copy in Base class ObExprOperator failed", K(ret));
} else {
this->is_seed_const_ = tmp_other->is_seed_const_;
}
}
return ret;
}
int ObExprRand::calc_random_expr_const_seed(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res_datum)
{
int ret = OB_SUCCESS;
ObDatum *seed_datum = NULL;
if (OB_UNLIKELY(0 != expr.arg_cnt_ && 1 != expr.arg_cnt_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected arg_cnt", K(ret), K(expr.arg_cnt_));
} else if (1 == expr.arg_cnt_ && OB_FAIL(expr.eval_param_value(ctx, seed_datum))) {
LOG_WARN("expr.eval_param_value failed", K(ret));
} else {
uint64_t op_id = expr.expr_ctx_id_;
ObExecContext &exec_ctx = ctx.exec_ctx_;
ObExprRandCtx *random_ctx = NULL;
if (OB_ISNULL(random_ctx = static_cast<ObExprRandCtx *>(
exec_ctx.get_expr_op_ctx(op_id)))) {
if (OB_FAIL(exec_ctx.create_expr_op_ctx(op_id, random_ctx))) {
LOG_WARN("failed to create operator ctx", K(ret), K(op_id));
} else {
uint32_t seed = 0;
if (expr.arg_cnt_ == 1) {
if(!seed_datum->is_null()) {
seed = static_cast<uint32_t> (seed_datum->get_int());
}
} else {
// use timestamp as the seed for rand expression
seed = static_cast<uint32_t> (ObTimeUtility::current_time());
}
random_ctx->set_seed(seed);
}
}
if (OB_SUCC(ret)) {
if (OB_ISNULL(random_ctx)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("random ctx is NULL", K(ret));
} else {
double rand_res = 0.0;
random_ctx->get_next_random(rand_res);
res_datum.set_double(rand_res);
}
}
}
return ret;
}
int ObExprRand::calc_random_expr_nonconst_seed(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res_datum)
{
int ret = OB_SUCCESS;
ObDatum *seed_datum = NULL;
if (OB_UNLIKELY(1 != expr.arg_cnt_)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("unexpected arg_cnt", K(ret), K(expr.arg_cnt_));
} else if (OB_FAIL(expr.eval_param_value(ctx, seed_datum))) {
LOG_WARN("expr.eval_param_value failed", K(ret));
} else {
uint64_t op_id = expr.expr_ctx_id_;
ObExecContext &exec_ctx = ctx.exec_ctx_;
ObExprRandCtx *random_ctx = NULL;
if (OB_ISNULL(random_ctx = static_cast<ObExprRandCtx *>(
exec_ctx.get_expr_op_ctx(op_id)))) {
if (OB_FAIL(exec_ctx.create_expr_op_ctx(op_id, random_ctx))) {
LOG_WARN("failed to create operator ctx", K(ret), K(op_id));
}
}
if (OB_SUCC(ret)) {
if (OB_ISNULL(random_ctx)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("random ctx is NULL", K(ret));
} else {
uint32_t seed = 0;
if(!seed_datum->is_null()) {
seed = static_cast<uint32_t>(seed_datum->get_int());
}
double rand_res = 0.0;
random_ctx->set_seed(seed);
random_ctx->get_next_random(rand_res);
res_datum.set_double(rand_res);
}
}
}
return ret;
}
int ObExprRand::cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
int ret = OB_SUCCESS;
UNUSED(expr_cg_ctx);
UNUSED(raw_expr);
if (is_seed_const_) {
rt_expr.eval_func_ = ObExprRand::calc_random_expr_const_seed;
} else {
rt_expr.eval_func_ = ObExprRand::calc_random_expr_nonconst_seed;
}
return ret;
}
} /* namespace sql */
} /* namespace oceanbase */