more code

source/dnode/vnode/src/tsdb/dev/inc/tsdbDataFReaderWriter.h

@@ -30,6 +30,11 @@ typedef struct SDataFileReaderConfig SDataFileReaderConfig;
 typedef struct SDataFileWriter       SDataFileWriter;
 typedef struct SDataFileWriterConfig SDataFileWriterConfig;
 
+int32_t tsdbDataFileWriterOpen(const SDataFileWriterConfig *config, SDataFileWriter **ppWriter);
+int32_t tsdbDataFileWriterClose(SDataFileWriter *pWriter);
+int32_t tsdbDataFileWriteTSData(SDataFileWriter *pWriter, SBlockData *pBlockData);
+int32_t tsdbDataFileWriteTSDataBlock(SDataFileWriter *pWriter, SBlockData *pBlockData);
+
 struct SDataFileReaderConfig {
   STsdb *pTsdb;
   // TODO

source/dnode/vnode/src/tsdb/dev/inc/tsdbSttFReaderWriter.h

@@ -22,7 +22,7 @@
 extern "C" {
 #endif
 
-// SSttFReader ==========================================
+// SSttFileReader ==========================================
 typedef struct SSttFSegReader       SSttFSegReader;
 typedef struct SSttFileReader       SSttFileReader;
 typedef struct SSttFileReaderConfig SSttFileReaderConfig;
@@ -50,7 +50,7 @@ struct SSttFileReaderConfig {
   // TODO
 };
 
-// SSttFWriter ==========================================
+// SSttFileWriter ==========================================
 typedef struct SSttFileWriter       SSttFileWriter;
 typedef struct SSttFileWriterConfig SSttFileWriterConfig;
 

source/dnode/vnode/src/tsdb/dev/tsdbDataFReaderWriter.c

@@ -24,3 +24,51 @@ struct SDataFileReader {
 struct SDataFileWriter {
   // TODO
 };
+
+int32_t tsdbDataFileWriterOpen(const SDataFileWriterConfig *config, SDataFileWriter **ppWriter) {
+  int32_t code = 0;
+  int32_t lino = 0;
+  int32_t vid = TD_VID(config->pTsdb->pVnode);
+// TODO
+_exit:
+  if (code) {
+    tsdbError("vgId:%d %s failed at line %d since %s", vid, __func__, lino, tstrerror(code));
+  }
+  return code;
+}
+
+int32_t tsdbDataFileWriterClose(SDataFileWriter *pWriter) {
+  int32_t code = 0;
+  int32_t lino = 0;
+  int32_t vid = 0;  // TODO: TD_VID(config->pTsdb->pVnode);
+// TODO
+_exit:
+  if (code) {
+    tsdbError("vgId:%d %s failed at line %d since %s", vid, __func__, lino, tstrerror(code));
+  }
+  return code;
+}
+
+int32_t tsdbDataFileWriteTSData(SDataFileWriter *pWriter, SBlockData *pBlockData) {
+  int32_t code = 0;
+  int32_t lino = 0;
+  int32_t vid = 0;  // TODO: TD_VID(config->pTsdb->pVnode);
+// TODO
+_exit:
+  if (code) {
+    tsdbError("vgId:%d %s failed at line %d since %s", vid, __func__, lino, tstrerror(code));
+  }
+  return code;
+}
+
+int32_t tsdbDataFileWriteTSDataBlock(SDataFileWriter *pWriter, SBlockData *pBlockData) {
+  int32_t code = 0;
+  int32_t lino = 0;
+  int32_t vid = 0;  // TODO: TD_VID(config->pTsdb->pVnode);
+// TODO
+_exit:
+  if (code) {
+    tsdbError("vgId:%d %s failed at line %d since %s", vid, __func__, lino, tstrerror(code));
+  }
+  return code;
+}

source/dnode/vnode/src/tsdb/dev/tsdbMerge.c

@@ -198,8 +198,12 @@ static int32_t tsdbMergeFileSetBegin(SMerger *merger) {
 
       // open the reader
       SSttFileReader      *reader;
-      // code = tsdbSttFileReaderOpen(&fobj->f.stt, &reader);
+      SSttFileReaderConfig config = {
-      // TSDB_CHECK_CODE(code, lino, _exit);
+          .pTsdb = merger->tsdb,
+          // TODO
+      };
+      code = tsdbSttFReaderOpen(&config, &reader);
+      TSDB_CHECK_CODE(code, lino, _exit);
 
       code = TARRAY2_APPEND(&merger->sttReaderArr, reader);
       TSDB_CHECK_CODE(code, lino, _exit);
@@ -216,6 +220,7 @@ static int32_t tsdbMergeFileSetBegin(SMerger *merger) {
   }
 
   // open stt file writer
+  if (lvl) {
     SSttFileWriterConfig config = {
         .pTsdb = merger->tsdb,
         .maxRow = merger->maxRow,
@@ -224,26 +229,41 @@ static int32_t tsdbMergeFileSetBegin(SMerger *merger) {
         .pSkmTb = &merger->skmTb,
         .pSkmRow = &merger->skmRow,
         .aBuf = merger->aBuf,
+        .file = fobj->f,
     };
-  if (lvl) {
+    code = tsdbSttFWriterOpen(&config, &merger->sttWriter);
-    config.file = fobj->f;
+    TSDB_CHECK_CODE(code, lino, _exit);
   } else {
-    config.file = (STFile){
+    SSttFileWriterConfig config = {
+        .pTsdb = merger->tsdb,
+        .maxRow = merger->maxRow,
+        .szPage = merger->szPage,
+        .cmprAlg = merger->cmprAlg,
+        .pSkmTb = &merger->skmTb,
+        .pSkmRow = &merger->skmRow,
+        .aBuf = merger->aBuf,
+        .file =
+            (STFile){
                 .type = TSDB_FTYPE_STT,
                 .did = {.level = 0, .id = 0},
                 .fid = fset->fid,
                 .cid = merger->cid,
                 .size = 0,
                 .stt = {.level = merger->ctx.level, .nseg = 0},
+            },
     };
-  }
     code = tsdbSttFWriterOpen(&config, &merger->sttWriter);
     TSDB_CHECK_CODE(code, lino, _exit);
+  }
 
   // open data file writer
   if (merger->ctx.toData) {
-    // code = tsdbDataFWriterOpen();
+    SDataFileWriterConfig config = {
-    // TSDB_CHECK_CODE(code, lino, _exit);
+        .pTsdb = merger->tsdb,
+        // TODO
+    };
+    code = tsdbDataFileWriterOpen(&config, &merger->dataWriter);
+    TSDB_CHECK_CODE(code, lino, _exit);
   }
 
 _exit: