init push

unittest/storage/ob_row_generate_adapter.h

@@ -0,0 +1,99 @@
+/**
+ * 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 UNITTEST_STORAGE_OB_ROW_GENERATE_ADAPTER_H_
+#define UNITTEST_STORAGE_OB_ROW_GENERATE_ADAPTER_H_
+
+#include "common/object/ob_object.h"
+#include "storage/ob_i_store.h"
+#include "common/row/ob_row_store.h"
+#include "lib/hash/ob_hashmap.h"
+#include "lib/time/ob_time_utility.h"
+#include "lib/string/ob_string.h"
+#include "storage/ob_i_store.h"
+#include "blocksstable/ob_row_generate.h"
+
+namespace oceanbase {
+
+using namespace blocksstable;
+
+namespace common {
+
+class ObMockIterWithLimit : public ObNewRowIterator {
+  public:
+  ObMockIterWithLimit() : need_row_count_(0), got_row_count_(0){};
+  ~ObMockIterWithLimit(){};
+
+  void set_need_row_count(int64_t count)
+  {
+    need_row_count_ = count;
+  }
+
+  int64_t get_need_row_count()
+  {
+    return need_row_count_;
+  }
+
+  bool is_iter_end()
+  {
+    return need_row_count_ == got_row_count_;
+  }
+
+  protected:
+  void advance_iter()
+  {
+    ++got_row_count_;
+  }
+
+  int64_t need_row_count_;
+  int64_t got_row_count_;
+};
+
+class ObRowGenerateAdapter : public ObMockIterWithLimit {
+
+  public:
+  ObRowGenerateAdapter(){};
+  ~ObRowGenerateAdapter(){};
+
+  virtual int get_next_row(ObNewRow*& row)
+  {
+    int ret = OB_SUCCESS;
+    if (is_iter_end()) {
+      ret = OB_ITER_END;
+    } else {
+      generate_.get_next_row(new_row_);
+      row = &new_row_.row_val_;
+      advance_iter();
+    }
+    return ret;
+  }
+
+  int init(const share::schema::ObTableSchema& src_schema, ObIAllocator& buf)
+  {
+    new_row_.row_val_.count_ = src_schema.get_column_count();
+    new_row_.row_val_.cells_ = static_cast<ObObj*>(buf.alloc(new_row_.row_val_.count_ * sizeof(ObObj)));
+    return generate_.init(src_schema);
+  }
+
+  virtual void reset()
+  {
+    generate_.reset();
+  }
+
+  private:
+  ObRowGenerate generate_;
+  ObStoreRow new_row_;
+};
+
+}  // namespace common
+}  // namespace oceanbase
+#endif  // OCEANBASE_UNITTEST_MOCK_ITERATOR_H_