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

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

View File

@ -0,0 +1,56 @@
/**
* 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.
*/
#ifndef OCEANBASE_SQL_OB_EXPR_FUNC_NORMAL_H_
#define OCEANBASE_SQL_OB_EXPR_FUNC_NORMAL_H_
#include <random>
#include "sql/engine/expr/ob_expr_operator.h"
namespace oceanbase
{
namespace sql
{
class ObExprNormal : public ObFuncExprOperator
{
class ObExprNormalCtx: public ObExprOperatorCtx
{
public:
ObExprNormalCtx() = default;
~ObExprNormalCtx() = default;
int initialize(ObEvalCtx &ctx, const ObExpr &expr);
int generate_next_value(int64_t sample, double &result);
private:
std::normal_distribution<double> normal_dist_;
std::mt19937_64 gen_; // map continuous small number to large sparse space
};
public:
explicit ObExprNormal(common::ObIAllocator &alloc);
virtual ~ObExprNormal();
virtual int calc_result_type3(ObExprResType &result_type,
ObExprResType &mean,
ObExprResType &stddev,
ObExprResType &rand_expr,
common::ObExprTypeCtx &type_ctx) const;
virtual bool need_rt_ctx() const override { return true; }
virtual int cg_expr(ObExprCGCtx &expr_cg_ctx,
const ObRawExpr &raw_expr,
ObExpr &rt_expr) const override;
static int eval_next_value(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &res_datum);
private:
DISALLOW_COPY_AND_ASSIGN(ObExprNormal);
};
} /* namespace sql */
} /* namespace oceanbase */
#endif /* OCEANBASE_SQL_OB_EXPR_FUNC_NORMAL_H_ */