Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
taosdata
TDengine
提交
824c87a6
T
TDengine
项目概览
taosdata
/
TDengine
大约 2 年 前同步成功
通知
1193
Star
22018
Fork
4786
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
TDengine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
824c87a6
编写于
3月 30, 2023
作者:
C
cademfly
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
binary format
上级
22b87dc2
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
190 addition
and
129 deletion
+190
-129
source/dnode/mnode/impl/inc/mndDef.h
source/dnode/mnode/impl/inc/mndDef.h
+2
-2
source/dnode/mnode/impl/src/mndUser.c
source/dnode/mnode/impl/src/mndUser.c
+188
-127
未找到文件。
source/dnode/mnode/impl/inc/mndDef.h
浏览文件 @
824c87a6
...
...
@@ -280,8 +280,8 @@ typedef struct {
SHashObj
*
readDbs
;
SHashObj
*
writeDbs
;
SHashObj
*
topics
;
SHashObj
*
read
St
bs
;
SHashObj
*
write
St
bs
;
SHashObj
*
read
T
bs
;
SHashObj
*
write
T
bs
;
SRWLatch
lock
;
}
SUserObj
;
...
...
source/dnode/mnode/impl/src/mndUser.c
浏览文件 @
824c87a6
...
...
@@ -125,14 +125,14 @@ SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
int32_t
numOfReadDbs
=
taosHashGetSize
(
pUser
->
readDbs
);
int32_t
numOfWriteDbs
=
taosHashGetSize
(
pUser
->
writeDbs
);
int32_t
numOfReadStbs
=
taosHashGetSize
(
pUser
->
read
St
bs
);
int32_t
numOfWriteStbs
=
taosHashGetSize
(
pUser
->
write
St
bs
);
int32_t
numOfReadStbs
=
taosHashGetSize
(
pUser
->
read
T
bs
);
int32_t
numOfWriteStbs
=
taosHashGetSize
(
pUser
->
write
T
bs
);
int32_t
numOfTopics
=
taosHashGetSize
(
pUser
->
topics
);
int32_t
size
=
sizeof
(
SUserObj
)
+
USER_RESERVE_SIZE
+
(
numOfReadDbs
+
numOfWriteDbs
)
*
TSDB_DB_FNAME_LEN
+
numOfTopics
*
TSDB_TOPIC_FNAME_LEN
;
char
*
stb
=
taosHashIterate
(
pUser
->
read
St
bs
,
NULL
);
char
*
stb
=
taosHashIterate
(
pUser
->
read
T
bs
,
NULL
);
while
(
stb
!=
NULL
)
{
size_t
keyLen
=
0
;
void
*
key
=
taosHashGetKey
(
stb
,
&
keyLen
);
...
...
@@ -143,10 +143,10 @@ SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
valueLen
=
strlen
(
stb
);
size
+=
sizeof
(
int32_t
);
size
+=
valueLen
;
stb
=
taosHashIterate
(
pUser
->
read
St
bs
,
stb
);
stb
=
taosHashIterate
(
pUser
->
read
T
bs
,
stb
);
}
stb
=
taosHashIterate
(
pUser
->
write
St
bs
,
NULL
);
stb
=
taosHashIterate
(
pUser
->
write
T
bs
,
NULL
);
while
(
stb
!=
NULL
)
{
size_t
keyLen
=
0
;
void
*
key
=
taosHashGetKey
(
stb
,
&
keyLen
);
...
...
@@ -157,7 +157,7 @@ SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
valueLen
=
strlen
(
stb
);
size
+=
sizeof
(
int32_t
);
size
+=
keyLen
;
stb
=
taosHashIterate
(
pUser
->
write
St
bs
,
stb
);
stb
=
taosHashIterate
(
pUser
->
write
T
bs
,
stb
);
}
SSdbRaw
*
pRaw
=
sdbAllocRaw
(
SDB_USER
,
USER_VER_NUMBER
,
size
);
...
...
@@ -198,7 +198,7 @@ SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
topic
=
taosHashIterate
(
pUser
->
topics
,
topic
);
}
stb
=
taosHashIterate
(
pUser
->
read
St
bs
,
NULL
);
stb
=
taosHashIterate
(
pUser
->
read
T
bs
,
NULL
);
while
(
stb
!=
NULL
)
{
size_t
keyLen
=
0
;
void
*
key
=
taosHashGetKey
(
stb
,
&
keyLen
);
...
...
@@ -206,13 +206,13 @@ SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
SDB_SET_BINARY
(
pRaw
,
dataPos
,
key
,
keyLen
,
_OVER
);
size_t
valueLen
=
0
;
valueLen
=
strlen
(
stb
);
valueLen
=
strlen
(
stb
)
+
1
;
SDB_SET_INT32
(
pRaw
,
dataPos
,
valueLen
,
_OVER
)
SDB_SET_BINARY
(
pRaw
,
dataPos
,
stb
,
valueLen
,
_OVER
);
stb
=
taosHashIterate
(
pUser
->
read
St
bs
,
stb
);
stb
=
taosHashIterate
(
pUser
->
read
T
bs
,
stb
);
}
stb
=
taosHashIterate
(
pUser
->
write
St
bs
,
NULL
);
stb
=
taosHashIterate
(
pUser
->
write
T
bs
,
NULL
);
while
(
stb
!=
NULL
)
{
size_t
keyLen
=
0
;
void
*
key
=
taosHashGetKey
(
stb
,
&
keyLen
);
...
...
@@ -220,10 +220,10 @@ SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
SDB_SET_BINARY
(
pRaw
,
dataPos
,
key
,
keyLen
,
_OVER
);
size_t
valueLen
=
0
;
valueLen
=
strlen
(
stb
);
valueLen
=
strlen
(
stb
)
+
1
;
SDB_SET_INT32
(
pRaw
,
dataPos
,
valueLen
,
_OVER
)
SDB_SET_BINARY
(
pRaw
,
dataPos
,
stb
,
valueLen
,
_OVER
);
stb
=
taosHashIterate
(
pUser
->
write
St
bs
,
stb
);
stb
=
taosHashIterate
(
pUser
->
write
T
bs
,
stb
);
}
SDB_SET_RESERVE
(
pRaw
,
dataPos
,
USER_RESERVE_SIZE
,
_OVER
)
...
...
@@ -292,8 +292,8 @@ static SSdbRow *mndUserActionDecode(SSdbRaw *pRaw) {
pUser
->
writeDbs
=
taosHashInit
(
numOfWriteDbs
,
taosGetDefaultHashFunction
(
TSDB_DATA_TYPE_BINARY
),
true
,
HASH_ENTRY_LOCK
);
pUser
->
topics
=
taosHashInit
(
numOfTopics
,
taosGetDefaultHashFunction
(
TSDB_DATA_TYPE_BINARY
),
true
,
HASH_ENTRY_LOCK
);
pUser
->
read
St
bs
=
taosHashInit
(
numOfReadStbs
,
taosGetDefaultHashFunction
(
TSDB_DATA_TYPE_BINARY
),
true
,
HASH_ENTRY_LOCK
);
pUser
->
write
St
bs
=
pUser
->
read
T
bs
=
taosHashInit
(
numOfReadStbs
,
taosGetDefaultHashFunction
(
TSDB_DATA_TYPE_BINARY
),
true
,
HASH_ENTRY_LOCK
);
pUser
->
write
T
bs
=
taosHashInit
(
numOfWriteStbs
,
taosGetDefaultHashFunction
(
TSDB_DATA_TYPE_BINARY
),
true
,
HASH_ENTRY_LOCK
);
if
(
pUser
->
readDbs
==
NULL
||
pUser
->
writeDbs
==
NULL
||
pUser
->
topics
==
NULL
)
goto
_OVER
;
...
...
@@ -325,15 +325,17 @@ static SSdbRow *mndUserActionDecode(SSdbRaw *pRaw) {
int32_t
keyLen
=
0
;
SDB_GET_INT32
(
pRaw
,
dataPos
,
&
keyLen
,
_OVER
);
char
*
key
=
taosMemoryCalloc
(
keyLen
+
1
,
sizeof
(
char
));
char
*
key
=
taosMemoryCalloc
(
keyLen
,
sizeof
(
char
));
memset
(
key
,
0
,
keyLen
);
SDB_GET_BINARY
(
pRaw
,
dataPos
,
key
,
keyLen
,
_OVER
);
int32_t
valuelen
=
0
;
SDB_GET_INT32
(
pRaw
,
dataPos
,
&
valuelen
,
_OVER
);
char
*
value
=
taosMemoryCalloc
(
valuelen
+
1
,
sizeof
(
char
));
char
*
value
=
taosMemoryCalloc
(
valuelen
,
sizeof
(
char
));
memset
(
value
,
0
,
keyLen
);
SDB_GET_BINARY
(
pRaw
,
dataPos
,
value
,
valuelen
,
_OVER
)
taosHashPut
(
pUser
->
read
St
bs
,
key
,
keyLen
,
value
,
valuelen
);
taosHashPut
(
pUser
->
read
T
bs
,
key
,
keyLen
,
value
,
valuelen
);
taosMemoryFree
(
key
);
taosMemoryFree
(
value
);
...
...
@@ -343,15 +345,17 @@ static SSdbRow *mndUserActionDecode(SSdbRaw *pRaw) {
int32_t
keyLen
=
0
;
SDB_GET_INT32
(
pRaw
,
dataPos
,
&
keyLen
,
_OVER
);
char
*
key
=
taosMemoryCalloc
(
keyLen
+
1
,
sizeof
(
char
));
char
*
key
=
taosMemoryCalloc
(
keyLen
,
sizeof
(
char
));
memset
(
key
,
0
,
keyLen
);
SDB_GET_BINARY
(
pRaw
,
dataPos
,
key
,
keyLen
,
_OVER
);
int32_t
valuelen
=
0
;
SDB_GET_INT32
(
pRaw
,
dataPos
,
&
valuelen
,
_OVER
);
char
*
value
=
taosMemoryCalloc
(
valuelen
+
1
,
sizeof
(
char
));
char
*
value
=
taosMemoryCalloc
(
valuelen
,
sizeof
(
char
));
memset
(
value
,
0
,
keyLen
);
SDB_GET_BINARY
(
pRaw
,
dataPos
,
value
,
valuelen
,
_OVER
)
taosHashPut
(
pUser
->
write
St
bs
,
key
,
keyLen
,
value
,
valuelen
);
taosHashPut
(
pUser
->
write
T
bs
,
key
,
keyLen
,
value
,
valuelen
);
taosMemoryFree
(
key
);
taosMemoryFree
(
value
);
...
...
@@ -370,8 +374,8 @@ _OVER:
taosHashCleanup
(
pUser
->
readDbs
);
taosHashCleanup
(
pUser
->
writeDbs
);
taosHashCleanup
(
pUser
->
topics
);
taosHashCleanup
(
pUser
->
read
St
bs
);
taosHashCleanup
(
pUser
->
write
St
bs
);
taosHashCleanup
(
pUser
->
read
T
bs
);
taosHashCleanup
(
pUser
->
write
T
bs
);
}
taosMemoryFreeClear
(
pRow
);
return
NULL
;
...
...
@@ -404,8 +408,8 @@ static int32_t mndUserDupObj(SUserObj *pUser, SUserObj *pNew) {
taosRLockLatch
(
&
pUser
->
lock
);
pNew
->
readDbs
=
mndDupDbHash
(
pUser
->
readDbs
);
pNew
->
writeDbs
=
mndDupDbHash
(
pUser
->
writeDbs
);
pNew
->
read
Stbs
=
mndDupTopicHash
(
pUser
->
readSt
bs
);
pNew
->
write
Stbs
=
mndDupTopicHash
(
pUser
->
writeSt
bs
);
pNew
->
read
Tbs
=
mndDupTopicHash
(
pUser
->
readT
bs
);
pNew
->
write
Tbs
=
mndDupTopicHash
(
pUser
->
writeT
bs
);
pNew
->
topics
=
mndDupTopicHash
(
pUser
->
topics
);
taosRUnLockLatch
(
&
pUser
->
lock
);
...
...
@@ -419,13 +423,13 @@ static void mndUserFreeObj(SUserObj *pUser) {
taosHashCleanup
(
pUser
->
readDbs
);
taosHashCleanup
(
pUser
->
writeDbs
);
taosHashCleanup
(
pUser
->
topics
);
taosHashCleanup
(
pUser
->
read
St
bs
);
taosHashCleanup
(
pUser
->
write
St
bs
);
taosHashCleanup
(
pUser
->
read
T
bs
);
taosHashCleanup
(
pUser
->
write
T
bs
);
pUser
->
readDbs
=
NULL
;
pUser
->
writeDbs
=
NULL
;
pUser
->
topics
=
NULL
;
pUser
->
read
St
bs
=
NULL
;
pUser
->
write
St
bs
=
NULL
;
pUser
->
read
T
bs
=
NULL
;
pUser
->
write
T
bs
=
NULL
;
}
static
int32_t
mndUserActionDelete
(
SSdb
*
pSdb
,
SUserObj
*
pUser
)
{
...
...
@@ -445,8 +449,8 @@ static int32_t mndUserActionUpdate(SSdb *pSdb, SUserObj *pOld, SUserObj *pNew) {
TSWAP
(
pOld
->
readDbs
,
pNew
->
readDbs
);
TSWAP
(
pOld
->
writeDbs
,
pNew
->
writeDbs
);
TSWAP
(
pOld
->
topics
,
pNew
->
topics
);
TSWAP
(
pOld
->
read
Stbs
,
pNew
->
readSt
bs
);
TSWAP
(
pOld
->
write
Stbs
,
pNew
->
writeSt
bs
);
TSWAP
(
pOld
->
read
Tbs
,
pNew
->
readT
bs
);
TSWAP
(
pOld
->
write
Tbs
,
pNew
->
writeT
bs
);
taosWUnLockLatch
(
&
pOld
->
lock
);
return
0
;
...
...
@@ -767,7 +771,7 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
if
(
taosHashPut
(
newUser
.
read
Stbs
,
tbFName
,
len
,
tbFName
,
TSDB_TABLE_NAME_LEN
)
!=
0
)
{
if
(
taosHashPut
(
newUser
.
read
Tbs
,
tbFName
,
len
,
tbFName
,
len
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
...
...
@@ -777,7 +781,7 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
pIter
=
sdbFetch
(
pSdb
,
SDB_STB
,
pIter
,
(
void
**
)
&
pStb
);
if
(
pIter
==
NULL
)
break
;
int32_t
len
=
strlen
(
pStb
->
name
)
+
1
;
taosHashPut
(
newUser
.
read
Stbs
,
pStb
->
name
,
len
,
pStb
->
name
,
TSDB_TABLE_NAME_LEN
);
taosHashPut
(
newUser
.
read
Tbs
,
pStb
->
name
,
len
,
pStb
->
name
,
len
);
sdbRelease
(
pSdb
,
pStb
);
}
}
...
...
@@ -794,7 +798,7 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
if
(
taosHashPut
(
newUser
.
write
Stbs
,
tbFName
,
len
,
tbFName
,
TSDB_TABLE_NAME_LEN
)
!=
0
)
{
if
(
taosHashPut
(
newUser
.
write
Tbs
,
tbFName
,
len
,
tbFName
,
len
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
...
...
@@ -804,12 +808,85 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
pIter
=
sdbFetch
(
pSdb
,
SDB_STB
,
pIter
,
(
void
**
)
&
pStb
);
if
(
pIter
==
NULL
)
break
;
int32_t
len
=
strlen
(
pStb
->
name
)
+
1
;
taosHashPut
(
newUser
.
write
St
bs
,
pStb
->
name
,
len
,
pStb
->
name
,
TSDB_DB_FNAME_LEN
);
taosHashPut
(
newUser
.
write
T
bs
,
pStb
->
name
,
len
,
pStb
->
name
,
TSDB_DB_FNAME_LEN
);
sdbRelease
(
pSdb
,
pStb
);
}
}
}
if
(
alterReq
.
alterType
==
TSDB_ALTER_USER_REMOVE_READ_TABLE
||
alterReq
.
alterType
==
TSDB_ALTER_USER_REMOVE_READ_TAG
||
alterReq
.
alterType
==
TSDB_ALTER_USER_REMOVE_ALL_TABLE
)
{
if
(
strcmp
(
alterReq
.
objname
,
"1.*"
)
!=
0
)
{
char
tbFName
[
TSDB_TABLE_FNAME_LEN
]
=
{
0
};
snprintf
(
tbFName
,
sizeof
(
tbFName
),
"%s.%s"
,
alterReq
.
objname
,
alterReq
.
tabName
);
int32_t
len
=
strlen
(
tbFName
)
+
1
;
SStbObj
*
pStb
=
mndAcquireStb
(
pMnode
,
tbFName
);
if
(
pStb
==
NULL
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
if
(
taosHashRemove
(
newUser
.
readTbs
,
tbFName
,
len
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
}
else
{
while
(
1
)
{
SStbObj
*
pStb
=
NULL
;
pIter
=
sdbFetch
(
pSdb
,
SDB_STB
,
pIter
,
(
void
**
)
&
pStb
);
if
(
pIter
==
NULL
)
break
;
int32_t
len
=
strlen
(
pStb
->
name
)
+
1
;
if
(
strcmp
(
pStb
->
db
,
alterReq
.
objname
)
==
0
){
if
(
taosHashRemove
(
newUser
.
readTbs
,
pStb
->
name
,
len
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
}
//taosHashPut(newUser.writeStbs, pStb->name, len, pStb->name, TSDB_DB_FNAME_LEN);
//sdbRelease(pSdb, pStb);
}
//taosHashClear(newUser.readStbs);
}
}
if
(
alterReq
.
alterType
==
TSDB_ALTER_USER_REMOVE_WRITE_TABLE
||
alterReq
.
alterType
==
TSDB_ALTER_USER_REMOVE_WRITE_TAG
||
alterReq
.
alterType
==
TSDB_ALTER_USER_REMOVE_ALL_TABLE
)
{
if
(
strcmp
(
alterReq
.
objname
,
"1.*"
)
!=
0
)
{
char
tbFName
[
TSDB_TABLE_FNAME_LEN
]
=
{
0
};
snprintf
(
tbFName
,
sizeof
(
tbFName
),
"%s.%s"
,
alterReq
.
objname
,
alterReq
.
tabName
);
int32_t
len
=
strlen
(
tbFName
)
+
1
;
SStbObj
*
pStb
=
mndAcquireStb
(
pMnode
,
tbFName
);
if
(
pStb
==
NULL
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
if
(
taosHashRemove
(
newUser
.
writeTbs
,
tbFName
,
len
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
}
else
{
while
(
1
)
{
SStbObj
*
pStb
=
NULL
;
pIter
=
sdbFetch
(
pSdb
,
SDB_STB
,
pIter
,
(
void
**
)
&
pStb
);
if
(
pIter
==
NULL
)
break
;
int32_t
len
=
strlen
(
pStb
->
name
)
+
1
;
if
(
strcmp
(
pStb
->
db
,
alterReq
.
objname
)
==
0
){
if
(
taosHashRemove
(
newUser
.
writeTbs
,
pStb
->
name
,
len
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
}
}
}
}
if
(
alterReq
.
alterType
==
TSDB_ALTER_USER_ADD_SUBSCRIBE_TOPIC
)
{
int32_t
len
=
strlen
(
alterReq
.
objname
)
+
1
;
SMqTopicObj
*
pTopic
=
mndAcquireTopic
(
pMnode
,
alterReq
.
objname
);
...
...
@@ -822,9 +899,9 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
if
(
alterReq
.
alterType
==
TSDB_ALTER_USER_ADD_READ_TAG
)
{
char
tbFName
[
TSDB_TABLE_FNAME_LEN
]
=
{
0
};
snprintf
(
tbFName
,
sizeof
(
tbFName
),
"%s.%s"
,
alterReq
.
objname
,
alterReq
.
tabName
);
snprintf
(
tbFName
,
TSDB_TABLE_FNAME_LEN
,
"%s.%s"
,
alterReq
.
objname
,
alterReq
.
tabName
);
int32_t
len
=
strlen
(
tbFName
)
+
1
;
SStbObj
*
pStb
=
mndAcquireStb
(
pMnode
,
tbFName
);
if
(
pStb
==
NULL
)
{
mndReleaseStb
(
pMnode
,
pStb
);
...
...
@@ -834,8 +911,9 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
int32_t
condLen
=
strlen
(
alterReq
.
tagCond
);
if
(
taosHashPut
(
newUser
.
readStbs
,
tbFName
,
len
,
alterReq
.
tagCond
,
strlen
(
alterReq
.
tagCond
))
!=
0
)
{
int32_t
condLen
=
strlen
(
alterReq
.
tagCond
)
+
1
;
if
(
taosHashPut
(
newUser
.
readTbs
,
tbFName
,
len
,
alterReq
.
tagCond
,
condLen
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
...
...
@@ -844,8 +922,8 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
if
(
alterReq
.
alterType
==
TSDB_ALTER_USER_ADD_WRITE_TAG
)
{
char
tbFName
[
TSDB_TABLE_FNAME_LEN
]
=
{
0
};
snprintf
(
tbFName
,
sizeof
(
tbFName
),
"%s.%s"
,
alterReq
.
objname
,
alterReq
.
tabName
);
int32_t
len
=
strlen
(
tbFName
)
+
1
;
SStbObj
*
pStb
=
mndAcquireStb
(
pMnode
,
tbFName
);
if
(
pStb
==
NULL
)
{
mndReleaseStb
(
pMnode
,
pStb
);
...
...
@@ -855,8 +933,8 @@ static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
int32_t
condLen
=
strlen
(
alterReq
.
tagCond
);
if
(
taosHashPut
(
newUser
.
write
St
bs
,
tbFName
,
len
,
alterReq
.
tagCond
,
condLen
)
!=
0
)
{
int32_t
condLen
=
strlen
(
alterReq
.
tagCond
)
+
1
;
if
(
taosHashPut
(
newUser
.
write
T
bs
,
tbFName
,
len
,
alterReq
.
tagCond
,
condLen
)
!=
0
)
{
mndReleaseStb
(
pMnode
,
pStb
);
goto
_OVER
;
}
...
...
@@ -1045,6 +1123,71 @@ static void mndCancelGetNextUser(SMnode *pMnode, void *pIter) {
sdbCancelFetch
(
pSdb
,
pIter
);
}
static
void
mndLoopHash
(
SHashObj
*
hash
,
char
*
priType
,
SSDataBlock
*
pBlock
,
int32_t
*
numOfRows
,
char
*
user
,
SShowObj
*
pShow
){
char
*
value
=
taosHashIterate
(
hash
,
NULL
);
int32_t
cols
=
0
;
while
(
value
!=
NULL
)
{
cols
=
0
;
char
userName
[
TSDB_USER_LEN
+
VARSTR_HEADER_SIZE
]
=
{
0
};
STR_WITH_MAXSIZE_TO_VARSTR
(
userName
,
user
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
SColumnInfoData
*
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
*
numOfRows
,
(
const
char
*
)
userName
,
false
);
char
privilege
[
20
]
=
{
0
};
STR_WITH_MAXSIZE_TO_VARSTR
(
privilege
,
priType
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
*
numOfRows
,
(
const
char
*
)
privilege
,
false
);
size_t
keyLen
=
0
;
void
*
key
=
taosHashGetKey
(
value
,
&
keyLen
);
char
tableName
[
TSDB_TABLE_NAME_LEN
]
=
{
0
};
mndExtractTbNameFromStbFullName
(
key
,
tableName
,
TSDB_TABLE_NAME_LEN
);
if
(
strcmp
(
key
,
value
)
==
0
){
char
*
obj
=
taosMemoryMalloc
(
TSDB_TABLE_NAME_LEN
+
VARSTR_HEADER_SIZE
);
STR_WITH_MAXSIZE_TO_VARSTR
(
obj
,
tableName
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
*
numOfRows
,
(
const
char
*
)
obj
,
false
);
taosMemoryFree
(
obj
);
}
else
{
SNode
*
pAst
=
NULL
;
int32_t
sqlLen
=
0
;
char
*
sql
=
NULL
;
if
(
nodesStringToNode
(
value
,
&
pAst
)
==
0
)
{
sql
=
taosMemoryMalloc
(
TSDB_EXPLAIN_RESULT_ROW_SIZE
);
nodesNodeToSQL
(
pAst
,
sql
,
TSDB_EXPLAIN_RESULT_ROW_SIZE
,
&
sqlLen
);
}
else
{
sqlLen
=
5
;
sql
=
taosMemoryMalloc
(
sqlLen
+
1
);
sprintf
(
sql
,
"error"
);
}
int32_t
contentLen
=
sqlLen
+
TSDB_TABLE_NAME_LEN
+
3
;
char
*
content
=
taosMemoryMalloc
(
contentLen
);
if
(
sql
!=
NULL
){
sprintf
(
content
,
"%s(%s)"
,
tableName
,
sql
);
taosMemoryFree
(
sql
);
}
char
*
obj
=
taosMemoryMalloc
(
contentLen
+
VARSTR_HEADER_SIZE
);
STR_WITH_MAXSIZE_TO_VARSTR
(
obj
,
content
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
taosMemoryFree
(
content
);
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
*
numOfRows
,
(
const
char
*
)
obj
,
false
);
taosMemoryFree
(
obj
);
}
(
*
numOfRows
)
++
;
value
=
taosHashIterate
(
hash
,
value
);
}
}
static
int32_t
mndRetrievePrivileges
(
SRpcMsg
*
pReq
,
SShowObj
*
pShow
,
SSDataBlock
*
pBlock
,
int32_t
rows
)
{
SMnode
*
pMnode
=
pReq
->
info
.
node
;
SSdb
*
pSdb
=
pMnode
->
pSdb
;
...
...
@@ -1132,91 +1275,9 @@ static int32_t mndRetrievePrivileges(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock
db
=
taosHashIterate
(
pUser
->
writeDbs
,
db
);
}
char
*
stb
=
taosHashIterate
(
pUser
->
readStbs
,
NULL
);
while
(
stb
!=
NULL
)
{
cols
=
0
;
char
userName
[
TSDB_USER_LEN
+
VARSTR_HEADER_SIZE
]
=
{
0
};
STR_WITH_MAXSIZE_TO_VARSTR
(
userName
,
pUser
->
user
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
SColumnInfoData
*
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
numOfRows
,
(
const
char
*
)
userName
,
false
);
char
privilege
[
20
]
=
{
0
};
STR_WITH_MAXSIZE_TO_VARSTR
(
privilege
,
"read"
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
numOfRows
,
(
const
char
*
)
privilege
,
false
);
size_t
superTableLen
=
0
;
void
*
superTable
=
taosHashGetKey
(
stb
,
&
superTableLen
);
char
objName
[
TSDB_TABLE_NAME_LEN
]
=
{
0
};
mndExtractTbNameFromStbFullName
(
superTable
,
objName
,
TSDB_TABLE_NAME_LEN
);
SNode
*
pAst
=
NULL
;
int32_t
code
=
nodesStringToNode
(
stb
,
&
pAst
);
char
*
buf
=
taosMemoryCalloc
(
TSDB_EXPLAIN_RESULT_ROW_SIZE
,
sizeof
(
char
));
int32_t
sqlLen
=
0
;
nodesNodeToSQL
(
pAst
,
buf
,
TSDB_EXPLAIN_RESULT_ROW_SIZE
*
sizeof
(
char
),
&
sqlLen
);
char
*
value
=
taosMemoryCalloc
(
sqlLen
+
TSDB_TABLE_NAME_LEN
+
1
,
sizeof
(
char
));
sprintf
(
value
,
"%s(%s)"
,
objName
,
buf
);
char
*
obj
=
taosMemoryCalloc
(
sqlLen
+
TSDB_TABLE_NAME_LEN
+
VARSTR_HEADER_SIZE
+
1
,
sizeof
(
char
));
STR_WITH_MAXSIZE_TO_VARSTR
(
obj
,
value
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
numOfRows
,
(
const
char
*
)
obj
,
false
);
taosMemoryFree
(
value
);
taosMemoryFree
(
buf
);
taosMemoryFree
(
obj
);
numOfRows
++
;
stb
=
taosHashIterate
(
pUser
->
readStbs
,
stb
);
}
stb
=
taosHashIterate
(
pUser
->
writeStbs
,
NULL
);
while
(
stb
!=
NULL
)
{
cols
=
0
;
char
userName
[
TSDB_USER_LEN
+
VARSTR_HEADER_SIZE
]
=
{
0
};
STR_WITH_MAXSIZE_TO_VARSTR
(
userName
,
pUser
->
user
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
SColumnInfoData
*
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
numOfRows
,
(
const
char
*
)
userName
,
false
);
mndLoopHash
(
pUser
->
readTbs
,
"read"
,
pBlock
,
&
numOfRows
,
pUser
->
user
,
pShow
);
char
privilege
[
20
]
=
{
0
};
STR_WITH_MAXSIZE_TO_VARSTR
(
privilege
,
"write"
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
numOfRows
,
(
const
char
*
)
privilege
,
false
);
size_t
superTableLen
=
0
;
void
*
superTable
=
taosHashGetKey
(
stb
,
&
superTableLen
);
char
objName
[
TSDB_TABLE_NAME_LEN
]
=
{
0
};
mndExtractTbNameFromStbFullName
(
superTable
,
objName
,
TSDB_TABLE_NAME_LEN
);
SNode
*
pAst
=
NULL
;
int32_t
code
=
nodesStringToNode
(
stb
,
&
pAst
);
char
*
buf
=
taosMemoryCalloc
(
TSDB_EXPLAIN_RESULT_ROW_SIZE
,
sizeof
(
char
));
int32_t
sqlLen
=
0
;
nodesNodeToSQL
(
pAst
,
buf
,
TSDB_EXPLAIN_RESULT_ROW_SIZE
*
sizeof
(
char
),
&
sqlLen
);
char
*
value
=
taosMemoryCalloc
(
sqlLen
+
TSDB_TABLE_NAME_LEN
+
1
,
sizeof
(
char
));
sprintf
(
value
,
"%s(%s)"
,
objName
,
buf
);
char
*
obj
=
taosMemoryCalloc
(
sqlLen
+
TSDB_TABLE_NAME_LEN
+
VARSTR_HEADER_SIZE
+
1
,
sizeof
(
char
));
STR_WITH_MAXSIZE_TO_VARSTR
(
obj
,
value
,
pShow
->
pMeta
->
pSchemas
[
cols
].
bytes
);
pColInfo
=
taosArrayGet
(
pBlock
->
pDataBlock
,
cols
++
);
colDataSetVal
(
pColInfo
,
numOfRows
,
(
const
char
*
)
obj
,
false
);
taosMemoryFree
(
value
);
taosMemoryFree
(
buf
);
taosMemoryFree
(
obj
);
numOfRows
++
;
stb
=
taosHashIterate
(
pUser
->
writeStbs
,
stb
);
}
mndLoopHash
(
pUser
->
writeTbs
,
"write"
,
pBlock
,
&
numOfRows
,
pUser
->
user
,
pShow
);
char
*
topic
=
taosHashIterate
(
pUser
->
topics
,
NULL
);
while
(
topic
!=
NULL
)
{
...
...
@@ -1340,8 +1401,8 @@ int32_t mndUserRemoveDb(SMnode *pMnode, STrans *pTrans, char *db) {
if
(
inRead
||
inWrite
)
{
(
void
)
taosHashRemove
(
newUser
.
readDbs
,
db
,
len
);
(
void
)
taosHashRemove
(
newUser
.
writeDbs
,
db
,
len
);
(
void
)
taosHashRemove
(
newUser
.
read
St
bs
,
db
,
len
);
(
void
)
taosHashRemove
(
newUser
.
write
St
bs
,
db
,
len
);
(
void
)
taosHashRemove
(
newUser
.
read
T
bs
,
db
,
len
);
(
void
)
taosHashRemove
(
newUser
.
write
T
bs
,
db
,
len
);
SSdbRaw
*
pCommitRaw
=
mndUserActionEncode
(
&
newUser
);
if
(
pCommitRaw
==
NULL
||
mndTransAppendCommitlog
(
pTrans
,
pCommitRaw
)
!=
0
)
break
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录