command.c

/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "command.h"
#include "tdatablock.h"

static int32_t getSchemaBytes(const SSchema* pSchema) {
  switch (pSchema->type) {
    case TSDB_DATA_TYPE_BINARY:
      return (pSchema->bytes - VARSTR_HEADER_SIZE);
    case TSDB_DATA_TYPE_NCHAR:
      return (pSchema->bytes - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE;
    default:
      return pSchema->bytes;
  }
}

static void buildRspData(const STableMeta* pMeta, char* pData) {
  int32_t* pColSizes = (int32_t*)pData;
  pData += DESCRIBE_RESULT_COLS * sizeof(int32_t);
  int32_t numOfRows = TABLE_TOTAL_COL_NUM(pMeta);

  // Field
  int32_t* pOffset = (int32_t*)pData;
  pData += numOfRows * sizeof(int32_t);
  for (int32_t i = 0; i < numOfRows; ++i) {
    STR_TO_VARSTR(pData, pMeta->schema[i].name);
    int16_t len = varDataTLen(pData);
    pData += len;
    *pOffset = pColSizes[0];
    pOffset += 1;
    pColSizes[0] += len;
  }
  
  // Type
  pOffset = (int32_t*)pData;
  pData += numOfRows * sizeof(int32_t);
  for (int32_t i = 0; i < numOfRows; ++i) {
    STR_TO_VARSTR(pData, tDataTypes[pMeta->schema[i].type].name);
    int16_t len = varDataTLen(pData);
    pData += len;
    *pOffset = pColSizes[1];
    pOffset += 1;
    pColSizes[1] += len;
  }

  // Length
  pData += BitmapLen(numOfRows);
  for (int32_t i = 0; i < numOfRows; ++i) {
    *(int32_t*)pData = getSchemaBytes(pMeta->schema + i);
    pData += sizeof(int32_t);
  }
  pColSizes[2] = sizeof(int32_t) * numOfRows;

  // Note
  pOffset = (int32_t*)pData;
  pData += numOfRows * sizeof(int32_t);
  for (int32_t i = 0; i < numOfRows; ++i) {
    STR_TO_VARSTR(pData, i >= pMeta->tableInfo.numOfColumns ? "TAG" : "");
    int16_t len = varDataTLen(pData);
    pData += len;
    *pOffset = pColSizes[3];
    pOffset += 1;
    pColSizes[3] += len;
  }

  for (int32_t i = 0; i < DESCRIBE_RESULT_COLS; ++i) {
    pColSizes[i] = htonl(pColSizes[i]);
  }
}

static int32_t calcRspSize(const STableMeta* pMeta) {
  int32_t numOfRows = TABLE_TOTAL_COL_NUM(pMeta);
  return sizeof(SRetrieveTableRsp) + 
      (numOfRows * sizeof(int32_t) + numOfRows * DESCRIBE_RESULT_FIELD_LEN) + 
      (numOfRows * sizeof(int32_t) + numOfRows * DESCRIBE_RESULT_TYPE_LEN) +
      (BitmapLen(numOfRows) + numOfRows * sizeof(int32_t)) + 
      (numOfRows * sizeof(int32_t) + numOfRows * DESCRIBE_RESULT_NOTE_LEN);
}

static int32_t execDescribe(SNode* pStmt, SRetrieveTableRsp** pRsp) {
  SDescribeStmt* pDesc = (SDescribeStmt*)pStmt;
  *pRsp = taosMemoryCalloc(1, calcRspSize(pDesc->pMeta));
  if (NULL == *pRsp) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  (*pRsp)->useconds = 0;
  (*pRsp)->completed = 1;
  (*pRsp)->precision = 0;
  (*pRsp)->compressed = 0;
  (*pRsp)->compLen = 0;
  (*pRsp)->numOfRows = htonl(TABLE_TOTAL_COL_NUM(pDesc->pMeta));
  buildRspData(pDesc->pMeta, (*pRsp)->data);
  return TSDB_CODE_SUCCESS;
}

static int32_t execResetQueryCache() {
  // todo
  return TSDB_CODE_SUCCESS;
}

int32_t qExecCommand(SNode* pStmt, SRetrieveTableRsp** pRsp) {
  switch (nodeType(pStmt)) {
    case QUERY_NODE_DESCRIBE_STMT:
      return execDescribe(pStmt, pRsp);
    case QUERY_NODE_RESET_QUERY_CACHE_STMT:
      return execResetQueryCache();
    default:
      break;
  }
  return TSDB_CODE_FAILED;
}