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

patch 4.0

Browse Source
This commit is contained in:
wangzelin.wzl
2022-10-24 10:34:53 +08:00
parent 4ad6e00ec3
commit 93a1074b0c
10533 changed files with 2588271 additions and 2299373 deletions
Download Patch File Download Diff File Expand all files Collapse all files

198
src/sql/engine/expr/ob_expr_random.cpp
View File

@ -10,122 +10,76 @@
* See the Mulan PubL v2 for more details.
*/
#define USING_LOG_PREFIX SQL_ENG
#define USING_LOG_PREFIX SQL_ENG
#include "lib/time/ob_time_utility.h" /* time */
#include "sql/engine/expr/ob_expr_random.h"
#include "sql/engine/ob_exec_context.h"
namespace oceanbase {
namespace oceanbase
{
using namespace common;
namespace sql {
namespace sql
{
OB_SERIALIZE_MEMBER((ObExprRandom, ObFuncExprOperator), is_seed_const_);
const uint64_t ObExprRandom::ObExprRandomCtx::max_value_ = 0x3FFFFFFFL;
ObExprRandom::ObExprRandomCtx::ObExprRandomCtx() : seed1_(1), seed2_(1)
{}
ObExprRandom::ObExprRandomCtx::ObExprRandomCtx()
: seed1_(1), seed2_(1)
{
}
ObExprRandom::ObExprRandomCtx::~ObExprRandomCtx()
{}
{
}
void ObExprRandom::ObExprRandomCtx::set_seed(uint32_t seed)
{
seed1_ = static_cast<uint32_t>(seed * 0x10001L + 55555555L) % max_value_;
seed2_ = static_cast<uint32_t>(seed * 0x10000001L) % max_value_;
seed1_ = static_cast<uint32_t>(seed * 0x10001L + 55555555L) % max_value_;
seed2_ = static_cast<uint32_t>(seed * 0x10000001L) % max_value_;
}
void ObExprRandom::ObExprRandomCtx::get_next_random(double& res)
void ObExprRandom::ObExprRandomCtx::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_);
seed1_ = (seed1_ * 3 + seed2_) % max_value_;
seed2_ = (seed1_ + seed2_ + 33) % max_value_;
res = static_cast<double> (seed1_) / static_cast<double> (max_value_);
}
ObExprRandom::ObExprRandom(common::ObIAllocator& alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_RAND, "rand", ZERO_OR_ONE, NOT_ROW_DIMENSION), is_seed_const_(true)
{}
ObExprRandom::ObExprRandom(common::ObIAllocator &alloc)
: ObFuncExprOperator(alloc, T_FUN_SYS_RAND, "rand", ZERO_OR_ONE, NOT_ROW_DIMENSION),
is_seed_const_(true)
{
}
ObExprRandom::~ObExprRandom()
{}
int ObExprRandom::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 ObExprRandom::calc_resultN(
common::ObObj& result, const common::ObObj* objs, int64_t param_num, common::ObExprCtx& expr_ctx) const
int ObExprRandom::calc_result_typeN(ObExprResType &type,
ObExprResType *types,
int64_t param_num,
common::ObExprTypeCtx &type_ctx) const
{
int ret = OB_SUCCESS;
if (OB_ISNULL(expr_ctx.calc_buf_)) {
ret = OB_NOT_INIT;
LOG_WARN("varchar buffer not init", K(ret));
} else if (OB_ISNULL(expr_ctx.exec_ctx_)) {
ret = OB_NOT_INIT;
LOG_WARN("execution ctx not init", K(ret));
} else if (param_num > 1) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid number of arguments", K(param_num));
} else {
ObExecContext* exec_ctx = expr_ctx.exec_ctx_;
ObExprRandomCtx* random_ctx = NULL;
if (NULL == (random_ctx = static_cast<ObExprRandomCtx*>(exec_ctx->get_expr_op_ctx(get_id())))) {
// create random ctx if it is the first time to calculate random expression
if (OB_FAIL(exec_ctx->create_expr_op_ctx(get_id(), random_ctx))) {
LOG_WARN("failed to create operator ctx", K(ret), K(get_id()));
} else {
if (is_seed_const_) {
uint32_t seed = 0;
if (param_num == 1) {
if (!objs[0].is_null()) {
TYPE_CHECK(objs[0], ObIntType);
seed = static_cast<uint32_t>(objs[0].get_int());
}
} else {
// use timestamp as the seed for rand expression
seed = static_cast<uint32_t>(ObTimeUtility::current_time());
}
random_ctx->set_seed(seed);
}
}
}
// set the random seed if necessary and generate random number
if (OB_SUCC(ret) && OB_NOT_NULL(random_ctx)) {
double rand_res = 0.0;
if (is_seed_const_) {
random_ctx->get_next_random(rand_res);
} else {
uint32_t seed = 0;
if (!objs[0].is_null()) {
TYPE_CHECK(objs[0], ObIntType);
seed = static_cast<uint32_t>(objs[0].get_int());
}
random_ctx->set_seed(seed);
random_ctx->get_next_random(rand_res);
}
result.set_double(rand_res);
}
}
return ret;
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 ObExprRandom::assign(const ObExprOperator& other)
int ObExprRandom::assign(const ObExprOperator &other)
{
int ret = OB_SUCCESS;
const ObExprRandom* tmp_other = dynamic_cast<const ObExprRandom*>(&other);
const ObExprRandom *tmp_other = dynamic_cast<const ObExprRandom*>(&other);
if (OB_UNLIKELY(NULL == tmp_other)) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument. wrong type for other", K(ret), K(other));
@ -139,10 +93,11 @@ int ObExprRandom::assign(const ObExprOperator& other)
return ret;
}
int ObExprRandom::calc_random_expr_const_seed(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res_datum)
int ObExprRandom::calc_random_expr_const_seed(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res_datum)
{
int ret = OB_SUCCESS;
ObDatum* seed_datum = NULL;
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_));
@ -150,25 +105,26 @@ int ObExprRandom::calc_random_expr_const_seed(const ObExpr& expr, ObEvalCtx& ctx
LOG_WARN("expr.eval_param_value failed", K(ret));
} else {
uint64_t op_id = expr.expr_ctx_id_;
ObExecContext& exec_ctx = ctx.exec_ctx_;
ObExprRandomCtx* random_ctx = NULL;
if (OB_ISNULL(random_ctx = static_cast<ObExprRandomCtx*>(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 {
ObExecContext &exec_ctx = ctx.exec_ctx_;
ObExprRandomCtx *random_ctx = NULL;
if (OB_ISNULL(random_ctx = static_cast<ObExprRandomCtx *>(
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());
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());
seed = static_cast<uint32_t> (ObTimeUtility::current_time());
}
random_ctx->set_seed(seed);
}
}
if (OB_SUCC(ret)) {
}
}
if (OB_SUCC(ret)) {
if (OB_ISNULL(random_ctx)) {
ret = OB_ERR_UNEXPECTED;
LOG_WARN("random ctx is NULL", K(ret));
@ -177,15 +133,16 @@ int ObExprRandom::calc_random_expr_const_seed(const ObExpr& expr, ObEvalCtx& ctx
random_ctx->get_next_random(rand_res);
res_datum.set_double(rand_res);
}
}
}
}
return ret;
}
int ObExprRandom::calc_random_expr_nonconst_seed(const ObExpr& expr, ObEvalCtx& ctx, ObDatum& res_datum)
int ObExprRandom::calc_random_expr_nonconst_seed(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res_datum)
{
int ret = OB_SUCCESS;
ObDatum* seed_datum = NULL;
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_));
@ -193,33 +150,35 @@ int ObExprRandom::calc_random_expr_nonconst_seed(const ObExpr& expr, ObEvalCtx&
LOG_WARN("expr.eval_param_value failed", K(ret));
} else {
uint64_t op_id = expr.expr_ctx_id_;
ObExecContext& exec_ctx = ctx.exec_ctx_;
ObExprRandomCtx* random_ctx = NULL;
if (OB_ISNULL(random_ctx = static_cast<ObExprRandomCtx*>(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)) {
ObExecContext &exec_ctx = ctx.exec_ctx_;
ObExprRandomCtx *random_ctx = NULL;
if (OB_ISNULL(random_ctx = static_cast<ObExprRandomCtx *>(
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());
}
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->set_seed(seed);
random_ctx->get_next_random(rand_res);
res_datum.set_double(rand_res);
}
}
}
}
return ret;
}
int ObExprRandom::cg_expr(ObExprCGCtx& expr_cg_ctx, const ObRawExpr& raw_expr, ObExpr& rt_expr) const
int ObExprRandom::cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
int ret = OB_SUCCESS;
UNUSED(expr_cg_ctx);
@ -233,3 +192,4 @@ int ObExprRandom::cg_expr(ObExprCGCtx& expr_cg_ctx, const ObRawExpr& raw_expr, O
}
} /* namespace sql */
} /* namespace oceanbase */