[FEAT MERGE] GIS

2022-12-23 12:08:27 +00:00
parent 38846e73d6
commit 40d215fc5a
519 changed files with 158600 additions and 8396 deletions
--- a/unittest/sql/engine/expr/test_gis_dispatcher.cpp
+++ b/unittest/sql/engine/expr/test_gis_dispatcher.cpp
@ -0,0 +1,268 @@
+#include <gtest/gtest.h>
+#include <sys/time.h>
+#include "lib/geo/ob_geo_func_register.h"
+#include "ob_geo_func_testx.h"
+#include "ob_geo_func_testy.h"
+#include "lib/json_type/ob_json_common.h"
+
+using namespace oceanbase::common;
+namespace oceanbase {
+namespace sql {
+
+class TestGisDispatcher : public ::testing::Test {
+public:
+  TestGisDispatcher()
+  {}
+  ~TestGisDispatcher()
+  {}
+  virtual void SetUp()
+  {}
+  virtual void TearDown()
+  {}
+  static void SetUpTestCase()
+  {}
+  static void TearDownTestCase()
+  {}
+private:
+  // disallow copy
+  DISALLOW_COPY_AND_ASSIGN(TestGisDispatcher);
+};
+
+// X will call functor Y
+enum class ObGeoFuncTestType
+{
+  NotImplemented = 0,
+  TypeX = 1,
+  TypeY = 2
+};
+
+template <ObGeoFuncTestType func_type>
+struct ObGisTestFunc {
+  typedef ObGeoFuncNotImplemented geo_func;
+};
+
+template <>
+struct ObGisTestFunc<ObGeoFuncTestType::TypeX> {
+  typedef ObGeoFuncTypeX geo_func;
+};
+
+template <>
+struct ObGisTestFunc<ObGeoFuncTestType::TypeY> {
+  typedef ObGeoFuncMockY geo_func;
+};
+
+TEST_F(TestGisDispatcher, not_impl)
+{
+  ObIWkbGeomPoint cp;
+  ObIWkbGeogPoint gp;
+  int ret = OB_SUCCESS;
+  int result;
+  ObGeoEvalCtx gis_context;
+  ret = ObGisTestFunc<ObGeoFuncTestType::NotImplemented>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_ERR_GIS_INVALID_DATA);
+
+  ret = OB_SUCCESS;
+  ret = ObGeoFunc<ObGeoFuncType::NotImplemented>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_ERR_GIS_INVALID_DATA);
+}
+
+TEST_F(TestGisDispatcher, dispatcher)
+{
+  ObString dumy_wkb(4, 4, "dumy");
+  ObIWkbGeomPoint cp;
+  ObIWkbGeomLineString cl;
+  ObIWkbGeomPolygon ca;
+  ObIWkbGeomMultiPoint cmp;
+  ObIWkbGeomMultiLineString cml;
+  ObIWkbGeomMultiPolygon cma;
+  ObIWkbGeomCollection cc;
+
+  ObIWkbGeogPoint gp;
+  ObIWkbGeogLineString gl;
+  ObIWkbGeogPolygon ga;
+  ObIWkbGeogMultiPoint gmp;
+  ObIWkbGeogMultiLineString gml;
+  ObIWkbGeogMultiPolygon gma;
+  ObIWkbGeogCollection gc;
+
+  ObArenaAllocator allocator(ObModIds::TEST);
+  ObSrsItem srs(NULL);
+  ObGeoEvalCtx gis_context(&allocator, &srs);
+  ObGeometry *gis_args[3][8] = {{0}};
+
+  gis_args[static_cast<int>(ObGeoCRS::Cartesian)][static_cast<int>(ObGeoType::POINT)] = &cp;
+  gis_args[static_cast<int>(ObGeoCRS::Cartesian)][static_cast<int>(ObGeoType::LINESTRING)] = &cl;
+  gis_args[static_cast<int>(ObGeoCRS::Cartesian)][static_cast<int>(ObGeoType::POLYGON)] = &ca;
+  gis_args[static_cast<int>(ObGeoCRS::Cartesian)][static_cast<int>(ObGeoType::MULTIPOINT)] = &cmp;
+  gis_args[static_cast<int>(ObGeoCRS::Cartesian)][static_cast<int>(ObGeoType::MULTILINESTRING)] = &cml;
+  gis_args[static_cast<int>(ObGeoCRS::Cartesian)][static_cast<int>(ObGeoType::MULTIPOLYGON)] = &cma;
+  gis_args[static_cast<int>(ObGeoCRS::Cartesian)][static_cast<int>(ObGeoType::GEOMETRYCOLLECTION)] = &cc;
+
+  gis_args[static_cast<int>(ObGeoCRS::Geographic)][static_cast<int>(ObGeoType::POINT)] = &gp;
+  gis_args[static_cast<int>(ObGeoCRS::Geographic)][static_cast<int>(ObGeoType::LINESTRING)] = &gl;
+  gis_args[static_cast<int>(ObGeoCRS::Geographic)][static_cast<int>(ObGeoType::POLYGON)] = &ga;
+  gis_args[static_cast<int>(ObGeoCRS::Geographic)][static_cast<int>(ObGeoType::MULTIPOINT)] = &gmp;
+  gis_args[static_cast<int>(ObGeoCRS::Geographic)][static_cast<int>(ObGeoType::MULTILINESTRING)] = &gml;
+  gis_args[static_cast<int>(ObGeoCRS::Geographic)][static_cast<int>(ObGeoType::MULTIPOLYGON)] = &gma;
+  gis_args[static_cast<int>(ObGeoCRS::Geographic)][static_cast<int>(ObGeoType::GEOMETRYCOLLECTION)] = &gc;
+
+  int result = 0;
+  int ret = OB_SUCCESS;
+
+  for (int crs_type = 1; crs_type < static_cast<int>(ObGeoCRS::Geographic) + 1; crs_type++) {
+    for (int g1_type = 1; g1_type < static_cast<int>(ObGeoType::GEOMETRYCOLLECTION) + 1; g1_type++) {
+      gis_args[crs_type][g1_type]->set_data(dumy_wkb);
+    }
+  }
+
+  for (int crs_type = 1; crs_type < static_cast<int>(ObGeoCRS::Geographic) + 1; crs_type++) {
+    for (int g1_type = 1; g1_type < static_cast<int>(ObGeoType::GEOMETRYCOLLECTION) + 1; g1_type++) {
+      gis_context.ut_set_geo_count(1);
+      gis_context.ut_set_geo_arg(0, gis_args[crs_type][g1_type]);
+      ret = ObGisTestFunc<ObGeoFuncTestType::TypeY>::geo_func::eval(gis_context, result);
+      ASSERT_EQ(ret, OB_SUCCESS);
+      ASSERT_EQ(result, g_test_geo_unary_func_result[crs_type][g1_type]);
+    }
+  }
+
+  for (int crs_type = 1; crs_type < static_cast<int>(ObGeoCRS::Geographic) + 1; crs_type++) {
+    for (int g1_type = 1; g1_type < static_cast<int>(ObGeoType::GEOMETRYCOLLECTION) + 1; g1_type++) {
+      for (int g2_type = 1; g2_type < static_cast<int>(ObGeoType::GEOMETRYCOLLECTION) + 1; g2_type++) {
+        gis_context.ut_set_geo_count(2);
+        gis_context.ut_set_geo_arg(0, gis_args[crs_type][g1_type]);
+        gis_context.ut_set_geo_arg(1, gis_args[crs_type][g2_type]);
+        ret = ObGisTestFunc<ObGeoFuncTestType::TypeY>::geo_func::eval(gis_context, result);
+        ASSERT_EQ(ret, OB_SUCCESS);
+        ASSERT_EQ(result, g_test_geo_binary_func_result[crs_type][g1_type][crs_type][g2_type]);
+      }
+    }
+  }
+}
+
+TEST_F(TestGisDispatcher, specialization_and_exception_test)
+{
+  ObIWkbGeomPoint cp;
+  ObIWkbGeomPolygon ca;
+  ObIWkbGeogPoint gp;
+  ObIWkbGeogLineString gl;
+  ObIWkbGeogPolygon ga;
+  ObIWkbGeogCollection gm;
+  ObString dumy_wkb(4, 4, "dumy");
+  cp.set_data(dumy_wkb);
+  ca.set_data(dumy_wkb);
+  gp.set_data(dumy_wkb);
+  gl.set_data(dumy_wkb);
+  ga.set_data(dumy_wkb);
+  gm.set_data(dumy_wkb);
+
+  int ret = OB_SUCCESS;
+  ObArenaAllocator allocator(ObModIds::TEST);
+  ObSrsItem srs(NULL);
+  ObGeoEvalCtx gis_context(&allocator, &srs);
+  ret = gis_context.append_geo_arg(&cp);
+  ASSERT_EQ(ret, OB_SUCCESS);
+
+  int result = 0;
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 1);
+
+  gis_context.ut_set_geo_arg(0, &ca);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 3);
+
+  gis_context.ut_set_geo_arg(0, &gp);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 1);
+
+  gis_context.ut_set_geo_arg(0, &ga);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 3);
+
+  gis_context.ut_set_geo_arg(0, &gm);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_ERR_NOT_IMPLEMENTED_FOR_GEOGRAPHIC_SRS);
+  ASSERT_EQ(result, 7);
+
+  gis_context.ut_set_geo_count(2);
+  gis_context.ut_set_geo_arg(0, &ca);
+  gis_context.ut_set_geo_arg(1, &ca);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 6);
+
+  gis_context.ut_set_geo_arg(0, &ga);
+  gis_context.ut_set_geo_arg(1, &ga);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 800);
+
+  gis_context.ut_set_geo_arg(0, &gm);
+  gis_context.ut_set_geo_arg(1, &gm);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_ERR_NOT_IMPLEMENTED_FOR_GEOGRAPHIC_SRS);
+  ASSERT_EQ(result, 14);
+
+  // test "functor" call "functor"
+  ObIWkbGeogMultiPoint gmp;
+  gmp.set_data(dumy_wkb);
+  gis_context.ut_set_geo_arg(0, &gmp);
+  gis_context.ut_set_geo_arg(1, &gmp);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 814);
+
+  gis_context.ut_set_geo_arg(0, &gp);
+  gis_context.ut_set_geo_arg(1, &gmp);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 100);
+
+  gis_context.ut_set_geo_arg(0, &gmp);
+  gis_context.ut_set_geo_arg(1, &gp);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, -100);
+
+  gis_context.ut_set_geo_arg(0, &gp);
+  gis_context.ut_set_geo_arg(1, &ga);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, -400);
+
+  gis_context.ut_set_geo_arg(0, &ga);
+  gis_context.ut_set_geo_arg(1, &gp);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_SUCCESS);
+  ASSERT_EQ(result, 400);
+
+  // exception test
+  gis_context.ut_set_geo_arg(0, &ga);
+  gis_context.ut_set_geo_arg(1, &gl);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_ERR_BOOST_GEOMETRY_CENTROID_EXCEPTION);
+  ASSERT_EQ(result, 777);
+
+  gis_context.ut_set_geo_arg(0, &gl);
+  gis_context.ut_set_geo_arg(1, &ga);
+  ret = ObGisTestFunc<ObGeoFuncTestType::TypeX>::geo_func::eval(gis_context, result);
+  ASSERT_EQ(ret, OB_ERR_BOOST_GEOMETRY_CENTROID_EXCEPTION);
+  ASSERT_EQ(result, 666);
+}
+
+} // namespace sql
+} // namespace oceanbase
+
+int main(int argc, char** argv)
+{
+  ::testing::InitGoogleTest(&argc, argv);
+  /*
+  system("rm -f test_gis_dispatcher.log");
+  OB_LOGGER.set_file_name("test_json_bin.log");
+  OB_LOGGER.set_log_level("INFO");
+  */
+  return RUN_ALL_TESTS();
+}