C++ ObExprObj类代码示例

TEST_F(ObObjCastTest, bool_to_xxx)
{
    ObExprObj in;
    bool cases[] = {true, false};
    const char *varchar_expected[] = {"true", "false"};
    const char *dec_expected[] = {"1", "0"};
    for (int64_t i = 0; i < static_cast<int64_t>(ARRAYSIZEOF(cases)); ++i)
    {
        bool v = cases[i];
        in.set_bool(v);

        MY_EXPECT(ObBoolType, in, ObNullType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObIntType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObFloatType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObDoubleType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObDateTimeType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObPreciseDateTimeType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObVarcharType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObSeqType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObCreateTimeType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObModifyTimeType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObExtendType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObBoolType, v, varchar_expected[i]);
        MY_EXPECT(ObBoolType, in, ObDecimalType, v, dec_expected[i]);
    }
}

 static int double_mtime(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObDoubleType);
     out.set_mtime(static_cast<ObModifyTime> (in.get_double()));
     return OB_SUCCESS;
 }

 static int bool_ctime(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObBoolType);
     out.set_ctime(static_cast<ObCreateTime> (in.get_bool()));
     return OB_SUCCESS;
 }

 static int int_datetime(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObIntType);
     out.set_datetime(static_cast<ObDateTime> (in.get_int()));
     return OB_SUCCESS;
 }

 static int double_int(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObDoubleType);
     out.set_int(static_cast<int64_t> (in.get_double()));
     return OB_SUCCESS;
 }

 static int ctime_float(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObCreateTimeType);
     out.set_float(static_cast<float> (in.get_ctime()));
     return OB_SUCCESS;
 }

 static int ctime_datetime(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObCreateTimeType);
     out.set_datetime(static_cast<ObDateTime> (in.get_ctime() / 1000000L));
     return OB_SUCCESS;
 }

 static int float_double(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObFloatType);
     out.set_double(static_cast<double> (in.get_float()));
     return OB_SUCCESS;
 }

 int obj_cast(const ObObj &orig_cell, const ObObj &expected_type,
         ObObj &casted_cell, const ObObj *&res_cell)
 {
     int ret = OB_SUCCESS;
     if (orig_cell.get_type() != expected_type.get_type())
     {
         // type cast
         ObObjCastParams params;
         ObExprObj from;
         ObExprObj to;
         from.assign(orig_cell);
         to.assign(casted_cell);
         if (OB_SUCCESS != (ret = OB_OBJ_CAST[orig_cell.get_type()][expected_type.get_type()](params, from, to)))
         {
             jlog(WARNING, "failed to type cast obj, err=%d", ret);
         }
         else if (OB_SUCCESS != (ret = to.to(casted_cell)))
         {
             jlog(WARNING, "failed to convert expr_obj to obj, err=%d", ret);
         }
         else
         {
             res_cell = &casted_cell;
         }
     }
     else
     {
         res_cell = &orig_cell;
     }
     return ret;
 }

 static int decimal_bool(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObDecimalType);
     out.set_bool(!in.get_decimal().is_zero());
     return OB_SUCCESS;
 }

 static int mtime_pdatetime(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObModifyTimeType);
     out.set_precise_datetime(static_cast<ObPreciseDateTime> (in.get_mtime()));
     return OB_SUCCESS;
 }

 static int bool_float(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObBoolType);
     out.set_float(static_cast<float> (in.get_bool()));
     return OB_SUCCESS;
 }

 static int mtime_bool(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObModifyTimeType);
     out.set_bool(static_cast<bool> (in.get_mtime()));
     return OB_SUCCESS;
 }

 static int int_varchar(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     int ret = OB_SUCCESS;
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObIntType);
     ret = varchar_printf(out, "%ld", in.get_int());
     return ret;
 }

 static int int_decimal(const ObObjCastParams &params, const ObExprObj &in, ObExprObj &out)
 {
     UNUSED(params);
     OB_ASSERT(in.get_type() == ObIntType);
     ObNumber num;
     num.from(in.get_int());
     out.set_decimal(num); // @todo optimize
     return OB_SUCCESS;
 }