qFill.h 3.6 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

H
Haojun Liao 已提交
16 17
#ifndef TDENGINE_QFILL_H
#define TDENGINE_QFILL_H
H
hzcheng 已提交
18 19 20 21 22

#ifdef __cplusplus
extern "C" {
#endif

23
#include "os.h"
H
Haojun Liao 已提交
24
#include "qExtbuffer.h"
25 26
#include "taosdef.h"

27 28 29 30
typedef struct {
  STColumn col;             // column info
  int16_t  functionId;      // sql function id
  int16_t  flag;            // column flag: TAG COLUMN|NORMAL COLUMN
H
Haojun Liao 已提交
31
  int16_t  tagIndex;        // index of current tag in SFillTagColInfo array list
32
  union {int64_t i; double d;} fillVal;
33
} SFillColInfo;
34 35 36 37 38

typedef struct {
  SSchema col;
  char*   tagVal;
} SFillTagColInfo;
39 40
  
typedef struct SFillInfo {
H
Haojun Liao 已提交
41 42 43 44 45 46 47 48 49 50 51 52 53
  TSKEY     start;                // start timestamp
  TSKEY     end;                  // endKey for fill
  TSKEY     currentKey;           // current active timestamp, the value may be changed during the fill procedure.
  int32_t   order;                // order [TSDB_ORDER_ASC|TSDB_ORDER_DESC]
  int32_t   type;                 // fill type
  int32_t   numOfRows;            // number of rows in the input data block
  int32_t   index;                // active row index
  int32_t   numOfTotal;           // number of filled rows in one round
  int32_t   numOfCurrent;         // number of filled rows in current results

  int32_t   numOfTags;            // number of tags
  int32_t   numOfCols;            // number of columns, including the tags columns
  int32_t   rowSize;              // size of each row
54
  SInterval interval;
H
Haojun Liao 已提交
55 56 57 58 59 60 61 62 63
  char *    prevValues;           // previous row of data, to generate the interpolation results
  char *    nextValues;           // next row of data
  char**    pData;                // original result data block involved in filling data
  int32_t   alloc;                // data buffer size in rows
  int8_t    precision;            // time resoluation

  SFillColInfo* pFillCol;         // column info for fill operations
  SFillTagColInfo* pTags;         // tags value for filling gap
  void*     handle;               // for dubug purpose
64
} SFillInfo;
H
hzcheng 已提交
65 66 67 68 69 70

typedef struct SPoint {
  int64_t key;
  void *  val;
} SPoint;

H
Haojun Liao 已提交
71 72
SFillInfo* taosInitFillInfo(int32_t order, TSKEY skey, int32_t numOfTags, int32_t capacity, int32_t numOfCols,
                            int64_t slidingTime, int8_t slidingUnit, int8_t precision, int32_t fillType,
H
Haojun Liao 已提交
73
                            SFillColInfo* pFillCol, void* handle);
H
hzcheng 已提交
74

75
void taosResetFillInfo(SFillInfo* pFillInfo, TSKEY startTimestamp);
H
hjxilinx 已提交
76

77
void* taosDestroyFillInfo(SFillInfo *pFillInfo);
H
hzcheng 已提交
78

79
void taosFillSetStartInfo(SFillInfo* pFillInfo, int32_t numOfRows, TSKEY endKey);
H
hzcheng 已提交
80

H
Haojun Liao 已提交
81
void taosFillCopyInputDataFromFilePage(SFillInfo* pFillInfo, const tFilePage** pInput);
H
hzcheng 已提交
82

H
Haojun Liao 已提交
83
void taosFillCopyInputDataFromOneFilePage(SFillInfo* pFillInfo, const tFilePage* pInput);
H
hzcheng 已提交
84

85
bool taosFillHasMoreResults(SFillInfo* pFillInfo);
H
hzcheng 已提交
86

87
int64_t getNumOfResultsAfterFillGap(SFillInfo* pFillInfo, int64_t ekey, int32_t maxNumOfRows);
88

H
Haojun Liao 已提交
89
int32_t taosGetLinearInterpolationVal(int32_t type, SPoint *point1, SPoint *point2, SPoint *point);
H
hzcheng 已提交
90

H
Haojun Liao 已提交
91
int64_t taosFillResultDataBlock(SFillInfo* pFillInfo, tFilePage** output, int32_t capacity);
92

H
hzcheng 已提交
93 94 95 96
#ifdef __cplusplus
}
#endif

H
Haojun Liao 已提交
97
#endif  // TDENGINE_QFILL_H