(tsfile) branch develop updated: Fix/cpp get device schema (#642)

[hongzhigao]

This is an automated email from the ASF dual-hosted git repository.

hongzhigao 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 1bb3780b Fix/cpp get device schema (#642)
1bb3780b is described below

commit 1bb3780bd21ff76b3396b72014b748fc49075f80
Author: Hongzhi Gao <761417...@qq.com>
AuthorDate: Thu Nov 20 10:01:00 2025 +0800

    Fix/cpp get device schema (#642)
    
    * fix tree-interface get_device_schema
    
    * fix tree-interface get_device_schema
---
 cpp/src/file/tsfile_io_reader.cc                   | 27 ++++++++++++++++------
 .../reader/tree_view/tsfile_reader_tree_test.cc    |  7 ++++++
 2 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/cpp/src/file/tsfile_io_reader.cc b/cpp/src/file/tsfile_io_reader.cc
index c70c429e..273f09a4 100644
--- a/cpp/src/file/tsfile_io_reader.cc
+++ b/cpp/src/file/tsfile_io_reader.cc
@@ -701,13 +701,26 @@ int TsFileIOReader::do_load_all_timeseries_index(
     for (const auto &index_node_entry : index_node_entry_list) {
         int64_t start_offset = index_node_entry.first->get_offset(),
                 end_offset = index_node_entry.second;
-        const std::string target_measurement_name(
-            index_node_entry.first->get_name().to_std_string());
-        ITimeseriesIndex *ts_idx;
-        ret = do_load_timeseries_index(target_measurement_name, start_offset,
-                                       end_offset, in_timeseries_index_pa,
-                                       ts_idx);
-        if (IS_SUCC(ret)) {
+        int32_t read_size = (int32_t)(end_offset - start_offset);
+        int32_t ret_read_len = 0;
+        char *ti_buf = in_timeseries_index_pa.alloc(read_size);
+        if (IS_NULL(ti_buf)) {
+            return E_OOM;
+        }
+        if (RET_FAIL(read_file_->read(start_offset, ti_buf, read_size,
+                                      ret_read_len))) {
+            return ret;
+        }
+        ByteStream bs;
+        bs.wrap_from(ti_buf, read_size);
+        while (bs.has_remaining()) {
+            void *buf = in_timeseries_index_pa.alloc(sizeof(TimeseriesIndex));
+            auto ts_idx = new (buf) TimeseriesIndex;
+            if (RET_FAIL(
+                    ts_idx->deserialize_from(bs, &in_timeseries_index_pa))) {
+                return ret;
+            }
+            if (ts_idx->get_measurement_name().len_ == 0) continue;
             ts_indexs.push_back(ts_idx);
         }
     }
diff --git a/cpp/test/reader/tree_view/tsfile_reader_tree_test.cc 
b/cpp/test/reader/tree_view/tsfile_reader_tree_test.cc
index e7466c7b..477ab24b 100644
--- a/cpp/test/reader/tree_view/tsfile_reader_tree_test.cc
+++ b/cpp/test/reader/tree_view/tsfile_reader_tree_test.cc
@@ -113,6 +113,7 @@ TEST_F(TsFileTreeReaderTest, ExtendedRowsAndColumnsTest) {
     std::vector<std::string> device_ids = {"device_1", "device_2", "device_3"};
     std::vector<std::string> measurement_ids = {"temperature", "humidity",
                                                 "pressure", "voltage"};
+    std::sort(measurement_ids.begin(), measurement_ids.end());
     std::vector<TSDataType> data_types = {INT64, DOUBLE, FLOAT, INT32};
     std::vector<MeasurementSchema*> measurements;
     for (size_t i = 0; i < measurement_ids.size(); ++i) {
@@ -168,6 +169,12 @@ TEST_F(TsFileTreeReaderTest, ExtendedRowsAndColumnsTest) {
         EXPECT_EQ(read_device_ids[i], device_ids[i]);
     }
 
+    auto device_schema = reader.get_device_schema(device_ids[0]);
+    for (int i = 0; i < measurements.size(); ++i) {
+        EXPECT_EQ(measurements[i]->measurement_name_,
+                  device_schema[i].measurement_name_);
+    }
+
     ResultSet* result;
     int ret =
         reader.query(device_ids, measurement_ids, 0, NUM_ROWS * 1000, result);