From 556a6fcd7fc27f9da69fb590add4b53b1ce55efb Mon Sep 17 00:00:00 2001
From: Hongze Cheng <hzcheng@taosdata.com>
Date: Tue, 20 Sep 2022 14:49:34 +0800
Subject: [PATCH] more code

---
 source/dnode/vnode/src/tsdb/tsdbDiskData.c | 205 ++++++++++++++++++++-
 1 file changed, 202 insertions(+), 3 deletions(-)

diff --git a/source/dnode/vnode/src/tsdb/tsdbDiskData.c b/source/dnode/vnode/src/tsdb/tsdbDiskData.c
index 0a42edc94c..b7b03e8288 100644
--- a/source/dnode/vnode/src/tsdb/tsdbDiskData.c
+++ b/source/dnode/vnode/src/tsdb/tsdbDiskData.c
@@ -16,14 +16,213 @@
 #include "tsdb.h"
 
 typedef struct SDiskData SDiskData;
+typedef struct SDiskCol  SDiskCol;
 
+struct SDiskCol {
+  int16_t      cid;
+  int8_t       type;
+  int8_t       flag;
+  int32_t      nVal;
+  SCompressor *pBitC;
+  SCompressor *pOffC;
+  SCompressor *pValC;
+};
+
+// SDiskCol ================================================
+static int32_t tDiskColReset(SDiskCol *pDiskCol, int16_t cid, int8_t type, uint8_t cmprAlg) {
+  int32_t code = 0;
+
+  pDiskCol->cid = cid;
+  pDiskCol->type = type;
+  pDiskCol->flag = 0;
+  pDiskCol->nVal = 0;
+
+  tCompressorReset(pDiskCol->pBitC, TSDB_DATA_TYPE_TINYINT, cmprAlg, 1);
+  tCompressorReset(pDiskCol->pOffC, TSDB_DATA_TYPE_INT, cmprAlg, 1);
+  tCompressorReset(pDiskCol->pValC, type, cmprAlg, 1);
+
+  return code;
+}
+
+static int32_t tDiskColAddVal0(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (pColVal->isNone) {
+    pDiskCol->flag = HAS_NONE;
+  } else if (pColVal->isNull) {
+    pDiskCol->flag = HAS_NULL;
+  } else {
+    pDiskCol->flag = HAS_VALUE;
+    code = tCompress(pDiskCol->pValC, pColVal->value.pData, pColVal->value.nData /*TODO*/);
+    if (code) goto _exit;
+  }
+  pDiskCol->nVal++;
+
+_exit:
+  return code;
+}
+static int32_t tDiskColAddVal1(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (!pColVal->isNone) {
+  }
+  pDiskCol->nVal++;
+
+_exit:
+  return code;
+}
+static int32_t tDiskColAddVal2(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (!pColVal->isNull) {
+  }
+  pDiskCol->nVal++;
+
+_exit:
+  return code;
+}
+static int32_t tDiskColAddVal3(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (pColVal->isNone) {
+  } else if (pColVal->isNull) {
+  } else {
+  }
+  pDiskCol->nVal++;
+
+  return code;
+}
+static int32_t tDiskColAddVal4(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (pColVal->isNone || pColVal->isNull) {
+  } else {
+  }
+
+  pDiskCol->nVal++;
+
+_exit:
+  return code;
+}
+static int32_t tDiskColAddVal5(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (pColVal->isNull) {
+  } else {
+  }
+  pDiskCol->nVal++;
+
+  return code;
+}
+static int32_t tDiskColAddVal6(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (pColVal->isNone) {
+  } else {
+  }
+  pDiskCol->nVal++;
+
+  return code;
+}
+static int32_t tDiskColAddVal7(SDiskCol *pDiskCol, SColVal *pColVal) {
+  int32_t code = 0;
+
+  if (pColVal->isNone) {
+  } else if (pColVal->isNull) {
+  } else {
+  }
+  pDiskCol->nVal++;
+
+  return code;
+}
+static int32_t (*tDiskColAddValImpl[])(SDiskCol *pDiskCol, SColVal *pColVal) = {
+    tDiskColAddVal0,  // 0
+    tDiskColAddVal1,  // HAS_NONE
+    tDiskColAddVal2,  // HAS_NULL
+    tDiskColAddVal3,  // HAS_NULL|HAS_NONE
+    tDiskColAddVal4,  // HAS_VALUE
+    tDiskColAddVal5,  // HAS_VALUE|HAS_NONE
+    tDiskColAddVal6,  // HAS_VALUE|HAS_NULL
+    tDiskColAddVal7,  // HAS_VALUE|HAS_NULL|HAS_NONE
+};
+
+// SDiskData ================================================
 struct SDiskData {
-  SDiskDataHdr hdr;
-  /* data */
+  int64_t      suid;
+  int64_t      uid;
+  uint8_t      cmprAlg;
+  SCompressor *pUidC;
+  SCompressor *pVerC;
+  SCompressor *pKeyC;
+  int32_t      nDiskCol;
+  SArray      *aDiskCol;
 };
 
-int32_t tDiskDataAddRow(SDiskData *pDiskData, TSDBROW *pRow) {
+int32_t tDiskDataCreate(SDiskData *pDiskData) {
   int32_t code = 0;
   // TODO
   return code;
 }
+
+int32_t tDiskDataDestroy(SDiskData *pDiskData) {
+  int32_t code = 0;
+  // TODO
+  return code;
+}
+
+int32_t tDiskDataReset(SDiskData *pDiskData, STSchema *pTSchema, TABLEID *pId, uint8_t cmprAlg) {
+  int32_t code = 0;
+
+  pDiskData->suid = pId->suid;
+  pDiskData->uid = pId->uid;
+  pDiskData->cmprAlg = cmprAlg;
+
+  for (int32_t iCol = 1; iCol < pTSchema->numOfCols; iCol++) {
+  }
+
+  return code;
+}
+
+int32_t tDiskDataAddRow(SDiskData *pDiskData, TSDBROW *pRow, STSchema *pTSchema, TABLEID *pId) {
+  int32_t code = 0;
+
+  ASSERT(pId->suid == pDiskData->suid);
+
+  // uid
+  code = tCompress(pDiskData->pUidC, &pId->uid, sizeof(int64_t));
+  if (code) goto _exit;
+
+  // version
+  int64_t version = TSDBROW_VERSION(pRow);
+  code = tCompress(pDiskData->pVerC, &version, sizeof(int64_t));
+  if (code) goto _exit;
+
+  // TSKEY
+  TSKEY ts = TSDBROW_TS(pRow);
+  code = tCompress(pDiskData->pVerC, &ts, sizeof(int64_t));
+  if (code) goto _exit;
+
+  SRowIter iter = {0};
+  tRowIterInit(&iter, pRow, pTSchema);
+
+  SColVal *pColVal = tRowIterNext(&iter);
+  for (int32_t iDiskCol = 0; iDiskCol < pDiskData->nDiskCol; iDiskCol++) {
+    SDiskCol *pDiskCol = (SDiskCol *)taosArrayGet(pDiskData->aDiskCol, iDiskCol);
+
+    while (pColVal && pColVal->cid < pDiskCol->cid) {
+      pColVal = tRowIterNext(&iter);
+    }
+
+    if (pColVal == NULL || pColVal->cid > pDiskCol->cid) {
+      code = tDiskColAddValImpl[pDiskCol->flag](pDiskCol, &COL_VAL_NONE(pDiskCol->cid, pDiskCol->type));
+      if (code) goto _exit;
+    } else {
+      code = tDiskColAddValImpl[pDiskCol->flag](pDiskCol, pColVal);
+      if (code) goto _exit;
+      pColVal = tRowIterNext(&iter);
+    }
+  }
+
+_exit:
+  return code;
+}
-- 
GitLab