Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
慢慢CG
TDengine
提交
f2eb4100
T
TDengine
项目概览
慢慢CG
/
TDengine
与 Fork 源项目一致
Fork自
taosdata / TDengine
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
TDengine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
f2eb4100
编写于
10月 13, 2020
作者:
H
Hongze Cheng
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refactor more code
上级
be7aaccb
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
130 addition
and
130 deletion
+130
-130
src/tsdb/inc/tsdbMain.h
src/tsdb/inc/tsdbMain.h
+6
-6
src/tsdb/src/tsdbMain.c
src/tsdb/src/tsdbMain.c
+1
-1
src/tsdb/src/tsdbRWHelper.c
src/tsdb/src/tsdbRWHelper.c
+109
-109
src/tsdb/src/tsdbRead.c
src/tsdb/src/tsdbRead.c
+14
-14
未找到文件。
src/tsdb/inc/tsdbMain.h
浏览文件 @
f2eb4100
...
...
@@ -229,7 +229,7 @@ typedef struct {
uint32_t
numOfBlocks
:
30
;
uint64_t
uid
;
TSKEY
maxKey
;
}
S
Comp
Idx
;
}
S
Block
Idx
;
typedef
struct
{
int64_t
last
:
1
;
...
...
@@ -288,7 +288,7 @@ typedef struct {
}
SHelperTable
;
typedef
struct
{
S
Comp
Idx
*
pIdxArray
;
S
Block
Idx
*
pIdxArray
;
int
numOfIdx
;
int
curIdx
;
}
SIdxH
;
...
...
@@ -301,7 +301,7 @@ typedef struct {
// For file set usage
SHelperFile
files
;
SIdxH
idxH
;
S
Comp
Idx
curCompIdx
;
S
Block
Idx
curCompIdx
;
void
*
pWIdx
;
// For table set usage
SHelperTable
tableInfo
;
...
...
@@ -318,7 +318,7 @@ typedef struct {
typedef
struct
{
SFileGroup
fGroup
;
int
numOfIdx
;
S
Comp
Idx
*
pCompIdx
;
S
Block
Idx
*
pCompIdx
;
SBlockInfo
*
pCompInfo
;
void
*
pBuf
;
FILE
*
tLogStream
;
...
...
@@ -521,9 +521,9 @@ int tsdbMoveLastBlockIfNeccessary(SRWHelper* pHelper);
int
tsdbWriteCompInfo
(
SRWHelper
*
pHelper
);
int
tsdbWriteCompIdx
(
SRWHelper
*
pHelper
);
int
tsdbLoadCompIdxImpl
(
SFile
*
pFile
,
uint32_t
offset
,
uint32_t
len
,
void
*
buffer
);
int
tsdbDecodeSCompIdxImpl
(
void
*
buffer
,
uint32_t
len
,
S
Comp
Idx
**
ppCompIdx
,
int
*
numOfIdx
);
int
tsdbDecodeSCompIdxImpl
(
void
*
buffer
,
uint32_t
len
,
S
Block
Idx
**
ppCompIdx
,
int
*
numOfIdx
);
int
tsdbLoadCompIdx
(
SRWHelper
*
pHelper
,
void
*
target
);
int
tsdbLoadCompInfoImpl
(
SFile
*
pFile
,
S
Comp
Idx
*
pIdx
,
SBlockInfo
**
ppCompInfo
);
int
tsdbLoadCompInfoImpl
(
SFile
*
pFile
,
S
Block
Idx
*
pIdx
,
SBlockInfo
**
ppCompInfo
);
int
tsdbLoadCompInfo
(
SRWHelper
*
pHelper
,
void
*
target
);
int
tsdbLoadCompData
(
SRWHelper
*
phelper
,
SBlock
*
pcompblock
,
void
*
target
);
void
tsdbGetDataStatis
(
SRWHelper
*
pHelper
,
SDataStatis
*
pStatis
,
int
numOfCols
);
...
...
src/tsdb/src/tsdbMain.c
浏览文件 @
f2eb4100
...
...
@@ -782,7 +782,7 @@ static int tsdbRestoreInfo(STsdbRepo *pRepo) {
STable
*
pTable
=
pMeta
->
tables
[
i
];
if
(
pTable
==
NULL
)
continue
;
if
(
tsdbSetHelperTable
(
&
rhelper
,
pTable
,
pRepo
)
<
0
)
goto
_err
;
S
Comp
Idx
*
pIdx
=
&
(
rhelper
.
curCompIdx
);
S
Block
Idx
*
pIdx
=
&
(
rhelper
.
curCompIdx
);
if
(
pIdx
->
offset
>
0
&&
pTable
->
lastKey
<
pIdx
->
maxKey
)
pTable
->
lastKey
=
pIdx
->
maxKey
;
}
...
...
src/tsdb/src/tsdbRWHelper.c
浏览文件 @
f2eb4100
...
...
@@ -29,13 +29,13 @@
#define TSDB_GET_COMPBLOCK_IDX(h, b) (POINTER_DISTANCE(b, (h)->pCompInfo->blocks)/sizeof(SCompBlock))
static
bool
tsdbShouldCreateNewLast
(
SRWHelper
*
pHelper
);
static
int
tsdbWriteBlockToFile
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
SDataCols
*
pDataCols
,
S
Comp
Block
*
pCompBlock
,
static
int
tsdbWriteBlockToFile
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
SDataCols
*
pDataCols
,
SBlock
*
pCompBlock
,
bool
isLast
,
bool
isSuperBlock
);
static
int
compareKeyBlock
(
const
void
*
arg1
,
const
void
*
arg2
);
static
int
tsdbAdjustInfoSizeIfNeeded
(
SRWHelper
*
pHelper
,
size_t
esize
);
static
int
tsdbInsertSuperBlock
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
int
blkIdx
);
static
int
tsdbAddSubBlock
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
int
blkIdx
,
int
rowsAdded
);
static
int
tsdbUpdateSuperBlock
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
int
blkIdx
);
static
int
tsdbInsertSuperBlock
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
int
blkIdx
);
static
int
tsdbAddSubBlock
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
int
blkIdx
,
int
rowsAdded
);
static
int
tsdbUpdateSuperBlock
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
int
blkIdx
);
static
void
tsdbResetHelperFileImpl
(
SRWHelper
*
pHelper
);
static
int
tsdbInitHelperFile
(
SRWHelper
*
pHelper
);
static
void
tsdbDestroyHelperFile
(
SRWHelper
*
pHelper
);
...
...
@@ -49,16 +49,16 @@ static int tsdbInitHelperBlock(SRWHelper *pHelper);
static
int
tsdbInitHelper
(
SRWHelper
*
pHelper
,
STsdbRepo
*
pRepo
,
tsdb_rw_helper_t
type
);
static
int
tsdbCheckAndDecodeColumnData
(
SDataCol
*
pDataCol
,
char
*
content
,
int32_t
len
,
int8_t
comp
,
int
numOfRows
,
int
maxPoints
,
char
*
buffer
,
int
bufferSize
);
static
int
tsdbLoadBlockDataColsImpl
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
SDataCols
*
pDataCols
,
int16_t
*
colIds
,
static
int
tsdbLoadBlockDataColsImpl
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
SDataCols
*
pDataCols
,
int16_t
*
colIds
,
int
numOfColIds
);
static
int
tsdbLoadBlockDataImpl
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
SDataCols
*
pDataCols
);
static
int
tsdbEncodeSCompIdx
(
void
**
buf
,
S
Comp
Idx
*
pIdx
);
static
void
*
tsdbDecodeSCompIdx
(
void
*
buf
,
S
Comp
Idx
*
pIdx
);
static
int
tsdbLoadBlockDataImpl
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
SDataCols
*
pDataCols
);
static
int
tsdbEncodeSCompIdx
(
void
**
buf
,
S
Block
Idx
*
pIdx
);
static
void
*
tsdbDecodeSCompIdx
(
void
*
buf
,
S
Block
Idx
*
pIdx
);
static
int
tsdbProcessAppendCommit
(
SRWHelper
*
pHelper
,
SCommitIter
*
pCommitIter
,
SDataCols
*
pDataCols
,
TSKEY
maxKey
);
static
void
tsdbDestroyHelperBlock
(
SRWHelper
*
pHelper
);
static
int
tsdbLoadColData
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
S
CompBlock
*
pCompBlock
,
SComp
Col
*
pCompCol
,
static
int
tsdbLoadColData
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
S
Block
*
pCompBlock
,
SBlock
Col
*
pCompCol
,
SDataCol
*
pDataCol
);
static
int
tsdbWriteBlockToProperFile
(
SRWHelper
*
pHelper
,
SDataCols
*
pDataCols
,
S
Comp
Block
*
pCompBlock
);
static
int
tsdbWriteBlockToProperFile
(
SRWHelper
*
pHelper
,
SDataCols
*
pDataCols
,
SBlock
*
pCompBlock
);
static
int
tsdbProcessMergeCommit
(
SRWHelper
*
pHelper
,
SCommitIter
*
pCommitIter
,
SDataCols
*
pDataCols
,
TSKEY
maxKey
,
int
*
blkIdx
);
static
int
tsdbLoadAndMergeFromCache
(
SDataCols
*
pDataCols
,
int
*
iter
,
SCommitIter
*
pCommitIter
,
SDataCols
*
pTarget
,
...
...
@@ -233,28 +233,28 @@ int tsdbSetHelperTable(SRWHelper *pHelper, STable *pTable, STsdbRepo *pRepo) {
if
(
pHelper
->
idxH
.
numOfIdx
>
0
)
{
while
(
true
)
{
if
(
pHelper
->
idxH
.
curIdx
>=
pHelper
->
idxH
.
numOfIdx
)
{
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Comp
Idx
));
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Block
Idx
));
break
;
}
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
idxH
.
pIdxArray
[
pHelper
->
idxH
.
curIdx
]);
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
idxH
.
pIdxArray
[
pHelper
->
idxH
.
curIdx
]);
if
(
pIdx
->
tid
==
TABLE_TID
(
pTable
))
{
if
(
pIdx
->
uid
==
TABLE_UID
(
pTable
))
{
pHelper
->
curCompIdx
=
*
pIdx
;
}
else
{
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Comp
Idx
));
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Block
Idx
));
}
pHelper
->
idxH
.
curIdx
++
;
break
;
}
else
if
(
pIdx
->
tid
>
TABLE_TID
(
pTable
))
{
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Comp
Idx
));
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Block
Idx
));
break
;
}
else
{
pHelper
->
idxH
.
curIdx
++
;
}
}
}
else
{
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Comp
Idx
));
memset
(
&
(
pHelper
->
curCompIdx
),
0
,
sizeof
(
S
Block
Idx
));
}
if
(
helperType
(
pHelper
)
==
TSDB_WRITE_HELPER
&&
pHelper
->
curCompIdx
.
hasLast
)
{
...
...
@@ -270,7 +270,7 @@ int tsdbSetHelperTable(SRWHelper *pHelper, STable *pTable, STsdbRepo *pRepo) {
int
tsdbCommitTableData
(
SRWHelper
*
pHelper
,
SCommitIter
*
pCommitIter
,
SDataCols
*
pDataCols
,
TSKEY
maxKey
)
{
ASSERT
(
helperType
(
pHelper
)
==
TSDB_WRITE_HELPER
);
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
int
blkIdx
=
0
;
ASSERT
(
pIdx
->
offset
==
0
||
pIdx
->
uid
==
TABLE_UID
(
pCommitIter
->
pTable
));
...
...
@@ -296,12 +296,12 @@ int tsdbMoveLastBlockIfNeccessary(SRWHelper *pHelper) {
STsdbCfg
*
pCfg
=
&
pHelper
->
pRepo
->
config
;
ASSERT
(
helperType
(
pHelper
)
==
TSDB_WRITE_HELPER
);
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Comp
Block
compBlock
=
{
0
};
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
SBlock
compBlock
=
{
0
};
if
(
TSDB_NLAST_FILE_OPENED
(
pHelper
)
&&
(
pHelper
->
hasOldLastBlock
))
{
if
(
tsdbLoadCompInfo
(
pHelper
,
NULL
)
<
0
)
return
-
1
;
S
Comp
Block
*
pCompBlock
=
blockAtIdx
(
pHelper
,
pIdx
->
numOfBlocks
-
1
);
SBlock
*
pCompBlock
=
blockAtIdx
(
pHelper
,
pIdx
->
numOfBlocks
-
1
);
ASSERT
(
pCompBlock
->
last
);
if
(
tsdbLoadBlockData
(
pHelper
,
pCompBlock
,
NULL
)
<
0
)
return
-
1
;
ASSERT
(
pHelper
->
pDataCols
[
0
]
->
numOfRows
==
pCompBlock
->
numOfRows
&&
...
...
@@ -351,7 +351,7 @@ int tsdbMoveLastBlockIfNeccessary(SRWHelper *pHelper) {
}
int
tsdbWriteCompInfo
(
SRWHelper
*
pHelper
)
{
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
off_t
offset
=
0
;
SFile
*
pFile
=
helperNewHeadF
(
pHelper
);
...
...
@@ -362,8 +362,8 @@ int tsdbWriteCompInfo(SRWHelper *pHelper) {
pHelper
->
pCompInfo
->
delimiter
=
TSDB_FILE_DELIMITER
;
pHelper
->
pCompInfo
->
uid
=
pHelper
->
tableInfo
.
uid
;
pHelper
->
pCompInfo
->
tid
=
pHelper
->
tableInfo
.
tid
;
ASSERT
(
pIdx
->
len
>
sizeof
(
S
Comp
Info
)
+
sizeof
(
TSCKSUM
)
&&
(
pIdx
->
len
-
sizeof
(
S
CompInfo
)
-
sizeof
(
TSCKSUM
))
%
sizeof
(
SComp
Block
)
==
0
);
ASSERT
(
pIdx
->
len
>
sizeof
(
S
Block
Info
)
+
sizeof
(
TSCKSUM
)
&&
(
pIdx
->
len
-
sizeof
(
S
BlockInfo
)
-
sizeof
(
TSCKSUM
))
%
sizeof
(
S
Block
)
==
0
);
taosCalcChecksumAppend
(
0
,
(
uint8_t
*
)
pHelper
->
pCompInfo
,
pIdx
->
len
);
}
...
...
@@ -387,7 +387,7 @@ int tsdbWriteCompInfo(SRWHelper *pHelper) {
return
-
1
;
}
if
(
taosTSizeof
(
pHelper
->
pWIdx
)
<
pFile
->
info
.
len
+
sizeof
(
S
Comp
Idx
)
+
12
)
{
if
(
taosTSizeof
(
pHelper
->
pWIdx
)
<
pFile
->
info
.
len
+
sizeof
(
S
Block
Idx
)
+
12
)
{
pHelper
->
pWIdx
=
taosTRealloc
(
pHelper
->
pWIdx
,
taosTSizeof
(
pHelper
->
pWIdx
)
==
0
?
1024
:
taosTSizeof
(
pHelper
->
pWIdx
)
*
2
);
if
(
pHelper
->
pWIdx
==
NULL
)
{
terrno
=
TSDB_CODE_TDB_OUT_OF_MEMORY
;
...
...
@@ -447,7 +447,7 @@ int tsdbWriteCompIdx(SRWHelper *pHelper) {
}
int
tsdbLoadCompIdxImpl
(
SFile
*
pFile
,
uint32_t
offset
,
uint32_t
len
,
void
*
buffer
)
{
const
char
*
prefixMsg
=
"failed to load S
Comp
Idx part"
;
const
char
*
prefixMsg
=
"failed to load S
Block
Idx part"
;
if
(
lseek
(
pFile
->
fd
,
offset
,
SEEK_SET
)
<
0
)
{
tsdbError
(
"%s: seek to file %s offset %u failed since %s"
,
prefixMsg
,
pFile
->
fname
,
offset
,
strerror
(
errno
));
terrno
=
TAOS_SYSTEM_ERROR
(
errno
);
...
...
@@ -470,14 +470,14 @@ int tsdbLoadCompIdxImpl(SFile *pFile, uint32_t offset, uint32_t len, void *buffe
return
0
;
}
int
tsdbDecodeSCompIdxImpl
(
void
*
buffer
,
uint32_t
len
,
S
Comp
Idx
**
ppCompIdx
,
int
*
numOfIdx
)
{
int
tsdbDecodeSCompIdxImpl
(
void
*
buffer
,
uint32_t
len
,
S
Block
Idx
**
ppCompIdx
,
int
*
numOfIdx
)
{
int
nIdx
=
0
;
void
*
pPtr
=
buffer
;
while
(
POINTER_DISTANCE
(
pPtr
,
buffer
)
<
(
int
)(
len
-
sizeof
(
TSCKSUM
)))
{
size_t
tlen
=
taosTSizeof
(
*
ppCompIdx
);
if
(
tlen
<
sizeof
(
S
Comp
Idx
)
*
(
nIdx
+
1
))
{
*
ppCompIdx
=
(
S
Comp
Idx
*
)
taosTRealloc
(
*
ppCompIdx
,
(
tlen
==
0
)
?
1024
:
tlen
*
2
);
if
(
tlen
<
sizeof
(
S
Block
Idx
)
*
(
nIdx
+
1
))
{
*
ppCompIdx
=
(
S
Block
Idx
*
)
taosTRealloc
(
*
ppCompIdx
,
(
tlen
==
0
)
?
1024
:
tlen
*
2
);
if
(
*
ppCompIdx
==
NULL
)
{
terrno
=
TSDB_CODE_TDB_OUT_OF_MEMORY
;
return
-
1
;
...
...
@@ -486,7 +486,7 @@ int tsdbDecodeSCompIdxImpl(void *buffer, uint32_t len, SCompIdx **ppCompIdx, int
pPtr
=
tsdbDecodeSCompIdx
(
pPtr
,
&
((
*
ppCompIdx
)[
nIdx
]));
if
(
pPtr
==
NULL
)
{
tsdbError
(
"failed to decode S
Comp
Idx part, idx:%d"
,
nIdx
);
tsdbError
(
"failed to decode S
Block
Idx part, idx:%d"
,
nIdx
);
terrno
=
TSDB_CODE_TDB_FILE_CORRUPTED
;
return
-
1
;
}
...
...
@@ -513,15 +513,15 @@ int tsdbLoadCompIdx(SRWHelper *pHelper, void *target) {
return
-
1
;
}
// Load S
Comp
Idx binary from file
// Load S
Block
Idx binary from file
if
(
tsdbLoadCompIdxImpl
(
pFile
,
pFile
->
info
.
offset
,
pFile
->
info
.
len
,
(
void
*
)(
pHelper
->
pBuffer
))
<
0
)
{
return
-
1
;
}
// Decode the S
Comp
Idx part
// Decode the S
Block
Idx part
if
(
tsdbDecodeSCompIdxImpl
(
pHelper
->
pBuffer
,
pFile
->
info
.
len
,
&
(
pHelper
->
idxH
.
pIdxArray
),
&
(
pHelper
->
idxH
.
numOfIdx
))
<
0
)
{
tsdbError
(
"vgId:%d failed to decode S
Comp
Idx part from file %s since %s"
,
REPO_ID
(
pHelper
->
pRepo
),
pFile
->
fname
,
tsdbError
(
"vgId:%d failed to decode S
Block
Idx part from file %s since %s"
,
REPO_ID
(
pHelper
->
pRepo
),
pFile
->
fname
,
tstrerror
(
errno
));
return
-
1
;
}
...
...
@@ -531,13 +531,13 @@ int tsdbLoadCompIdx(SRWHelper *pHelper, void *target) {
// Copy the memory for outside usage
if
(
target
&&
pHelper
->
idxH
.
numOfIdx
>
0
)
memcpy
(
target
,
pHelper
->
idxH
.
pIdxArray
,
sizeof
(
S
Comp
Idx
)
*
pHelper
->
idxH
.
numOfIdx
);
memcpy
(
target
,
pHelper
->
idxH
.
pIdxArray
,
sizeof
(
S
Block
Idx
)
*
pHelper
->
idxH
.
numOfIdx
);
return
0
;
}
int
tsdbLoadCompInfoImpl
(
SFile
*
pFile
,
S
CompIdx
*
pIdx
,
SComp
Info
**
ppCompInfo
)
{
const
char
*
prefixMsg
=
"failed to load S
CompInfo/SComp
Block part"
;
int
tsdbLoadCompInfoImpl
(
SFile
*
pFile
,
S
BlockIdx
*
pIdx
,
SBlock
Info
**
ppCompInfo
)
{
const
char
*
prefixMsg
=
"failed to load S
BlockInfo/S
Block part"
;
if
(
lseek
(
pFile
->
fd
,
pIdx
->
offset
,
SEEK_SET
)
<
0
)
{
tsdbError
(
"%s: seek to file %s offset %u failed since %s"
,
prefixMsg
,
pFile
->
fname
,
pIdx
->
offset
,
strerror
(
errno
));
...
...
@@ -570,7 +570,7 @@ int tsdbLoadCompInfoImpl(SFile *pFile, SCompIdx *pIdx, SCompInfo **ppCompInfo) {
int
tsdbLoadCompInfo
(
SRWHelper
*
pHelper
,
void
*
target
)
{
ASSERT
(
helperHasState
(
pHelper
,
TSDB_HELPER_TABLE_SET
));
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
SFile
*
pFile
=
helperHeadF
(
pHelper
);
...
...
@@ -591,7 +591,7 @@ int tsdbLoadCompInfo(SRWHelper *pHelper, void *target) {
return
0
;
}
int
tsdbLoadCompData
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
void
*
target
)
{
int
tsdbLoadCompData
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
void
*
target
)
{
ASSERT
(
pCompBlock
->
numOfSubBlocks
<=
1
);
SFile
*
pFile
=
(
pCompBlock
->
last
)
?
helperLastF
(
pHelper
)
:
helperDataF
(
pHelper
);
...
...
@@ -630,7 +630,7 @@ int tsdbLoadCompData(SRWHelper *pHelper, SCompBlock *pCompBlock, void *target) {
}
void
tsdbGetDataStatis
(
SRWHelper
*
pHelper
,
SDataStatis
*
pStatis
,
int
numOfCols
)
{
S
Comp
Data
*
pCompData
=
pHelper
->
pCompData
;
S
Block
Data
*
pCompData
=
pHelper
->
pCompData
;
for
(
int
i
=
0
,
j
=
0
;
i
<
numOfCols
;)
{
if
(
j
>=
pCompData
->
numOfCols
)
{
...
...
@@ -657,13 +657,13 @@ void tsdbGetDataStatis(SRWHelper *pHelper, SDataStatis *pStatis, int numOfCols)
}
}
int
tsdbLoadBlockDataCols
(
SRWHelper
*
pHelper
,
S
CompBlock
*
pCompBlock
,
SComp
Info
*
pCompInfo
,
int16_t
*
colIds
,
int
numOfColIds
)
{
int
tsdbLoadBlockDataCols
(
SRWHelper
*
pHelper
,
S
Block
*
pCompBlock
,
SBlock
Info
*
pCompInfo
,
int16_t
*
colIds
,
int
numOfColIds
)
{
ASSERT
(
pCompBlock
->
numOfSubBlocks
>=
1
);
// Must be super block
S
Comp
Block
*
pTCompBlock
=
pCompBlock
;
SBlock
*
pTCompBlock
=
pCompBlock
;
int
numOfSubBlocks
=
pCompBlock
->
numOfSubBlocks
;
if
(
numOfSubBlocks
>
1
)
pTCompBlock
=
(
S
Comp
Block
*
)
POINTER_SHIFT
((
pCompInfo
==
NULL
)
?
pHelper
->
pCompInfo
:
pCompInfo
,
pCompBlock
->
offset
);
pTCompBlock
=
(
SBlock
*
)
POINTER_SHIFT
((
pCompInfo
==
NULL
)
?
pHelper
->
pCompInfo
:
pCompInfo
,
pCompBlock
->
offset
);
tdResetDataCols
(
pHelper
->
pDataCols
[
0
]);
if
(
tsdbLoadBlockDataColsImpl
(
pHelper
,
pTCompBlock
,
pHelper
->
pDataCols
[
0
],
colIds
,
numOfColIds
)
<
0
)
goto
_err
;
...
...
@@ -684,12 +684,12 @@ _err:
return
-
1
;
}
int
tsdbLoadBlockData
(
SRWHelper
*
pHelper
,
S
CompBlock
*
pCompBlock
,
SComp
Info
*
pCompInfo
)
{
S
Comp
Block
*
pTCompBlock
=
pCompBlock
;
int
tsdbLoadBlockData
(
SRWHelper
*
pHelper
,
S
Block
*
pCompBlock
,
SBlock
Info
*
pCompInfo
)
{
SBlock
*
pTCompBlock
=
pCompBlock
;
int
numOfSubBlock
=
pCompBlock
->
numOfSubBlocks
;
if
(
numOfSubBlock
>
1
)
pTCompBlock
=
(
S
Comp
Block
*
)
POINTER_SHIFT
((
pCompInfo
==
NULL
)
?
pHelper
->
pCompInfo
:
pCompInfo
,
pCompBlock
->
offset
);
pTCompBlock
=
(
SBlock
*
)
POINTER_SHIFT
((
pCompInfo
==
NULL
)
?
pHelper
->
pCompInfo
:
pCompInfo
,
pCompBlock
->
offset
);
tdResetDataCols
(
pHelper
->
pDataCols
[
0
]);
if
(
tsdbLoadBlockDataImpl
(
pHelper
,
pTCompBlock
,
pHelper
->
pDataCols
[
0
])
<
0
)
goto
_err
;
...
...
@@ -719,10 +719,10 @@ static bool tsdbShouldCreateNewLast(SRWHelper *pHelper) {
return
false
;
}
static
int
tsdbWriteBlockToFile
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
SDataCols
*
pDataCols
,
S
Comp
Block
*
pCompBlock
,
static
int
tsdbWriteBlockToFile
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
SDataCols
*
pDataCols
,
SBlock
*
pCompBlock
,
bool
isLast
,
bool
isSuperBlock
)
{
STsdbCfg
*
pCfg
=
&
(
pHelper
->
pRepo
->
config
);
S
CompData
*
pCompData
=
(
SComp
Data
*
)(
pHelper
->
pBuffer
);
S
BlockData
*
pCompData
=
(
SBlock
Data
*
)(
pHelper
->
pBuffer
);
int64_t
offset
=
0
;
int
rowsToWrite
=
pDataCols
->
numOfRows
;
...
...
@@ -740,7 +740,7 @@ static int tsdbWriteBlockToFile(SRWHelper *pHelper, SFile *pFile, SDataCols *pDa
int
nColsNotAllNull
=
0
;
for
(
int
ncol
=
1
;
ncol
<
pDataCols
->
numOfCols
;
ncol
++
)
{
// ncol from 1, we skip the timestamp column
SDataCol
*
pDataCol
=
pDataCols
->
cols
+
ncol
;
S
Comp
Col
*
pCompCol
=
pCompData
->
cols
+
nColsNotAllNull
;
S
Block
Col
*
pCompCol
=
pCompData
->
cols
+
nColsNotAllNull
;
if
(
isNEleNull
(
pDataCol
,
rowsToWrite
))
{
// all data to commit are NULL, just ignore it
continue
;
...
...
@@ -770,7 +770,7 @@ static int tsdbWriteBlockToFile(SRWHelper *pHelper, SFile *pFile, SDataCols *pDa
if
(
ncol
!=
0
&&
tcol
>=
nColsNotAllNull
)
break
;
SDataCol
*
pDataCol
=
pDataCols
->
cols
+
ncol
;
S
Comp
Col
*
pCompCol
=
pCompData
->
cols
+
tcol
;
S
Block
Col
*
pCompCol
=
pCompData
->
cols
+
tcol
;
if
(
ncol
!=
0
&&
(
pDataCol
->
colId
!=
pCompCol
->
colId
))
continue
;
void
*
tptr
=
POINTER_SHIFT
(
pCompData
,
lsize
);
...
...
@@ -859,7 +859,7 @@ _err:
static
int
compareKeyBlock
(
const
void
*
arg1
,
const
void
*
arg2
)
{
TSKEY
key
=
*
(
TSKEY
*
)
arg1
;
S
CompBlock
*
pBlock
=
(
SComp
Block
*
)
arg2
;
S
Block
*
pBlock
=
(
S
Block
*
)
arg2
;
if
(
key
<
pBlock
->
keyFirst
)
{
return
-
1
;
...
...
@@ -872,42 +872,42 @@ static int compareKeyBlock(const void *arg1, const void *arg2) {
static
int
tsdbAdjustInfoSizeIfNeeded
(
SRWHelper
*
pHelper
,
size_t
esize
)
{
if
(
taosTSizeof
((
void
*
)
pHelper
->
pCompInfo
)
<=
esize
)
{
size_t
tsize
=
esize
+
sizeof
(
S
Comp
Block
)
*
16
;
pHelper
->
pCompInfo
=
(
S
Comp
Info
*
)
taosTRealloc
(
pHelper
->
pCompInfo
,
tsize
);
size_t
tsize
=
esize
+
sizeof
(
SBlock
)
*
16
;
pHelper
->
pCompInfo
=
(
S
Block
Info
*
)
taosTRealloc
(
pHelper
->
pCompInfo
,
tsize
);
if
(
pHelper
->
pCompInfo
==
NULL
)
return
-
1
;
}
return
0
;
}
static
int
tsdbInsertSuperBlock
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
int
blkIdx
)
{
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
static
int
tsdbInsertSuperBlock
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
int
blkIdx
)
{
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
ASSERT
(
blkIdx
>=
0
&&
blkIdx
<=
(
int
)
pIdx
->
numOfBlocks
);
ASSERT
(
pCompBlock
->
numOfSubBlocks
==
1
);
// Adjust memory if no more room
if
(
pIdx
->
len
==
0
)
pIdx
->
len
=
sizeof
(
S
Comp
Info
)
+
sizeof
(
TSCKSUM
);
if
(
tsdbAdjustInfoSizeIfNeeded
(
pHelper
,
pIdx
->
len
+
sizeof
(
S
Comp
Info
))
<
0
)
goto
_err
;
if
(
pIdx
->
len
==
0
)
pIdx
->
len
=
sizeof
(
S
Block
Info
)
+
sizeof
(
TSCKSUM
);
if
(
tsdbAdjustInfoSizeIfNeeded
(
pHelper
,
pIdx
->
len
+
sizeof
(
S
Block
Info
))
<
0
)
goto
_err
;
// Change the offset
for
(
uint32_t
i
=
0
;
i
<
pIdx
->
numOfBlocks
;
i
++
)
{
S
Comp
Block
*
pTCompBlock
=
&
pHelper
->
pCompInfo
->
blocks
[
i
];
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
+=
sizeof
(
S
Comp
Block
);
SBlock
*
pTCompBlock
=
&
pHelper
->
pCompInfo
->
blocks
[
i
];
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
+=
sizeof
(
SBlock
);
}
// Memmove if needed
int
tsize
=
pIdx
->
len
-
(
sizeof
(
S
CompInfo
)
+
sizeof
(
SComp
Block
)
*
blkIdx
);
int
tsize
=
pIdx
->
len
-
(
sizeof
(
S
BlockInfo
)
+
sizeof
(
S
Block
)
*
blkIdx
);
if
(
tsize
>
0
)
{
ASSERT
(
sizeof
(
S
CompInfo
)
+
sizeof
(
SComp
Block
)
*
(
blkIdx
+
1
)
<
taosTSizeof
(
pHelper
->
pCompInfo
));
ASSERT
(
sizeof
(
S
CompInfo
)
+
sizeof
(
SComp
Block
)
*
(
blkIdx
+
1
)
+
tsize
<=
taosTSizeof
(
pHelper
->
pCompInfo
));
memmove
(
POINTER_SHIFT
(
pHelper
->
pCompInfo
,
sizeof
(
S
CompInfo
)
+
sizeof
(
SComp
Block
)
*
(
blkIdx
+
1
)),
POINTER_SHIFT
(
pHelper
->
pCompInfo
,
sizeof
(
S
CompInfo
)
+
sizeof
(
SComp
Block
)
*
blkIdx
),
tsize
);
ASSERT
(
sizeof
(
S
BlockInfo
)
+
sizeof
(
S
Block
)
*
(
blkIdx
+
1
)
<
taosTSizeof
(
pHelper
->
pCompInfo
));
ASSERT
(
sizeof
(
S
BlockInfo
)
+
sizeof
(
S
Block
)
*
(
blkIdx
+
1
)
+
tsize
<=
taosTSizeof
(
pHelper
->
pCompInfo
));
memmove
(
POINTER_SHIFT
(
pHelper
->
pCompInfo
,
sizeof
(
S
BlockInfo
)
+
sizeof
(
S
Block
)
*
(
blkIdx
+
1
)),
POINTER_SHIFT
(
pHelper
->
pCompInfo
,
sizeof
(
S
BlockInfo
)
+
sizeof
(
S
Block
)
*
blkIdx
),
tsize
);
}
pHelper
->
pCompInfo
->
blocks
[
blkIdx
]
=
*
pCompBlock
;
pIdx
->
numOfBlocks
++
;
pIdx
->
len
+=
sizeof
(
S
Comp
Block
);
pIdx
->
len
+=
sizeof
(
SBlock
);
ASSERT
(
pIdx
->
len
<=
taosTSizeof
(
pHelper
->
pCompInfo
));
pIdx
->
maxKey
=
blockAtIdx
(
pHelper
,
pIdx
->
numOfBlocks
-
1
)
->
keyLast
;
pIdx
->
hasLast
=
(
uint32_t
)
blockAtIdx
(
pHelper
,
pIdx
->
numOfBlocks
-
1
)
->
last
;
...
...
@@ -925,17 +925,17 @@ _err:
return
-
1
;
}
static
int
tsdbAddSubBlock
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
int
blkIdx
,
int
rowsAdded
)
{
static
int
tsdbAddSubBlock
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
int
blkIdx
,
int
rowsAdded
)
{
ASSERT
(
pCompBlock
->
numOfSubBlocks
==
0
);
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
ASSERT
(
blkIdx
>=
0
&&
blkIdx
<
(
int
)
pIdx
->
numOfBlocks
);
S
Comp
Block
*
pSCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
blkIdx
;
SBlock
*
pSCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
blkIdx
;
ASSERT
(
pSCompBlock
->
numOfSubBlocks
>=
1
&&
pSCompBlock
->
numOfSubBlocks
<
TSDB_MAX_SUBBLOCKS
);
size_t
spaceNeeded
=
(
pSCompBlock
->
numOfSubBlocks
==
1
)
?
pIdx
->
len
+
sizeof
(
S
CompBlock
)
*
2
:
pIdx
->
len
+
sizeof
(
SComp
Block
);
(
pSCompBlock
->
numOfSubBlocks
==
1
)
?
pIdx
->
len
+
sizeof
(
S
Block
)
*
2
:
pIdx
->
len
+
sizeof
(
S
Block
);
if
(
tsdbAdjustInfoSizeIfNeeded
(
pHelper
,
spaceNeeded
)
<
0
)
goto
_err
;
pSCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
blkIdx
;
...
...
@@ -944,28 +944,28 @@ static int tsdbAddSubBlock(SRWHelper *pHelper, SCompBlock *pCompBlock, int blkId
if
(
pSCompBlock
->
numOfSubBlocks
>
1
)
{
size_t
tsize
=
(
size_t
)(
pIdx
->
len
-
(
pSCompBlock
->
offset
+
pSCompBlock
->
len
));
if
(
tsize
>
0
)
{
memmove
((
void
*
)((
char
*
)(
pHelper
->
pCompInfo
)
+
pSCompBlock
->
offset
+
pSCompBlock
->
len
+
sizeof
(
S
Comp
Block
)),
memmove
((
void
*
)((
char
*
)(
pHelper
->
pCompInfo
)
+
pSCompBlock
->
offset
+
pSCompBlock
->
len
+
sizeof
(
SBlock
)),
(
void
*
)((
char
*
)(
pHelper
->
pCompInfo
)
+
pSCompBlock
->
offset
+
pSCompBlock
->
len
),
tsize
);
for
(
uint32_t
i
=
blkIdx
+
1
;
i
<
pIdx
->
numOfBlocks
;
i
++
)
{
S
Comp
Block
*
pTCompBlock
=
&
pHelper
->
pCompInfo
->
blocks
[
i
];
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
+=
sizeof
(
S
Comp
Block
);
SBlock
*
pTCompBlock
=
&
pHelper
->
pCompInfo
->
blocks
[
i
];
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
+=
sizeof
(
SBlock
);
}
}
*
(
S
Comp
Block
*
)((
char
*
)(
pHelper
->
pCompInfo
)
+
pSCompBlock
->
offset
+
pSCompBlock
->
len
)
=
*
pCompBlock
;
*
(
SBlock
*
)((
char
*
)(
pHelper
->
pCompInfo
)
+
pSCompBlock
->
offset
+
pSCompBlock
->
len
)
=
*
pCompBlock
;
pSCompBlock
->
numOfSubBlocks
++
;
ASSERT
(
pSCompBlock
->
numOfSubBlocks
<=
TSDB_MAX_SUBBLOCKS
);
pSCompBlock
->
len
+=
sizeof
(
S
Comp
Block
);
pSCompBlock
->
len
+=
sizeof
(
SBlock
);
pSCompBlock
->
numOfRows
+=
rowsAdded
;
pSCompBlock
->
keyFirst
=
MIN
(
pSCompBlock
->
keyFirst
,
pCompBlock
->
keyFirst
);
pSCompBlock
->
keyLast
=
MAX
(
pSCompBlock
->
keyLast
,
pCompBlock
->
keyLast
);
pIdx
->
len
+=
sizeof
(
S
Comp
Block
);
pIdx
->
len
+=
sizeof
(
SBlock
);
}
else
{
// Need to create two sub-blocks
void
*
ptr
=
NULL
;
for
(
uint32_t
i
=
blkIdx
+
1
;
i
<
pIdx
->
numOfBlocks
;
i
++
)
{
S
Comp
Block
*
pTCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
i
;
SBlock
*
pTCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
i
;
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
{
ptr
=
POINTER_SHIFT
(
pHelper
->
pCompInfo
,
pTCompBlock
->
offset
);
break
;
...
...
@@ -976,26 +976,26 @@ static int tsdbAddSubBlock(SRWHelper *pHelper, SCompBlock *pCompBlock, int blkId
size_t
tsize
=
pIdx
->
len
-
((
char
*
)
ptr
-
(
char
*
)(
pHelper
->
pCompInfo
));
if
(
tsize
>
0
)
{
memmove
(
POINTER_SHIFT
(
ptr
,
sizeof
(
S
Comp
Block
)
*
2
),
ptr
,
tsize
);
memmove
(
POINTER_SHIFT
(
ptr
,
sizeof
(
SBlock
)
*
2
),
ptr
,
tsize
);
for
(
uint32_t
i
=
blkIdx
+
1
;
i
<
pIdx
->
numOfBlocks
;
i
++
)
{
S
Comp
Block
*
pTCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
i
;
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
+=
(
sizeof
(
S
Comp
Block
)
*
2
);
SBlock
*
pTCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
i
;
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
+=
(
sizeof
(
SBlock
)
*
2
);
}
}
((
S
Comp
Block
*
)
ptr
)[
0
]
=
*
pSCompBlock
;
((
S
Comp
Block
*
)
ptr
)[
0
].
numOfSubBlocks
=
0
;
((
SBlock
*
)
ptr
)[
0
]
=
*
pSCompBlock
;
((
SBlock
*
)
ptr
)[
0
].
numOfSubBlocks
=
0
;
((
S
Comp
Block
*
)
ptr
)[
1
]
=
*
pCompBlock
;
((
SBlock
*
)
ptr
)[
1
]
=
*
pCompBlock
;
pSCompBlock
->
numOfSubBlocks
=
2
;
pSCompBlock
->
numOfRows
+=
rowsAdded
;
pSCompBlock
->
offset
=
((
char
*
)
ptr
)
-
((
char
*
)
pHelper
->
pCompInfo
);
pSCompBlock
->
len
=
sizeof
(
S
Comp
Block
)
*
2
;
pSCompBlock
->
keyFirst
=
MIN
(((
S
CompBlock
*
)
ptr
)[
0
].
keyFirst
,
((
SComp
Block
*
)
ptr
)[
1
].
keyFirst
);
pSCompBlock
->
keyLast
=
MAX
(((
S
CompBlock
*
)
ptr
)[
0
].
keyLast
,
((
SComp
Block
*
)
ptr
)[
1
].
keyLast
);
pSCompBlock
->
len
=
sizeof
(
SBlock
)
*
2
;
pSCompBlock
->
keyFirst
=
MIN
(((
S
Block
*
)
ptr
)[
0
].
keyFirst
,
((
S
Block
*
)
ptr
)[
1
].
keyFirst
);
pSCompBlock
->
keyLast
=
MAX
(((
S
Block
*
)
ptr
)[
0
].
keyLast
,
((
S
Block
*
)
ptr
)[
1
].
keyLast
);
pIdx
->
len
+=
(
sizeof
(
S
Comp
Block
)
*
2
);
pIdx
->
len
+=
(
sizeof
(
SBlock
)
*
2
);
}
pIdx
->
maxKey
=
pHelper
->
pCompInfo
->
blocks
[
pIdx
->
numOfBlocks
-
1
].
keyLast
;
...
...
@@ -1009,14 +1009,14 @@ _err:
return
-
1
;
}
static
int
tsdbUpdateSuperBlock
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
int
blkIdx
)
{
static
int
tsdbUpdateSuperBlock
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
int
blkIdx
)
{
ASSERT
(
pCompBlock
->
numOfSubBlocks
==
1
);
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
ASSERT
(
blkIdx
>=
0
&&
blkIdx
<
(
int
)
pIdx
->
numOfBlocks
);
S
Comp
Block
*
pSCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
blkIdx
;
SBlock
*
pSCompBlock
=
pHelper
->
pCompInfo
->
blocks
+
blkIdx
;
ASSERT
(
pSCompBlock
->
numOfSubBlocks
>=
1
);
...
...
@@ -1029,11 +1029,11 @@ static int tsdbUpdateSuperBlock(SRWHelper *pHelper, SCompBlock *pCompBlock, int
}
for
(
uint32_t
i
=
blkIdx
+
1
;
i
<
pIdx
->
numOfBlocks
;
i
++
)
{
S
Comp
Block
*
pTCompBlock
=
&
pHelper
->
pCompInfo
->
blocks
[
i
];
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
-=
(
sizeof
(
S
Comp
Block
)
*
pSCompBlock
->
numOfSubBlocks
);
SBlock
*
pTCompBlock
=
&
pHelper
->
pCompInfo
->
blocks
[
i
];
if
(
pTCompBlock
->
numOfSubBlocks
>
1
)
pTCompBlock
->
offset
-=
(
sizeof
(
SBlock
)
*
pSCompBlock
->
numOfSubBlocks
);
}
pIdx
->
len
-=
(
sizeof
(
S
Comp
Block
)
*
pSCompBlock
->
numOfSubBlocks
);
pIdx
->
len
-=
(
sizeof
(
SBlock
)
*
pSCompBlock
->
numOfSubBlocks
);
}
*
pSCompBlock
=
*
pCompBlock
;
...
...
@@ -1139,7 +1139,7 @@ static int tsdbInitHelper(SRWHelper *pHelper, STsdbRepo *pRepo, tsdb_rw_helper_t
// TODO: pMeta->maxRowBytes and pMeta->maxCols may change here causing invalid write
pHelper
->
pBuffer
=
taosTMalloc
(
sizeof
(
S
CompData
)
+
(
sizeof
(
SComp
Col
)
+
sizeof
(
TSCKSUM
)
+
COMP_OVERFLOW_BYTES
)
*
pMeta
->
maxCols
+
taosTMalloc
(
sizeof
(
S
BlockData
)
+
(
sizeof
(
SBlock
Col
)
+
sizeof
(
TSCKSUM
)
+
COMP_OVERFLOW_BYTES
)
*
pMeta
->
maxCols
+
pMeta
->
maxRowBytes
*
pCfg
->
maxRowsPerFileBlock
+
sizeof
(
TSCKSUM
));
if
(
pHelper
->
pBuffer
==
NULL
)
{
terrno
=
TSDB_CODE_TDB_OUT_OF_MEMORY
;
...
...
@@ -1187,7 +1187,7 @@ static int tsdbCheckAndDecodeColumnData(SDataCol *pDataCol, char *content, int32
return
0
;
}
static
int
tsdbLoadColData
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
S
CompBlock
*
pCompBlock
,
SComp
Col
*
pCompCol
,
static
int
tsdbLoadColData
(
SRWHelper
*
pHelper
,
SFile
*
pFile
,
S
Block
*
pCompBlock
,
SBlock
Col
*
pCompCol
,
SDataCol
*
pDataCol
)
{
ASSERT
(
pDataCol
->
colId
==
pCompCol
->
colId
);
int
tsize
=
pDataCol
->
bytes
*
pCompBlock
->
numOfRows
+
COMP_OVERFLOW_BYTES
;
...
...
@@ -1228,14 +1228,14 @@ static int tsdbLoadColData(SRWHelper *pHelper, SFile *pFile, SCompBlock *pCompBl
return
0
;
}
static
int
tsdbLoadBlockDataColsImpl
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
SDataCols
*
pDataCols
,
int16_t
*
colIds
,
int
numOfColIds
)
{
static
int
tsdbLoadBlockDataColsImpl
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
SDataCols
*
pDataCols
,
int16_t
*
colIds
,
int
numOfColIds
)
{
ASSERT
(
pCompBlock
->
numOfSubBlocks
<=
1
);
ASSERT
(
colIds
[
0
]
==
0
);
SFile
*
pFile
=
(
pCompBlock
->
last
)
?
helperLastF
(
pHelper
)
:
helperDataF
(
pHelper
);
S
Comp
Col
compCol
=
{
0
};
S
Block
Col
compCol
=
{
0
};
// If only load timestamp column, no need to load S
Comp
Data part
// If only load timestamp column, no need to load S
Block
Data part
if
(
numOfColIds
>
1
&&
tsdbLoadCompData
(
pHelper
,
pCompBlock
,
NULL
)
<
0
)
goto
_err
;
pDataCols
->
numOfRows
=
pCompBlock
->
numOfRows
;
...
...
@@ -1245,7 +1245,7 @@ static int tsdbLoadBlockDataColsImpl(SRWHelper *pHelper, SCompBlock *pCompBlock,
for
(
int
i
=
0
;
i
<
numOfColIds
;
i
++
)
{
int16_t
colId
=
colIds
[
i
];
SDataCol
*
pDataCol
=
NULL
;
S
Comp
Col
*
pCompCol
=
NULL
;
S
Block
Col
*
pCompCol
=
NULL
;
while
(
true
)
{
if
(
dcol
>=
pDataCols
->
numOfCols
)
{
...
...
@@ -1305,7 +1305,7 @@ _err:
return
-
1
;
}
static
int
tsdbLoadBlockDataImpl
(
SRWHelper
*
pHelper
,
S
Comp
Block
*
pCompBlock
,
SDataCols
*
pDataCols
)
{
static
int
tsdbLoadBlockDataImpl
(
SRWHelper
*
pHelper
,
SBlock
*
pCompBlock
,
SDataCols
*
pDataCols
)
{
ASSERT
(
pCompBlock
->
numOfSubBlocks
<=
1
);
SFile
*
pFile
=
(
pCompBlock
->
last
)
?
helperLastF
(
pHelper
)
:
helperDataF
(
pHelper
);
...
...
@@ -1316,7 +1316,7 @@ static int tsdbLoadBlockDataImpl(SRWHelper *pHelper, SCompBlock *pCompBlock, SDa
goto
_err
;
}
S
CompData
*
pCompData
=
(
SComp
Data
*
)
pHelper
->
pBuffer
;
S
BlockData
*
pCompData
=
(
SBlock
Data
*
)
pHelper
->
pBuffer
;
int
fd
=
pFile
->
fd
;
if
(
lseek
(
fd
,
(
off_t
)
pCompBlock
->
offset
,
SEEK_SET
)
<
0
)
{
...
...
@@ -1344,7 +1344,7 @@ static int tsdbLoadBlockDataImpl(SRWHelper *pHelper, SCompBlock *pCompBlock, SDa
pDataCols
->
numOfRows
=
pCompBlock
->
numOfRows
;
// Recover the data
int
ccol
=
0
;
// loop iter for S
Comp
Col object
int
ccol
=
0
;
// loop iter for S
Block
Col object
int
dcol
=
0
;
// loop iter for SDataCols object
while
(
dcol
<
pDataCols
->
numOfCols
)
{
SDataCol
*
pDataCol
=
&
(
pDataCols
->
cols
[
dcol
]);
...
...
@@ -1360,7 +1360,7 @@ static int tsdbLoadBlockDataImpl(SRWHelper *pHelper, SCompBlock *pCompBlock, SDa
int32_t
tlen
=
pCompBlock
->
keyLen
;
if
(
dcol
!=
0
)
{
S
Comp
Col
*
pCompCol
=
&
(
pCompData
->
cols
[
ccol
]);
S
Block
Col
*
pCompCol
=
&
(
pCompData
->
cols
[
ccol
]);
tcolId
=
pCompCol
->
colId
;
toffset
=
pCompCol
->
offset
;
tlen
=
pCompCol
->
len
;
...
...
@@ -1404,7 +1404,7 @@ _err:
return
-
1
;
}
static
int
tsdbEncodeSCompIdx
(
void
**
buf
,
S
Comp
Idx
*
pIdx
)
{
static
int
tsdbEncodeSCompIdx
(
void
**
buf
,
S
Block
Idx
*
pIdx
)
{
int
tlen
=
0
;
tlen
+=
taosEncodeVariantI32
(
buf
,
pIdx
->
tid
);
...
...
@@ -1418,7 +1418,7 @@ static int tsdbEncodeSCompIdx(void **buf, SCompIdx *pIdx) {
return
tlen
;
}
static
void
*
tsdbDecodeSCompIdx
(
void
*
buf
,
S
Comp
Idx
*
pIdx
)
{
static
void
*
tsdbDecodeSCompIdx
(
void
*
buf
,
S
Block
Idx
*
pIdx
)
{
uint8_t
hasLast
=
0
;
uint32_t
numOfBlocks
=
0
;
uint64_t
value
=
0
;
...
...
@@ -1441,15 +1441,15 @@ static void *tsdbDecodeSCompIdx(void *buf, SCompIdx *pIdx) {
static
int
tsdbProcessAppendCommit
(
SRWHelper
*
pHelper
,
SCommitIter
*
pCommitIter
,
SDataCols
*
pDataCols
,
TSKEY
maxKey
)
{
STsdbCfg
*
pCfg
=
&
(
pHelper
->
pRepo
->
config
);
STable
*
pTable
=
pCommitIter
->
pTable
;
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
TSKEY
keyFirst
=
tsdbNextIterKey
(
pCommitIter
->
pIter
);
int
defaultRowsInBlock
=
pCfg
->
maxRowsPerFileBlock
*
4
/
5
;
S
Comp
Block
compBlock
=
{
0
};
SBlock
compBlock
=
{
0
};
ASSERT
(
pIdx
->
len
<=
0
||
keyFirst
>
pIdx
->
maxKey
);
if
(
pIdx
->
hasLast
)
{
// append to with last block
ASSERT
(
pIdx
->
len
>
0
);
S
Comp
Block
*
pCompBlock
=
blockAtIdx
(
pHelper
,
pIdx
->
numOfBlocks
-
1
);
SBlock
*
pCompBlock
=
blockAtIdx
(
pHelper
,
pIdx
->
numOfBlocks
-
1
);
ASSERT
(
pCompBlock
->
last
&&
pCompBlock
->
numOfRows
<
pCfg
->
minRowsPerFileBlock
);
tdResetDataCols
(
pDataCols
);
int
rowsRead
=
tsdbLoadDataFromCache
(
pTable
,
pCommitIter
->
pIter
,
maxKey
,
defaultRowsInBlock
-
pCompBlock
->
numOfRows
,
...
...
@@ -1493,8 +1493,8 @@ static int tsdbProcessMergeCommit(SRWHelper *pHelper, SCommitIter *pCommitIter,
int
*
blkIdx
)
{
STsdbCfg
*
pCfg
=
&
(
pHelper
->
pRepo
->
config
);
STable
*
pTable
=
pCommitIter
->
pTable
;
S
Comp
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
S
Comp
Block
compBlock
=
{
0
};
S
Block
Idx
*
pIdx
=
&
(
pHelper
->
curCompIdx
);
SBlock
compBlock
=
{
0
};
TSKEY
keyFirst
=
tsdbNextIterKey
(
pCommitIter
->
pIter
);
int
defaultRowsInBlock
=
pCfg
->
maxRowsPerFileBlock
*
4
/
5
;
SDataCols
*
pDataCols0
=
pHelper
->
pDataCols
[
0
];
...
...
@@ -1503,8 +1503,8 @@ static int tsdbProcessMergeCommit(SRWHelper *pHelper, SCommitIter *pCommitIter,
ASSERT
(
keyFirst
<=
pIdx
->
maxKey
);
S
Comp
Block
*
pCompBlock
=
taosbsearch
((
void
*
)(
&
keyFirst
),
(
void
*
)
blockAtIdx
(
pHelper
,
*
blkIdx
),
pIdx
->
numOfBlocks
-
*
blkIdx
,
sizeof
(
S
Comp
Block
),
compareKeyBlock
,
TD_GE
);
SBlock
*
pCompBlock
=
taosbsearch
((
void
*
)(
&
keyFirst
),
(
void
*
)
blockAtIdx
(
pHelper
,
*
blkIdx
),
pIdx
->
numOfBlocks
-
*
blkIdx
,
sizeof
(
SBlock
),
compareKeyBlock
,
TD_GE
);
ASSERT
(
pCompBlock
!=
NULL
);
int
tblkIdx
=
(
int32_t
)(
TSDB_GET_COMPBLOCK_IDX
(
pHelper
,
pCompBlock
));
...
...
@@ -1675,7 +1675,7 @@ static int tsdbLoadAndMergeFromCache(SDataCols *pDataCols, int *iter, SCommitIte
return
numOfRows
;
}
static
int
tsdbWriteBlockToProperFile
(
SRWHelper
*
pHelper
,
SDataCols
*
pDataCols
,
S
Comp
Block
*
pCompBlock
)
{
static
int
tsdbWriteBlockToProperFile
(
SRWHelper
*
pHelper
,
SDataCols
*
pDataCols
,
SBlock
*
pCompBlock
)
{
STsdbCfg
*
pCfg
=
&
(
pHelper
->
pRepo
->
config
);
SFile
*
pFile
=
NULL
;
bool
isLast
=
false
;
...
...
src/tsdb/src/tsdbRead.c
浏览文件 @
f2eb4100
...
...
@@ -69,7 +69,7 @@ typedef struct STableCheckInfo {
STableId
tableId
;
TSKEY
lastKey
;
STable
*
pTableObj
;
S
Comp
Info
*
pCompInfo
;
S
Block
Info
*
pCompInfo
;
int32_t
compSize
;
int32_t
numOfBlocks
;
// number of qualified data blocks not the original blocks
SDataCols
*
pDataCols
;
...
...
@@ -80,7 +80,7 @@ typedef struct STableCheckInfo {
}
STableCheckInfo
;
typedef
struct
STableBlockInfo
{
S
Comp
Block
*
compBlock
;
SBlock
*
compBlock
;
STableCheckInfo
*
pTableCheckInfo
;
}
STableBlockInfo
;
...
...
@@ -136,7 +136,7 @@ typedef struct STableGroupSupporter {
static
STimeWindow
changeTableGroupByLastrow
(
STableGroupInfo
*
groupList
);
static
void
changeQueryHandleForInterpQuery
(
TsdbQueryHandleT
pHandle
);
static
void
doMergeTwoLevelData
(
STsdbQueryHandle
*
pQueryHandle
,
STableCheckInfo
*
pCheckInfo
,
S
Comp
Block
*
pBlock
);
static
void
doMergeTwoLevelData
(
STsdbQueryHandle
*
pQueryHandle
,
STableCheckInfo
*
pCheckInfo
,
SBlock
*
pBlock
);
static
int32_t
binarySearchForKey
(
char
*
pValue
,
int
num
,
TSKEY
key
,
int
order
);
static
int
tsdbReadRowsFromCache
(
STableCheckInfo
*
pCheckInfo
,
TSKEY
maxKey
,
int
maxRowsToRead
,
STimeWindow
*
win
,
STsdbQueryHandle
*
pQueryHandle
);
...
...
@@ -585,7 +585,7 @@ static int32_t getFileIdFromKey(TSKEY key, int32_t daysPerFile, int32_t precisio
return
(
int32_t
)
fid
;
}
static
int32_t
binarySearchForBlock
(
S
Comp
Block
*
pBlock
,
int32_t
numOfBlocks
,
TSKEY
skey
,
int32_t
order
)
{
static
int32_t
binarySearchForBlock
(
SBlock
*
pBlock
,
int32_t
numOfBlocks
,
TSKEY
skey
,
int32_t
order
)
{
int32_t
firstSlot
=
0
;
int32_t
lastSlot
=
numOfBlocks
-
1
;
...
...
@@ -628,7 +628,7 @@ static int32_t getFileCompInfo(STsdbQueryHandle* pQueryHandle, int32_t* numOfBlo
break
;
}
S
Comp
Idx
*
compIndex
=
&
pQueryHandle
->
rhelper
.
curCompIdx
;
S
Block
Idx
*
compIndex
=
&
pQueryHandle
->
rhelper
.
curCompIdx
;
// no data block in this file, try next file
if
(
compIndex
->
len
==
0
||
compIndex
->
numOfBlocks
==
0
||
compIndex
->
uid
!=
pCheckInfo
->
tableId
.
uid
)
{
...
...
@@ -645,12 +645,12 @@ static int32_t getFileCompInfo(STsdbQueryHandle* pQueryHandle, int32_t* numOfBlo
break
;
}
pCheckInfo
->
pCompInfo
=
(
S
Comp
Info
*
)
t
;
pCheckInfo
->
pCompInfo
=
(
S
Block
Info
*
)
t
;
pCheckInfo
->
compSize
=
compIndex
->
len
;
}
tsdbLoadCompInfo
(
&
(
pQueryHandle
->
rhelper
),
(
void
*
)(
pCheckInfo
->
pCompInfo
));
S
Comp
Info
*
pCompInfo
=
pCheckInfo
->
pCompInfo
;
S
Block
Info
*
pCompInfo
=
pCheckInfo
->
pCompInfo
;
TSKEY
s
=
TSKEY_INITIAL_VAL
,
e
=
TSKEY_INITIAL_VAL
;
...
...
@@ -679,7 +679,7 @@ static int32_t getFileCompInfo(STsdbQueryHandle* pQueryHandle, int32_t* numOfBlo
pCheckInfo
->
numOfBlocks
=
(
end
-
start
);
if
(
start
>
0
)
{
memmove
(
pCompInfo
->
blocks
,
&
pCompInfo
->
blocks
[
start
],
pCheckInfo
->
numOfBlocks
*
sizeof
(
S
Comp
Block
));
memmove
(
pCompInfo
->
blocks
,
&
pCompInfo
->
blocks
[
start
],
pCheckInfo
->
numOfBlocks
*
sizeof
(
SBlock
));
}
(
*
numOfBlocks
)
+=
pCheckInfo
->
numOfBlocks
;
...
...
@@ -688,7 +688,7 @@ static int32_t getFileCompInfo(STsdbQueryHandle* pQueryHandle, int32_t* numOfBlo
return
code
;
}
static
int32_t
doLoadFileDataBlock
(
STsdbQueryHandle
*
pQueryHandle
,
S
Comp
Block
*
pBlock
,
STableCheckInfo
*
pCheckInfo
,
int32_t
slotIndex
)
{
static
int32_t
doLoadFileDataBlock
(
STsdbQueryHandle
*
pQueryHandle
,
SBlock
*
pBlock
,
STableCheckInfo
*
pCheckInfo
,
int32_t
slotIndex
)
{
STsdbRepo
*
pRepo
=
pQueryHandle
->
pTsdb
;
int64_t
st
=
taosGetTimestampUs
();
...
...
@@ -766,7 +766,7 @@ static void moveDataToFront(STsdbQueryHandle* pQueryHandle, int32_t numOfRows, i
static
void
doCheckGeneratedBlockRange
(
STsdbQueryHandle
*
pQueryHandle
);
static
void
copyAllRemainRowsFromFileBlock
(
STsdbQueryHandle
*
pQueryHandle
,
STableCheckInfo
*
pCheckInfo
,
SDataBlockInfo
*
pBlockInfo
,
int32_t
endPos
);
static
int32_t
handleDataMergeIfNeeded
(
STsdbQueryHandle
*
pQueryHandle
,
S
Comp
Block
*
pBlock
,
STableCheckInfo
*
pCheckInfo
){
static
int32_t
handleDataMergeIfNeeded
(
STsdbQueryHandle
*
pQueryHandle
,
SBlock
*
pBlock
,
STableCheckInfo
*
pCheckInfo
){
SQueryFilePos
*
cur
=
&
pQueryHandle
->
cur
;
SDataBlockInfo
binfo
=
GET_FILE_DATA_BLOCK_INFO
(
pCheckInfo
,
pBlock
);
int32_t
code
=
TSDB_CODE_SUCCESS
;
...
...
@@ -848,7 +848,7 @@ static int32_t handleDataMergeIfNeeded(STsdbQueryHandle* pQueryHandle, SCompBloc
return
code
;
}
static
int32_t
loadFileDataBlock
(
STsdbQueryHandle
*
pQueryHandle
,
S
Comp
Block
*
pBlock
,
STableCheckInfo
*
pCheckInfo
,
bool
*
exists
)
{
static
int32_t
loadFileDataBlock
(
STsdbQueryHandle
*
pQueryHandle
,
SBlock
*
pBlock
,
STableCheckInfo
*
pCheckInfo
,
bool
*
exists
)
{
SQueryFilePos
*
cur
=
&
pQueryHandle
->
cur
;
int32_t
code
=
TSDB_CODE_SUCCESS
;
...
...
@@ -1255,7 +1255,7 @@ int32_t getEndPosInDataBlock(STsdbQueryHandle* pQueryHandle, SDataBlockInfo* pBl
// only return the qualified data to client in terms of query time window, data rows in the same block but do not
// be included in the query time window will be discarded
static
void
doMergeTwoLevelData
(
STsdbQueryHandle
*
pQueryHandle
,
STableCheckInfo
*
pCheckInfo
,
S
Comp
Block
*
pBlock
)
{
static
void
doMergeTwoLevelData
(
STsdbQueryHandle
*
pQueryHandle
,
STableCheckInfo
*
pCheckInfo
,
SBlock
*
pBlock
)
{
SQueryFilePos
*
cur
=
&
pQueryHandle
->
cur
;
SDataBlockInfo
blockInfo
=
GET_FILE_DATA_BLOCK_INFO
(
pCheckInfo
,
pBlock
);
...
...
@@ -1534,7 +1534,7 @@ static int32_t createDataBlocksInfo(STsdbQueryHandle* pQueryHandle, int32_t numO
continue
;
}
S
Comp
Block
*
pBlock
=
pTableCheck
->
pCompInfo
->
blocks
;
SBlock
*
pBlock
=
pTableCheck
->
pCompInfo
->
blocks
;
sup
.
numOfBlocksPerTable
[
numOfQualTables
]
=
pTableCheck
->
numOfBlocks
;
char
*
buf
=
calloc
(
1
,
sizeof
(
STableBlockInfo
)
*
pTableCheck
->
numOfBlocks
);
...
...
@@ -2238,7 +2238,7 @@ SArray* tsdbRetrieveDataBlock(TsdbQueryHandleT* pQueryHandle, SArray* pIdList) {
pBlockLoadInfo
->
tid
==
pCheckInfo
->
pTableObj
->
tableId
.
tid
)
{
return
pHandle
->
pColumns
;
}
else
{
// only load the file block
S
Comp
Block
*
pBlock
=
pBlockInfo
->
compBlock
;
SBlock
*
pBlock
=
pBlockInfo
->
compBlock
;
if
(
doLoadFileDataBlock
(
pHandle
,
pBlock
,
pCheckInfo
,
pHandle
->
cur
.
slot
)
!=
TSDB_CODE_SUCCESS
)
{
return
NULL
;
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录