This is an automated email from the ASF dual-hosted git repository.
jackietien pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/tsfile.git
The following commit(s) were added to refs/heads/develop by this push:
new 36f974d0 [CPP].add cwrapper and cexamples to cpp. (#97)
36f974d0 is described below
commit 36f974d0a32892620a7a070c106a0f4f106b168f
Author: Colin Lee <[email protected]>
AuthorDate: Thu Jun 6 14:55:02 2024 +0800
[CPP].add cwrapper and cexamples to cpp. (#97)
---
cpp/examples/CMakeLists.txt | 3 +-
cpp/examples/{ => c_examples}/CMakeLists.txt | 18 +-
cpp/examples/c_examples/c_examples.c | 134 ++++
.../{examples.cc => c_examples/c_examples.h} | 19 +-
cpp/examples/cpp_examples/demo_read.cpp | 2 +-
cpp/examples/cpp_examples/demo_write.cpp | 9 +-
cpp/examples/examples.cc | 4 +
cpp/src/CMakeLists.txt | 3 +-
cpp/{examples => src/cwrapper}/CMakeLists.txt | 20 +-
cpp/src/cwrapper/TsFile-cwrapper.cc | 846 +++++++++++++++++++++
cpp/src/cwrapper/TsFile-cwrapper.h | 232 ++++++
11 files changed, 1245 insertions(+), 45 deletions(-)
diff --git a/cpp/examples/CMakeLists.txt b/cpp/examples/CMakeLists.txt
index c99ac0f4..4b433c96 100644
--- a/cpp/examples/CMakeLists.txt
+++ b/cpp/examples/CMakeLists.txt
@@ -29,6 +29,7 @@ include_directories(${SDK_INCLUDE_DIR})
find_library(my_tsfile_lib NAMES tsfile PATHS ${SDK_LIB_DIR_RELEASE}
NO_DEFAULT_PATH REQUIRED)
add_subdirectory(cpp_examples)
+add_subdirectory(c_examples)
add_executable(examples examples.cc)
-target_link_libraries(examples ${my_tsfile_lib} cpp_examples_obj)
\ No newline at end of file
+target_link_libraries(examples ${my_tsfile_lib} cpp_examples_obj
c_examples_obj)
\ No newline at end of file
diff --git a/cpp/examples/CMakeLists.txt
b/cpp/examples/c_examples/CMakeLists.txt
similarity index 56%
copy from cpp/examples/CMakeLists.txt
copy to cpp/examples/c_examples/CMakeLists.txt
index c99ac0f4..a1cf714e 100644
--- a/cpp/examples/CMakeLists.txt
+++ b/cpp/examples/c_examples/CMakeLists.txt
@@ -16,19 +16,7 @@ KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
]]
-cmake_minimum_required(VERSION 3.10)
-project(examples)
-message("Running in exampes directory")
-set(SDK_INCLUDE_DIR ${PROJECT_SOURCE_DIR}/../src/)
-message("SDK_INCLUDE_DIR: ${SDK_INCLUDE_DIR}")
-set(SDK_LIB_DIR_RELEASE ${PROJECT_SOURCE_DIR}/../build/Release/lib)
-message("SDK_LIB_DIR_RELEASE: ${SDK_LIB_DIR_RELEASE}")
-
-include_directories(${SDK_INCLUDE_DIR})
-find_library(my_tsfile_lib NAMES tsfile PATHS ${SDK_LIB_DIR_RELEASE}
NO_DEFAULT_PATH REQUIRED)
-
-add_subdirectory(cpp_examples)
-
-add_executable(examples examples.cc)
-target_link_libraries(examples ${my_tsfile_lib} cpp_examples_obj)
\ No newline at end of file
+message("Running in examples/c_examples directory")
+aux_source_directory(. c_SRC_LIST)
+add_library(c_examples_obj OBJECT ${c_SRC_LIST})
diff --git a/cpp/examples/c_examples/c_examples.c
b/cpp/examples/c_examples/c_examples.c
new file mode 100644
index 00000000..d2398fae
--- /dev/null
+++ b/cpp/examples/c_examples/c_examples.c
@@ -0,0 +1,134 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * License); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#include "c_examples.h"
+
+#include <fcntl.h>
+#include <malloc.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <unistd.h>
+
+
+#define HANDLE_ERROR(err_no) \
+ do { \
+ if (err_no != 0) { \
+ printf("get err no: %d", err_no); \
+ return err_no; \
+ } \
+ } while (0)
+
+ErrorCode write_tsfile() {
+ ErrorCode err_code;
+ CTsFileWriter writer;
+ if (access("c_rw.tsfile", 0) == 0) {
+ if (remove("test.tsfile") != 0) {
+ printf("Failed to delete test.tsfile file\n");
+ return -1;
+ }
+ }
+ writer = ts_writer_open("c_rw.tsfile", &err_code);
+ if (NULL == writer) {
+ return err_code;
+ }
+ ColumnSchema columnSchema;
+ columnSchema.name = "temperature";
+ columnSchema.column_def = TS_TYPE_INT32;
+ err_code =
+ tsfile_register_table_column(writer, "test_table", &columnSchema);
+ HANDLE_ERROR(err_code);
+ TableSchema tableSchema;
+ tableSchema.column_num = 3;
+ tableSchema.table_name = "test_table";
+ tableSchema.column_schema =
+ (ColumnSchema **)malloc(tableSchema.column_num * sizeof(TableSchema
*));
+ tableSchema.column_schema[0] = (ColumnSchema
*)malloc(sizeof(ColumnSchema));
+ tableSchema.column_schema[0]->column_def = TS_TYPE_DOUBLE;
+ tableSchema.column_schema[0]->name = "level";
+ tableSchema.column_schema[1] = (ColumnSchema
*)malloc(sizeof(ColumnSchema));
+ tableSchema.column_schema[1]->column_def = TS_TYPE_BOOLEAN;
+ tableSchema.column_schema[1]->name = "up";
+ tableSchema.column_schema[2] = (ColumnSchema
*)malloc(sizeof(ColumnSchema));
+ tableSchema.column_schema[2]->column_def = TS_TYPE_FLOAT;
+ tableSchema.column_schema[2]->name = "humi";
+ err_code = tsfile_register_table(writer, &tableSchema);
+ free(tableSchema.column_schema[0]);
+ free(tableSchema.column_schema[1]);
+ free(tableSchema.column_schema[2]);
+ free(tableSchema.column_schema);
+ HANDLE_ERROR(err_code);
+ printf("register table success\n");
+ TsFileRowData rowData = create_tsfile_row("test_table", 1, 4);
+ insert_data_into_tsfile_row_double(rowData, "level", 10);
+ insert_data_into_tsfile_row_float(rowData, "humi", 10.0f);
+ insert_data_into_tsfile_row_boolean(rowData, "up", true);
+ insert_data_into_tsfile_row_int32(rowData, "temperature", 10);
+ err_code = tsfile_write_row_data(writer, rowData);
+
+ rowData = create_tsfile_row("test_table", 2, 4);
+ insert_data_into_tsfile_row_double(rowData, "level", 12);
+ err_code = tsfile_write_row_data(writer, rowData);
+
+ for (int ind = 10; ind < 2000; ind++) {
+ rowData = create_tsfile_row("test_table", ind, 4);
+ insert_data_into_tsfile_row_double(rowData, "level", 12 + ind);
+ insert_data_into_tsfile_row_float(rowData, "humi", 12.0f + ind);
+ insert_data_into_tsfile_row_boolean(rowData, "up", true);
+ insert_data_into_tsfile_row_int32(rowData, "temperature", 12 + ind);
+ err_code = tsfile_write_row_data(writer, rowData);
+ }
+ printf("writer row data success\n");
+ HANDLE_ERROR(err_code);
+ HANDLE_ERROR(tsfile_flush_data(writer));
+ printf("flush data success\n");
+ HANDLE_ERROR(ts_writer_close(writer));
+ printf("close writer success\n");
+ return 0;
+}
+
+ErrorCode read_tsfile() {
+ ErrorCode err_code;
+ CTsFileReader reader;
+ reader = ts_reader_open("c_rw.tsfile", &err_code);
+ if (NULL == reader) {
+ return err_code;
+ }
+ const char *columns[] = {"temperature", "level", "up", "humi"};
+ // TimeFilterExpression* exp = create_andquery_timefilter();
+ // TimeFilterExpression* time_filter = create_time_filter("test_table",
+ // "temperature", GT, 11); TimeFilterExpression* time_filter2 =
+ // create_time_filter("test_table", "humi", GT, 10); TimeFilterExpression*
+ // time_filter3 = create_time_filter("test_table", "level", LE, 20);
+ // add_time_filter_to_and_query(exp, time_filter);
+ // add_time_filter_to_and_query(exp, time_filter2);
+ // add_time_filter_to_and_query(exp, time_filter3);
+
+ QueryDataRet ret = ts_reader_query(reader, "test_table", columns, 4, NULL);
+ printf("query success\n");
+ DataResult *result = ts_next(ret, 20);
+ if (result == NULL) {
+ printf("get result failed\n");
+ return -1;
+ }
+ print_data_result(result);
+ // destory_time_filter_query(exp);
+ HANDLE_ERROR(destory_query_dataret(ret));
+ HANDLE_ERROR(destory_tablet(result));
+ return 0;
+}
diff --git a/cpp/examples/examples.cc b/cpp/examples/c_examples/c_examples.h
similarity index 83%
copy from cpp/examples/examples.cc
copy to cpp/examples/c_examples/c_examples.h
index e800f03e..d29090cb 100644
--- a/cpp/examples/examples.cc
+++ b/cpp/examples/c_examples/c_examples.h
@@ -17,11 +17,16 @@
* under the License.
*/
-#include "cpp_examples/cpp_examples.h"
-int main() {
- // C++ examples
- demo_write();
- demo_read();
- return 0;
-}
\ No newline at end of file
+#include "cwrapper/TsFile-cwrapper.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+ErrorCode write_tsfile();
+ErrorCode read_tsfile();
+
+#ifdef __cplusplus
+}
+#endif
\ No newline at end of file
diff --git a/cpp/examples/cpp_examples/demo_read.cpp
b/cpp/examples/cpp_examples/demo_read.cpp
index ce87dc88..37a0fa33 100644
--- a/cpp/examples/cpp_examples/demo_read.cpp
+++ b/cpp/examples/cpp_examples/demo_read.cpp
@@ -68,7 +68,7 @@ int demo_read()
common::init_config_value();
storage::TsFileReader reader;
- int ret = reader.open("demo_ts.tsfile");
+ int ret = reader.open("cpp_rw.tsfile");
std::cout << "begin to query expr" << std::endl;
ASSERT(ret == 0);
diff --git a/cpp/examples/cpp_examples/demo_write.cpp
b/cpp/examples/cpp_examples/demo_write.cpp
index bb394149..7acc4f58 100644
--- a/cpp/examples/cpp_examples/demo_write.cpp
+++ b/cpp/examples/cpp_examples/demo_write.cpp
@@ -29,9 +29,11 @@ int demo_write()
std::string device_name = "root.db001.dev001";
std::string measurement_name = "m001";
storage::libtsfile_init();
- tsfile_writer.open("demo_ts.tsfile",O_CREAT | O_RDWR, 0644);
- int ret = tsfile_writer.register_timeseries(device_name, measurement_name,
common::INT32, common::PLAIN, common::UNCOMPRESSED);
- std::cout<<"register finish" << device_name<< std::endl;
+ int ret = tsfile_writer.open("cpp_rw.tsfile", O_CREAT | O_RDWR, 0644);
+ ASSERT(ret == 0);
+ ret = tsfile_writer.register_timeseries(device_name, measurement_name,
common::INT32, common::PLAIN, common::UNCOMPRESSED);
+ ASSERT(ret == 0);
+ std::cout<<"get open ret: "<<ret<<std::endl;
int row_count = 100;
for (int i = 1; i < row_count; ++i) {
@@ -39,7 +41,6 @@ int demo_write()
storage::TsRecord record(i, device_name,1);
record.points_.push_back(point);
ret = tsfile_writer.write_record(record);
- std::cout<< "write point:" << measurement_name << " " << 10000 + i <<
std::endl;
ASSERT(ret == 0);
}
diff --git a/cpp/examples/examples.cc b/cpp/examples/examples.cc
index e800f03e..9d20b8bb 100644
--- a/cpp/examples/examples.cc
+++ b/cpp/examples/examples.cc
@@ -18,10 +18,14 @@
*/
#include "cpp_examples/cpp_examples.h"
+#include "c_examples/c_examples.h"
int main() {
// C++ examples
demo_write();
demo_read();
+ // C examples
+ write_tsfile();
+ read_tsfile();
return 0;
}
\ No newline at end of file
diff --git a/cpp/src/CMakeLists.txt b/cpp/src/CMakeLists.txt
index 0675eb00..9f323c77 100644
--- a/cpp/src/CMakeLists.txt
+++ b/cpp/src/CMakeLists.txt
@@ -21,6 +21,7 @@ include(${PROJECT_SOURCE_DIR}/cmake/build_function.cmake)
add_subdirectory(common)
add_subdirectory(compress)
+add_subdirectory(cwrapper)
add_subdirectory(encoding)
add_subdirectory(file)
add_subdirectory(reader)
@@ -28,7 +29,7 @@ add_subdirectory(utils)
add_subdirectory(writer)
add_library(tsfile SHARED)
-target_link_libraries(tsfile common_obj compress_obj file_obj read_obj
write_obj )
+target_link_libraries(tsfile common_obj compress_obj cwrapper_obj file_obj
read_obj write_obj )
set(LIBTSFILE_PROJECT_VERSION 1.0)
set(LIBTSFILE_SO_VERSION 1.0)
diff --git a/cpp/examples/CMakeLists.txt b/cpp/src/cwrapper/CMakeLists.txt
similarity index 56%
copy from cpp/examples/CMakeLists.txt
copy to cpp/src/cwrapper/CMakeLists.txt
index c99ac0f4..345c66d3 100644
--- a/cpp/examples/CMakeLists.txt
+++ b/cpp/src/cwrapper/CMakeLists.txt
@@ -16,19 +16,7 @@ KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
]]
-cmake_minimum_required(VERSION 3.10)
-project(examples)
-message("Running in exampes directory")
-
-set(SDK_INCLUDE_DIR ${PROJECT_SOURCE_DIR}/../src/)
-message("SDK_INCLUDE_DIR: ${SDK_INCLUDE_DIR}")
-set(SDK_LIB_DIR_RELEASE ${PROJECT_SOURCE_DIR}/../build/Release/lib)
-message("SDK_LIB_DIR_RELEASE: ${SDK_LIB_DIR_RELEASE}")
-
-include_directories(${SDK_INCLUDE_DIR})
-find_library(my_tsfile_lib NAMES tsfile PATHS ${SDK_LIB_DIR_RELEASE}
NO_DEFAULT_PATH REQUIRED)
-
-add_subdirectory(cpp_examples)
-
-add_executable(examples examples.cc)
-target_link_libraries(examples ${my_tsfile_lib} cpp_examples_obj)
\ No newline at end of file
+message("Running in cwrapper directory")
+set(CMAKE_POSITION_INDEPENDENT_CODE ON)
+aux_source_directory(. CWRAPPER_SRC_LIST)
+add_library(cwrapper_obj OBJECT ${CWRAPPER_SRC_LIST})
\ No newline at end of file
diff --git a/cpp/src/cwrapper/TsFile-cwrapper.cc
b/cpp/src/cwrapper/TsFile-cwrapper.cc
new file mode 100644
index 00000000..d2153a1d
--- /dev/null
+++ b/cpp/src/cwrapper/TsFile-cwrapper.cc
@@ -0,0 +1,846 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * License); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#include "cwrapper/TsFile-cwrapper.h"
+
+#include <iomanip>
+
+#include "common/global.h"
+#include "reader/expression.h"
+#include "reader/filter/and_filter.h"
+#include "reader/filter/filter.h"
+#include "reader/filter/time_filter.h"
+#include "reader/filter/time_operator.h"
+#include "reader/query_data_set.h"
+#include "reader/tsfile_reader.h"
+#include "utils/errno_define.h"
+#include "writer/tsfile_writer.h"
+
+static bool is_init = false;
+
+#define INSERT_DATA_INTO_RECORD(record, column, value) \
+ do { \
+ DataPoint point(column, value); \
+ if (record->points_.size() + 1 > record->points_.capacity()) \
+ return E_BUF_NOT_ENOUGH; \
+ record->points_.push_back(point); \
+ return E_OK; \
+ } while (0)
+
+#define CONSTRUCT_EXP_INTERNAL(exp, column_name) \
+ do { \
+ exp.column_name = column_name; \
+ exp.operatype = oper; \
+ exp.children_length = 0; \
+ } while (0)
+
+#define INSERT_DATA_TABLET_STEP \
+ do { \
+ for (int i = 0; i < tablet->column_num; i++) { \
+ if (strcmp(tablet->column_schema[i]->name, column_name) == 0) { \
+ column_id = i; \
+ break; \
+ } \
+ } \
+ if (column_id == -1) { \
+ return tablet; \
+ } \
+ if (tablet->cur_num + 1 > tablet->max_capacity) { \
+ return tablet; \
+ } \
+ tablet->times[line_id] = timestamp; \
+ } while (0)
+#define TSDataType common::TSDataType
+#define TSEncoding common::TSEncoding
+#define CompressionType common::CompressionType
+#define TsFileReader storage::TsFileReader
+#define TsFileWriter storage::TsFileWriter
+#define E_OK common::E_OK
+#define TsRecord storage::TsRecord
+#define DataPoint storage::DataPoint
+#define E_BUF_NOT_ENOUGH common::E_BUF_NOT_ENOUGH
+
+TSDataType get_datatype(SchemaInfo schema_info) {
+ if (schema_info & TS_TYPE_BOOLEAN) {
+ return TSDataType::BOOLEAN;
+ } else if (schema_info & TS_TYPE_DOUBLE) {
+ return TSDataType::DOUBLE;
+ } else if (schema_info & TS_TYPE_FLOAT) {
+ return TSDataType::FLOAT;
+ } else if (schema_info & TS_TYPE_INT32) {
+ return TSDataType::INT32;
+ } else if (schema_info & TS_TYPE_INT64) {
+ return TSDataType::INT64;
+ } else if (schema_info & TS_TYPE_TEXT) {
+ return TSDataType::TEXT;
+ }
+ return TSDataType::INVALID_DATATYPE;
+}
+
+TSEncoding get_data_encoding(SchemaInfo schema_info) {
+ if (schema_info & TS_ENCODING_PLAIN) {
+ return TSEncoding::PLAIN;
+ } else if (schema_info & TS_ENCODING_TS_DIFF) {
+ return TSEncoding::DIFF;
+ } else if (schema_info & TS_ENCODING_BITMAP) {
+ return TSEncoding::BITMAP;
+ } else if (schema_info & TS_ENCODING_GORILLA) {
+ return TSEncoding::GORILLA;
+ }
+ return TSEncoding::PLAIN;
+}
+
+CompressionType get_data_compression(SchemaInfo schema_info) {
+ if (schema_info & TS_COMPRESS_UNCOMPRESS) {
+ return CompressionType::UNCOMPRESSED;
+ } else if (schema_info & TS_COMPRESS_LZ4) {
+ return CompressionType::LZ4;
+ }
+ return CompressionType::UNCOMPRESSED;
+}
+
+SchemaInfo get_schema_info(TSDataType type) {
+ switch (type) {
+ case TSDataType::BOOLEAN:
+ return TS_TYPE_BOOLEAN;
+ case TSDataType::DOUBLE:
+ return TS_TYPE_DOUBLE;
+ case TSDataType::FLOAT:
+ return TS_TYPE_FLOAT;
+ case TSDataType::INT32:
+ return TS_TYPE_INT32;
+ case TSDataType::INT64:
+ return TS_TYPE_INT64;
+ case TSDataType::TEXT:
+ return TS_TYPE_TEXT;
+ default:
+ return 0;
+ }
+}
+
+void init_tsfile_config() {
+ if (!is_init) {
+ common::init_config_value();
+ is_init = true;
+ }
+}
+
+CTsFileReader ts_reader_open(const char* pathname, ErrorCode* err_code) {
+ init_tsfile_config();
+ TsFileReader* reader = new TsFileReader();
+ int ret = reader->open(pathname);
+ if (ret != E_OK) {
+ std::cout << "open file failed" << std::endl;
+ *err_code = ret;
+ delete reader;
+ return nullptr;
+ }
+ return reader;
+}
+
+CTsFileWriter ts_writer_open(const char* pathname, ErrorCode* err_code) {
+ init_tsfile_config();
+ TsFileWriter* writer = new TsFileWriter();
+ int ret = writer->open(pathname, O_CREAT | O_RDWR, 0644);
+ if (ret != E_OK) {
+ delete writer;
+ *err_code = ret;
+ return nullptr;
+ }
+ return writer;
+}
+
+CTsFileWriter ts_writer_open_flag(const char* pathname, mode_t flag,
+ ErrorCode* err_code) {
+ init_tsfile_config();
+ TsFileWriter* writer = new TsFileWriter();
+ int ret = writer->open(pathname, O_CREAT | O_RDWR, flag);
+ if (ret != E_OK) {
+ delete writer;
+ *err_code = ret;
+ return nullptr;
+ }
+ return writer;
+}
+
+CTsFileWriter ts_writer_open_conf(const char* pathname, int flag,
+ ErrorCode* err_code, TsFileConf* conf) {
+ *err_code = common::E_INVALID_ARG;
+ return nullptr;
+}
+
+ErrorCode ts_writer_close(CTsFileWriter writer) {
+ TsFileWriter* w = (TsFileWriter*)writer;
+ int ret = w->close();
+ delete w;
+ return ret;
+}
+
+ErrorCode ts_reader_close(CTsFileReader reader) {
+ TsFileReader* ts_reader = (TsFileReader*)reader;
+ delete ts_reader;
+ return E_OK;
+}
+
+ErrorCode tsfile_register_table_column(CTsFileWriter writer,
+ const char* table_name,
+ ColumnSchema* schema) {
+ TsFileWriter* w = (TsFileWriter*)writer;
+ int ret = w->register_timeseries(table_name, schema->name,
+ get_datatype(schema->column_def),
+ get_data_encoding(schema->column_def),
+ get_data_compression(schema->column_def));
+ std::cout << "register table column name" << table_name << std::endl;
+ std::cout << "register column name" << schema->name << std::endl;
+ std::cout << "register column type" << get_datatype(schema->column_def)
+ << std::endl;
+ return ret;
+}
+
+ErrorCode tsfile_register_table(CTsFileWriter writer,
+ TableSchema* table_schema) {
+ TsFileWriter* w = (TsFileWriter*)writer;
+ for (int column_id = 0; column_id < table_schema->column_num; column_id++)
{
+ ColumnSchema* schema = table_schema->column_schema[column_id];
+ ErrorCode ret =
+ w->register_timeseries(table_schema->table_name, schema->name,
+ get_datatype(schema->column_def),
+ get_data_encoding(schema->column_def),
+ get_data_compression(schema->column_def));
+ if (ret != E_OK) {
+ return ret;
+ }
+ }
+ return E_OK;
+}
+
+TsFileRowData create_tsfile_row(const char* table_name, int64_t timestamp,
+ int column_length) {
+ TsRecord* record = new TsRecord(timestamp, table_name, column_length);
+ return record;
+}
+
+Tablet* create_tablet(const char* table_name, int max_capacity) {
+ Tablet* tablet = new Tablet();
+ tablet->table_name = strdup(table_name);
+ tablet->max_capacity = max_capacity;
+ tablet->times = (timestamp*)malloc(max_capacity * sizeof(int64_t));
+ return tablet;
+}
+
+int get_size_from_schema_info(SchemaInfo schema_info) {
+ if (schema_info & TS_TYPE_BOOLEAN) {
+ return sizeof(bool);
+ } else if (schema_info & TS_TYPE_DOUBLE) {
+ return sizeof(double);
+ } else if (schema_info & TS_TYPE_FLOAT) {
+ return sizeof(float);
+ } else if (schema_info & TS_TYPE_INT32) {
+ return sizeof(int32_t);
+ } else if (schema_info & TS_TYPE_INT64) {
+ return sizeof(int64_t);
+ } else if (schema_info & TS_TYPE_TEXT) {
+ return sizeof(char*);
+ }
+ return 0;
+}
+
+Tablet* add_column_to_tablet(Tablet* tablet, char* column_name,
+ SchemaInfo column_def) {
+ tablet->column_num++;
+ tablet->column_schema = (ColumnSchema**)realloc(
+ tablet->column_schema, tablet->column_num * sizeof(ColumnSchema*));
+ tablet->bitmap =
+ (bool**)realloc(tablet->bitmap, tablet->column_num * sizeof(bool*));
+ tablet->bitmap[tablet->column_num - 1] =
+ (bool*)malloc(tablet->max_capacity * sizeof(bool));
+ std::memset(tablet->bitmap[tablet->column_num - 1], 0,
+ tablet->max_capacity * sizeof(bool));
+ ColumnSchema* schema = new ColumnSchema();
+ schema->name = column_name;
+ schema->column_def = column_def;
+ tablet->column_schema[tablet->column_num - 1] = schema;
+ tablet->value =
+ (void**)realloc(tablet->value, tablet->column_num * sizeof(void*));
+ tablet->value[tablet->column_num - 1] =
+ (void*)malloc(tablet->max_capacity * sizeof(int64_t));
+ return tablet;
+}
+
+Tablet* add_data_to_tablet_i64(Tablet* tablet, int line_id, int64_t timestamp,
+ const char* column_name, int64_t value) {
+ int column_id = -1;
+ INSERT_DATA_TABLET_STEP;
+ memcpy((int64_t*)tablet->value[column_id] + line_id, &value,
+ sizeof(int64_t));
+ tablet->bitmap[column_id][line_id] = true;
+ line_id > tablet->cur_num ? tablet->cur_num = line_id : 0;
+ return tablet;
+}
+
+Tablet* add_data_to_tablet_i32(Tablet* tablet, int line_id, int64_t timestamp,
+ const char* column_name, int32_t value) {
+ int column_id = -1;
+ INSERT_DATA_TABLET_STEP;
+ memcpy((int32_t*)tablet->value[column_id] + line_id, &value,
+ sizeof(int32_t));
+ tablet->bitmap[column_id][line_id] = true;
+ line_id > tablet->cur_num ? tablet->cur_num = line_id : 0;
+ return tablet;
+}
+
+Tablet* add_data_to_tablet_float(Tablet* tablet, int line_id, int64_t
timestamp,
+ const char* column_name, float value) {
+ int column_id = -1;
+ INSERT_DATA_TABLET_STEP;
+ memcpy((float*)tablet->value[column_id] + line_id, &value, sizeof(float));
+ tablet->bitmap[column_id][line_id] = true;
+ line_id > tablet->cur_num ? tablet->cur_num = line_id : 0;
+ return tablet;
+}
+
+Tablet* add_data_to_tablet_double(Tablet* tablet, int line_id,
+ int64_t timestamp, const char* column_name,
+ double value) {
+ int column_id = -1;
+ INSERT_DATA_TABLET_STEP;
+ memcpy((double*)tablet->value[column_id] + line_id, &value,
sizeof(double));
+ tablet->bitmap[column_id][line_id] = true;
+ line_id > tablet->cur_num ? tablet->cur_num = line_id : 0;
+ return tablet;
+}
+
+Tablet* add_data_to_tablet_bool(Tablet* tablet, int line_id, int64_t timestamp,
+ const char* column_name, bool value) {
+ int column_id = -1;
+ INSERT_DATA_TABLET_STEP;
+ memcpy((bool*)tablet->value[column_id] + line_id, &value, sizeof(bool));
+ tablet->bitmap[column_id][line_id] = true;
+ line_id > tablet->cur_num ? tablet->cur_num = line_id : 0;
+ return tablet;
+}
+
+Tablet* add_data_to_tablet_char(Tablet* tablet, int line_id, int64_t timestamp,
+ const char* column_name, char* value) {
+ int column_id = -1;
+ INSERT_DATA_TABLET_STEP;
+ memcpy((char*)tablet->value[column_id] + line_id, &value, sizeof(char*));
+ tablet->bitmap[column_id][line_id] = true;
+ line_id > tablet->cur_num ? tablet->cur_num = line_id : 0;
+ return tablet;
+}
+
+// Tablet* add_null_to_tablet(Tablet* tablet, int line_id, int64_t timestamp,
+// const char* column_num) {
+// int column_id = -1;
+// for (int i = 0; i < tablet->column_num; i++) {
+// if (strcmp(tablet->column_schema[i]->name, column_num) == 0) {
+// column_id = i;
+// break;
+// }
+// }
+// if (column_id == -1) {
+// return tablet;
+// }
+
+// if (tablet->cur_num + 1 > tablet->max_capacity) {
+// return tablet;
+// }
+// tablet->times[line_id] = timestamp;
+// memcpy((int64_t*)tablet->value[column_id] + line_id, 0,
sizeof(int64_t));
+// line_id > tablet->cur_num ? tablet->cur_num = line_id : 0;
+// return tablet;
+// }
+
+ErrorCode destory_tablet(Tablet* tablet) {
+ free(tablet->table_name);
+ tablet->table_name = nullptr;
+ free(tablet->times);
+ tablet->times = nullptr;
+ for (int i = 0; i < tablet->column_num; i++) {
+ free(tablet->column_schema[i]);
+ free(tablet->value[i]);
+ free(tablet->bitmap[i]);
+ }
+ free(tablet->bitmap);
+ free(tablet->column_schema);
+ free(tablet->value);
+ delete tablet;
+ return E_OK;
+}
+
+ErrorCode insert_data_into_tsfile_row_int32(TsFileRowData data, char*
columname,
+ int32_t value) {
+ TsRecord* record = (TsRecord*)data;
+ INSERT_DATA_INTO_RECORD(record, columname, value);
+}
+
+ErrorCode insert_data_into_tsfile_row_boolean(TsFileRowData data,
+ char* columname, bool value) {
+ TsRecord* record = (TsRecord*)data;
+ INSERT_DATA_INTO_RECORD(record, columname, value);
+}
+
+ErrorCode insert_data_into_tsfile_row_int64(TsFileRowData data, char*
columname,
+ int64_t value) {
+ TsRecord* record = (TsRecord*)data;
+ INSERT_DATA_INTO_RECORD(record, columname, value);
+}
+
+ErrorCode insert_data_into_tsfile_row_float(TsFileRowData data, char*
columname,
+ float value) {
+ TsRecord* record = (TsRecord*)data;
+ INSERT_DATA_INTO_RECORD(record, columname, value);
+}
+
+ErrorCode insert_data_into_tsfile_row_double(TsFileRowData data,
+ char* columname, double value) {
+ TsRecord* record = (TsRecord*)data;
+ INSERT_DATA_INTO_RECORD(record, columname, value);
+}
+
+ErrorCode tsfile_write_row_data(CTsFileWriter writer, TsFileRowData data) {
+ TsFileWriter* w = (TsFileWriter*)writer;
+ TsRecord* record = (TsRecord*)data;
+ int ret = w->write_record(*record);
+ if (ret == E_OK) {
+ delete record;
+ }
+ return ret;
+}
+
+ErrorCode destory_tsfile_row(TsFileRowData data) {
+ TsRecord* record = (TsRecord*)data;
+ if (record != nullptr) {
+ delete record;
+ record = nullptr;
+ }
+ return E_OK;
+}
+
+ErrorCode tsfile_flush_data(CTsFileWriter writer) {
+ TsFileWriter* w = (TsFileWriter*)writer;
+ int ret = w->flush();
+ return ret;
+}
+
+Expression create_column_filter(const char* column_name, OperatorType oper,
+ int32_t int32_value) {
+ Expression exp;
+ CONSTRUCT_EXP_INTERNAL(exp, column_name);
+ exp.const_condition.value_condition = int32_value;
+ exp.const_condition.type = TS_TYPE_INT32;
+ return exp;
+}
+
+Expression create_column_filter(const char* column_name, OperatorType oper,
+ int64_t int64_value) {
+ Expression exp;
+ CONSTRUCT_EXP_INTERNAL(exp, column_name);
+ exp.const_condition.value_condition = int64_value;
+ exp.const_condition.type = TS_TYPE_INT64;
+ return exp;
+}
+Expression create_column_filter(const char* column_name, OperatorType oper,
+ bool bool_value) {
+ Expression exp;
+ CONSTRUCT_EXP_INTERNAL(exp, column_name);
+ exp.const_condition.value_condition = bool_value ? 1 : 0;
+ exp.const_condition.type = TS_TYPE_BOOLEAN;
+ return exp;
+}
+Expression create_column_filter(const char* column_name, OperatorType oper,
+ float float_value) {
+ Expression exp;
+ CONSTRUCT_EXP_INTERNAL(exp, column_name);
+ memcpy(&exp.const_condition.value_condition, &float_value, sizeof(float));
+ exp.const_condition.type = TS_TYPE_FLOAT;
+ return exp;
+}
+Expression create_column_filter(const char* column_name, OperatorType oper,
+ double double_value) {
+ Expression exp;
+ CONSTRUCT_EXP_INTERNAL(exp, column_name);
+ exp.const_condition.value_condition = double_value;
+ exp.const_condition.type = TS_TYPE_DOUBLE;
+ return exp;
+}
+Expression create_column_filter(const char* column_name, OperatorType oper,
+ const char* char_value) {
+ Expression exp;
+ CONSTRUCT_EXP_INTERNAL(exp, column_name);
+ exp.const_condition.value_condition =
reinterpret_cast<int64_t>(char_value);
+ exp.const_condition.type = TS_TYPE_TEXT;
+ return exp;
+}
+
+TimeFilterExpression* create_andquery_timefilter() {
+ storage::Expression* exp = new storage::Expression(storage::AND_EXPR);
+ return (TimeFilterExpression*)exp;
+}
+
+TimeFilterExpression* create_time_filter(const char* table_name,
+ const char* column_name,
+ OperatorType oper, int64_t timestamp)
{
+ std::string table_name_str(table_name);
+ std::string column_name_str(column_name);
+ storage::Path path(table_name_str, column_name_str);
+ storage::Filter* filter;
+ switch (oper) {
+ case GT:
+ filter = storage::TimeFilter::gt(timestamp);
+ break;
+ case LT:
+ filter = storage::TimeFilter::lt(timestamp);
+ break;
+ case EQ:
+ filter = storage::TimeFilter::eq(timestamp);
+ break;
+ case NOTEQ:
+ filter = storage::TimeFilter::not_eqt(timestamp);
+ break;
+ case GE:
+ filter = storage::TimeFilter::gt_eq(timestamp);
+ break;
+ case LE:
+ filter = storage::TimeFilter::lt_eq(timestamp);
+ break;
+ default:
+ filter = nullptr;
+ break;
+ }
+ storage::Expression* exp =
+ new storage::Expression(storage::SERIES_EXPR, path, filter);
+ return (TimeFilterExpression*)exp;
+}
+
+TimeFilterExpression* add_time_filter_to_and_query(
+ TimeFilterExpression* exp_and, TimeFilterExpression* exp) {
+ storage::Expression* and_exp = (storage::Expression*)exp_and;
+ storage::Expression* time_exp = (storage::Expression*)exp;
+ if (and_exp->left_ == nullptr) {
+ and_exp->left_ = time_exp;
+ } else if (and_exp->right_ == nullptr) {
+ and_exp->right_ = time_exp;
+ } else {
+ storage::Expression* new_exp =
+ new storage::Expression(storage::AND_EXPR);
+ new_exp->left_ = and_exp->right_;
+ and_exp->right_ = new_exp;
+ add_time_filter_to_and_query((TimeFilterExpression*)new_exp, exp);
+ }
+ return exp_and;
+}
+
+void destory_time_filter_query(TimeFilterExpression* expression) {
+ if (expression == nullptr) {
+ return;
+ }
+
+ destory_time_filter_query(
+ (TimeFilterExpression*)((storage::Expression*)expression)->left_);
+ destory_time_filter_query(
+ (TimeFilterExpression*)((storage::Expression*)expression)->right_);
+ storage::Expression* exp = (storage::Expression*)expression;
+ if (exp->type_ == storage::ExpressionType::SERIES_EXPR) {
+ delete exp->filter_;
+ } else {
+ delete exp;
+ }
+}
+
+Expression create_global_time_expression(OperatorType oper, int64_t timestamp)
{
+ Expression exp;
+ exp.operatype = oper;
+ exp.expression_type = GLOBALTIME;
+ exp.const_condition.value_condition = timestamp;
+ exp.const_condition.type = TS_TYPE_INT64;
+ return exp;
+}
+
+Expression* and_filter_to_and_query(Expression* exp_and, Expression* exp) {
+ if (exp_and->children_length >= MAX_COLUMN_FILTER_NUM - 1) {
+ return nullptr;
+ }
+ exp_and->children[exp_and->children_length++] = exp;
+ return exp_and;
+}
+
+QueryDataRet ts_reader_query(CTsFileReader reader, const char* table_name,
+ const char** columns_name, int column_num,
+ TimeFilterExpression* expression) {
+ TsFileReader* r = (TsFileReader*)reader;
+ std::string table_name_str(table_name);
+ std::vector<storage::Path> selected_paths;
+ for (int i = 0; i < column_num; i++) {
+ std::string column_name(columns_name[i]);
+ selected_paths.push_back(storage::Path(table_name_str, column_name));
+ }
+
+ storage::QueryDataSet* qds = nullptr;
+ storage::QueryExpression* query_expression =
+ storage::QueryExpression::create(selected_paths,
+ (storage::Expression*)expression);
+ r->query(query_expression, qds);
+ QueryDataRet ret = (QueryDataRet)malloc(sizeof(struct query_data_ret));
+ ret->data = qds;
+ ret->column_names = (char**)malloc(column_num * sizeof(char*));
+ ret->column_num = column_num;
+ memcpy(ret->column_names, columns_name, column_num * sizeof(char*));
+ storage::QueryExpression::destory(query_expression);
+ return ret;
+}
+
+QueryDataRet ts_reader_begin_end(CTsFileReader reader, const char* table_name,
+ char** columns_name, int column_num,
+ timestamp begin, timestamp end) {
+ TsFileReader* r = (TsFileReader*)reader;
+ std::string table_name_str(table_name);
+ std::vector<storage::Path> selected_paths;
+ for (int i = 0; i < column_num; i++) {
+ std::string column_name(columns_name[i]);
+ selected_paths.push_back(storage::Path(table_name_str, column_name));
+ }
+
+ storage::QueryDataSet* qds = nullptr;
+ storage::Filter* filter_low = nullptr;
+ storage::Filter* filter_high = nullptr;
+ storage::Expression* exp = nullptr;
+ storage::Filter* and_filter = nullptr;
+ if (begin != -1) {
+ filter_low = storage::TimeFilter::gt_eq(begin);
+ }
+ if (end != -1) {
+ filter_high = storage::TimeFilter::lt_eq(end);
+ }
+ if (filter_low != nullptr && filter_high != nullptr) {
+ and_filter = new storage::AndFilter(filter_low, filter_high);
+ exp = new storage::Expression(storage::GLOBALTIME_EXPR, and_filter);
+ } else if (filter_low != nullptr && filter_high == nullptr) {
+ exp = new storage::Expression(storage::GLOBALTIME_EXPR, filter_low);
+ } else if (filter_high != nullptr && filter_low == nullptr) {
+ exp = new storage::Expression(storage::GLOBALTIME_EXPR, filter_high);
+ }
+ storage::QueryExpression* query_expr =
+ storage::QueryExpression::create(selected_paths, exp);
+ r->query(query_expr, qds);
+ QueryDataRet ret = (QueryDataRet)malloc(sizeof(struct query_data_ret));
+ ret->data = qds;
+ ret->column_num = column_num;
+ ret->column_names = (char**)malloc(column_num * sizeof(char*));
+ memcpy(ret->column_names, columns_name, column_num * sizeof(char*));
+ storage::QueryExpression::destory(query_expr);
+ return ret;
+}
+
+QueryDataRet ts_reader_read(CTsFileReader reader, const char* table_name,
+ char** columns_name, int column_num) {
+ TsFileReader* r = (TsFileReader*)reader;
+ std::string table_name_str(table_name);
+ std::vector<storage::Path> selected_paths;
+ for (int i = 0; i < column_num; i++) {
+ std::string column_name(columns_name[i]);
+ selected_paths.push_back(storage::Path(table_name_str, column_name));
+ }
+ storage::QueryDataSet* qds = nullptr;
+ storage::QueryExpression* query_expr =
+ storage::QueryExpression::create(selected_paths, nullptr);
+ r->query(query_expr, qds);
+ QueryDataRet ret = (QueryDataRet)malloc(sizeof(struct query_data_ret));
+ ret->data = qds;
+ ret->column_names = (char**)malloc(column_num * sizeof(char*));
+ ret->column_num = column_num;
+ memcpy(ret->column_names, columns_name, column_num * sizeof(char*));
+ storage::QueryExpression::destory(query_expr);
+ return ret;
+}
+
+ErrorCode destory_query_dataret(QueryDataRet data) {
+ storage::QueryDataSet* qds = (storage::QueryDataSet*)data->data;
+ delete qds;
+ free(data->column_names);
+ free(data);
+ return E_OK;
+}
+
+DataResult* ts_next(QueryDataRet data, int expect_line_count) {
+ storage::QueryDataSet* qds = (storage::QueryDataSet*)data->data;
+ DataResult* result = create_tablet("result", expect_line_count);
+ storage::RowRecord* record;
+ bool init_tablet = false;
+ for (int i = 0; i < expect_line_count; i++) {
+ record = qds->get_next();
+ if (record == nullptr) {
+ break;
+ std::cout << "record null now"
+ << "i = " << i << std::endl;
+ }
+ int column_num = record->get_fields()->size();
+ if (!init_tablet) {
+ for (int col = 0; col < column_num; col++) {
+ storage::Field* field = record->get_field(col);
+ result = add_column_to_tablet(result, data->column_names[col],
+ get_schema_info(field->type_));
+ }
+ init_tablet = true;
+ std::cout << "init finished" << std::endl;
+ }
+ for (int col = 0; col < column_num; col++) {
+ storage::Field* field = record->get_field(col);
+ switch (field->type_) {
+ // all data will stored as 8 bytes
+ case TSDataType::BOOLEAN:
+ result = add_data_to_tablet_bool(
+ result, i, record->get_timestamp(),
+ data->column_names[col], field->value_.bval_);
+ break;
+ case TSDataType::INT32:
+ result = add_data_to_tablet_i32(
+ result, i, record->get_timestamp(),
+ data->column_names[col], field->value_.ival_);
+ break;
+ case TSDataType::INT64:
+ result = add_data_to_tablet_i64(
+ result, i, record->get_timestamp(),
+ data->column_names[col], field->value_.lval_);
+ break;
+ case TSDataType::FLOAT:
+ result = add_data_to_tablet_float(
+ result, i, record->get_timestamp(),
+ data->column_names[col], field->value_.fval_);
+ break;
+ case TSDataType::DOUBLE:
+ result = add_data_to_tablet_double(
+ result, i, record->get_timestamp(),
+ data->column_names[col], field->value_.dval_);
+ break;
+ case TSDataType::TEXT:
+ result = add_data_to_tablet_char(
+ result, i, record->get_timestamp(),
+ data->column_names[col], field->value_.sval_);
+ break;
+ case TSDataType::NULL_TYPE:
+ // result = add_data_to_tablet(result, i ,
+ // record->get_timestamp(),
+ // data->column_names[col], 0);
+ // skip null data
+ break;
+ default:
+ std::cout << field->type_ << std::endl;
+ std::cout << "error here" << std::endl;
+ return nullptr;
+ }
+ }
+ }
+ return result;
+}
+
+void print_data_result(DataResult* result) {
+ std::cout << std::left << std::setw(15) << "timestamp";
+ for (int i = 0; i < result->column_num; i++) {
+ std::cout << std::left << std::setw(15)
+ << result->column_schema[i]->name;
+ }
+ std::cout << std::endl;
+ for (int i = 0; i < result->cur_num; i++) {
+ std::cout << std::left << std::setw(15);
+ std::cout << result->times[i];
+ for (int j = 0; j < result->column_num; j++) {
+ ColumnSchema* schema = result->column_schema[j];
+ double dval;
+ float fval;
+ std::cout << std::left << std::setw(15);
+ switch (get_datatype(schema->column_def)) {
+ case TSDataType::BOOLEAN:
+ std::cout
+ << ((*((int64_t*)result->value[j] + i)) > 0 ? "true"
+ : "false");
+ break;
+ case TSDataType::INT32:
+ std::cout << *((int64_t*)result->value[j] + i);
+ break;
+ case TSDataType::INT64:
+ std::cout << *((int64_t*)result->value[j] + i);
+ break;
+ case TSDataType::FLOAT:
+ memcpy(&fval, (int64_t*)result->value[j] + i,
+ sizeof(float));
+ std::cout << fval;
+ break;
+ case TSDataType::DOUBLE:
+ memcpy(&dval, (int64_t*)result->value[j] + i,
+ sizeof(double));
+ std::cout << dval;
+ break;
+ default:
+ std::cout << "";
+ }
+ }
+ std::cout << std::endl;
+ }
+}
+
+// }
+
+// storage::Expression construct_query(Expression* exp) {
+// int column_num = exp->children_length;
+// std::vector<storage::Path> paths;
+// for (int i = 0; i < column_num; i++) {
+// Expression* exp = exp->children[i];
+// if (exp->expression_type != )
+// if (exp->column_name != nullptr ) {
+// std::string column_name = exp->column_name;
+
+// } else if (column->expression_type == AND) {
+// storage::Expression and_exp = construct_query(table_name,
+// column);
+// // add and_exp to the query
+// }
+// column++;
+// }
+// // construct the query using paths and other information
+// // return the constructed query
+// }
+
+// storage::Filter get_filter(int operate_type, Constant condition) {
+// switch(operate_type) {
+// case GT:
+// return storage::TimeFilter::gt();
+
+// }
+
+// }
+
+// storage::Expression construct_query(const char* table_name,
+// Expression exp) {
+// std::string table = table_name;
+// int column_num = exp.children_length;
+// std::vector<storage::Path> paths;
+// paths.reserve(column_num);
+// Expression* column = exp.children;
+// for (int i = 0; i < column_num;i++) {
+// if (column_num == 1) {
+// std::string column_name = column->column_name;
+// // select_list
+// paths.push_back(storage::Path(table, column_name));
+// int operate = column->operatype;
+// Filter filter = get_filter(operate, column->const_condition);
+// }
+// }
+// }
diff --git a/cpp/src/cwrapper/TsFile-cwrapper.h
b/cpp/src/cwrapper/TsFile-cwrapper.h
new file mode 100644
index 00000000..08b664d5
--- /dev/null
+++ b/cpp/src/cwrapper/TsFile-cwrapper.h
@@ -0,0 +1,232 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * License); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#ifndef CWRAPPER_TSFILE_CWRAPPER_H
+#define CWRAPPER_TSFILE_CWRAPPER_H
+
+#include <fcntl.h>
+#include <stdbool.h>
+#include <stddef.h>
+#include <stdint.h>
+
+typedef long long SchemaInfo;
+typedef long long timestamp;
+typedef void* CTsFileReader;
+typedef void* CTsFileWriter;
+typedef void* TsFileRowData;
+typedef int ErrorCode;
+typedef void* TimeFilterExpression;
+
+// DATA TYPE
+#define TS_TYPE_INT32 1 << 8
+#define TS_TYPE_BOOLEAN 1 << 9
+#define TS_TYPE_FLOAT 1 << 10
+#define TS_TYPE_DOUBLE 1 << 11
+#define TS_TYPE_INT64 1 << 12
+#define TS_TYPE_TEXT 1 << 13
+
+// ENCODING TYPE
+#define TS_ENCODING_PLAIN 1 << 16
+#define TS_ENCODING_TS_DIFF 1 << 17
+#define TS_ENCODING_DICTIONARY 1 << 18
+#define TS_ENCODING_RLE 1 << 19
+#define TS_ENCODING_BITMAP 1 << 20
+#define TS_ENCODING_GORILLA_V1 1 << 21
+#define TS_ENCODING_REGULAR 1 << 22
+#define TS_ENCODING_GORILLA 1 << 23
+#define TS_ENCODING_ZIGZAG 1 << 24
+#define TS_ENCODING_FREQ 1 << 25
+
+// COMPRESS TYPE
+#define TS_COMPRESS_UNCOMPRESS 1LL << 32
+#define TS_COMPRESS_SNAPPY 1LL << 33
+#define TS_COMPRESS_GZIP 1LL << 34
+#define TS_COMPRESS_LZO 1LL << 35
+#define TS_COMPRESS_SDT 1LL << 36
+#define TS_COMPRESS_PAA 1LL << 37
+#define TS_COMPRESS_PLA 1LL << 38
+#define TS_COMPRESS_LZ4 1LL << 39
+
+#define MAX_COLUMN_FILTER_NUM 10
+
+typedef struct column_schema {
+ char* name;
+ SchemaInfo column_def;
+} ColumnSchema;
+
+typedef struct table_shcema {
+ char* table_name;
+ ColumnSchema** column_schema;
+ int column_num;
+} TableSchema;
+
+typedef enum operator_type {
+ LT,
+ LE,
+ EQ,
+ GT,
+ GE,
+ NOTEQ,
+} OperatorType;
+
+typedef enum expression_type {
+ OR,
+ AND,
+ GLOBALTIME,
+} ExpressionType;
+
+typedef struct constant {
+ int64_t value_condition;
+ int type;
+} Constant;
+
+typedef struct expression {
+ const char* column_name;
+ Constant const_condition;
+ ExpressionType expression_type;
+ OperatorType operatype;
+ struct expression* children[MAX_COLUMN_FILTER_NUM];
+ int children_length;
+} Expression;
+
+typedef struct tablet {
+ char* table_name;
+ ColumnSchema** column_schema;
+ int column_num;
+ timestamp* times;
+ bool** bitmap;
+ void** value;
+ int cur_num;
+ int max_capacity;
+} Tablet;
+
+typedef struct tsfile_conf {
+ int mem_threshold_kb;
+} TsFileConf;
+
+typedef Tablet DataResult;
+
+typedef void* QueryDataRetINTERNAL;
+typedef struct query_data_ret {
+ char** column_names;
+ int column_num;
+ QueryDataRetINTERNAL data;
+} * QueryDataRet;
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+CTsFileReader ts_reader_open(const char* pathname, ErrorCode* err_code);
+CTsFileWriter ts_writer_open(const char* pathname, ErrorCode* err_code);
+CTsFileWriter ts_writer_open_flag(const char* pathname, mode_t flag,
+ ErrorCode* err_code);
+CTsFileWriter ts_writer_open_conf(const char* pathname, mode_t flag,
+ ErrorCode* err_code, TsFileConf* conf);
+
+ErrorCode ts_writer_close(CTsFileWriter writer);
+ErrorCode ts_reader_close(CTsFileReader reader);
+
+ErrorCode tsfile_register_table_column(CTsFileWriter writer,
+ const char* table_name,
+ ColumnSchema* schema);
+ErrorCode tsfile_register_table(CTsFileWriter writer,
+ TableSchema* table_shcema);
+
+TsFileRowData create_tsfile_row(const char* tablename, int64_t timestamp,
+ int column_length);
+
+ErrorCode insert_data_into_tsfile_row_int32(TsFileRowData data, char*
columname,
+ int32_t value);
+ErrorCode insert_data_into_tsfile_row_boolean(TsFileRowData data,
+ char* columname, bool value);
+ErrorCode insert_data_into_tsfile_row_int64(TsFileRowData data, char*
columname,
+ int64_t value);
+ErrorCode insert_data_into_tsfile_row_float(TsFileRowData data, char*
columname,
+ float value);
+ErrorCode insert_data_into_tsfile_row_double(TsFileRowData data,
+ char* columname, double value);
+
+ErrorCode tsfile_write_row_data(CTsFileWriter writer, TsFileRowData data);
+ErrorCode destory_tsfile_row(TsFileRowData data);
+
+Tablet* create_tablet(const char* table_name, int max_capacity);
+Tablet* add_column_to_tablet(Tablet* tablet, char* column_name,
+ SchemaInfo column_def);
+Tablet add_data_to_tablet(Tablet tablet, int line_id, int64_t timestamp,
+ const char* column_name, int64_t value);
+
+ErrorCode destory_tablet(Tablet* tablet);
+
+ErrorCode tsfile_flush_data(CTsFileWriter writer);
+
+Expression create_column_filter_I32(const char* column_name, OperatorType oper,
+ int32_t int32_value);
+Expression create_column_filter_I64(const char* column_name, OperatorType oper,
+ int64_t int64_value);
+Expression create_column_filter_bval(const char* column_name, OperatorType
oper,
+ bool bool_value);
+Expression create_column_filter_fval(const char* column_name, OperatorType
oper,
+ float float_value);
+Expression create_column_filter_dval(const char* column_name, OperatorType
oper,
+ double double_value);
+Expression create_column_filter_cval(const char* column_name, OperatorType
oper,
+ const char* char_value);
+
+TimeFilterExpression* create_andquery_timefilter();
+
+TimeFilterExpression* create_time_filter(const char* table_name,
+ const char* column_name,
+ OperatorType oper, int64_t timestamp);
+
+TimeFilterExpression* add_time_filter_to_and_query(
+ TimeFilterExpression* exp_and, TimeFilterExpression* exp);
+
+void destory_time_filter_query(TimeFilterExpression* expression);
+
+Expression* create_time_expression(const char* column_name, OperatorType oper,
+ int64_t timestamp);
+
+Expression* add_and_filter_to_and_query(Expression* exp_and, Expression* exp);
+
+QueryDataRet ts_reader_query(CTsFileReader reader, const char* table_name,
+ const char** columns, int colum_num,
+ TimeFilterExpression* expression);
+
+QueryDataRet ts_reader_begin_end(CTsFileReader reader, const char* table_name,
+ char** columns, int colum_num, timestamp
begin,
+ timestamp end);
+
+QueryDataRet ts_reader_read(CTsFileReader reader, const char* table_name,
+ char** columns, int colum_num);
+
+ErrorCode destory_query_dataret(QueryDataRet query_data_set);
+
+DataResult* ts_next(QueryDataRet data, int expect_line_count);
+
+void print_data_result(DataResult* result);
+
+void clean_data_record(DataResult data_result);
+void clean_query_ret(QueryDataRet query_data_set);
+void clean_query_tree(Expression* expression);
+
+#ifdef __cplusplus
+}
+#endif
+#endif // CWRAPPER_TSFILE_CWRAPPER_H