test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-11-30 02:48:45 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

enhance: add the unmarshal method for the rbac msg (#35692)

Browse Source 
/kind improvement
- issue: #35391

Signed-off-by: SimFG <bang.fu@zilliz.com>
This commit is contained in:
SimFG 2024-08-30 07:23:00 +08:00 committed by GitHub
parent 671112d17b
commit 07bb071aaf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/mq/msgstream/unmarshal.go
View File

@ -65,6 +65,7 @@ func (pudf *ProtoUDFactory) NewUnmarshalDispatcher() *ProtoUnmarshalDispatcher {
createIndexMsg := CreateIndexMsg{}
dropIndexMsg := DropIndexMsg{}
alterIndexMsg := AlterIndexMsg{}
loadCollectionMsg := LoadCollectionMsg{}
releaseCollectionMsg := ReleaseCollectionMsg{}
@ -74,6 +75,15 @@ func (pudf *ProtoUDFactory) NewUnmarshalDispatcher() *ProtoUnmarshalDispatcher {
createDatabaseMsg := CreateDatabaseMsg{}
dropDatabaseMsg := DropDatabaseMsg{}
alterDatabaseMsg := AlterDatabaseMsg{}
createCredentialMsg := CreateUserMsg{}
deleteCredentialMsg := DeleteUserMsg{}
updateCredentialMsg := UpdateUserMsg{}
createRoleMsg := CreateRoleMsg{}
dropRoleMsg := DropRoleMsg{}
operateUserRoleMsg := OperateUserRoleMsg{}
operatePrivilegeMsg := OperatePrivilegeMsg{}
p := &ProtoUnmarshalDispatcher{}
p.TempMap = make(map[commonpb.MsgType]UnmarshalFunc)
@ -87,6 +97,7 @@ func (pudf *ProtoUDFactory) NewUnmarshalDispatcher() *ProtoUnmarshalDispatcher {
p.TempMap[commonpb.MsgType_DataNodeTt] = dataNodeTtMsg.Unmarshal
p.TempMap[commonpb.MsgType_CreateIndex] = createIndexMsg.Unmarshal
p.TempMap[commonpb.MsgType_DropIndex] = dropIndexMsg.Unmarshal
p.TempMap[commonpb.MsgType_AlterIndex] = alterIndexMsg.Unmarshal
p.TempMap[commonpb.MsgType_LoadCollection] = loadCollectionMsg.Unmarshal
p.TempMap[commonpb.MsgType_ReleaseCollection] = releaseCollectionMsg.Unmarshal
p.TempMap[commonpb.MsgType_LoadPartitions] = loadPartitionsMsg.Unmarshal
@ -94,6 +105,14 @@ func (pudf *ProtoUDFactory) NewUnmarshalDispatcher() *ProtoUnmarshalDispatcher {
p.TempMap[commonpb.MsgType_Flush] = flushMsg.Unmarshal
p.TempMap[commonpb.MsgType_CreateDatabase] = createDatabaseMsg.Unmarshal
p.TempMap[commonpb.MsgType_DropDatabase] = dropDatabaseMsg.Unmarshal
p.TempMap[commonpb.MsgType_AlterDatabase] = alterDatabaseMsg.Unmarshal
p.TempMap[commonpb.MsgType_CreateCredential] = createCredentialMsg.Unmarshal
p.TempMap[commonpb.MsgType_DeleteCredential] = deleteCredentialMsg.Unmarshal
p.TempMap[commonpb.MsgType_UpdateCredential] = updateCredentialMsg.Unmarshal
p.TempMap[commonpb.MsgType_CreateRole] = createRoleMsg.Unmarshal
p.TempMap[commonpb.MsgType_DropRole] = dropRoleMsg.Unmarshal
p.TempMap[commonpb.MsgType_OperateUserRole] = operateUserRoleMsg.Unmarshal
p.TempMap[commonpb.MsgType_OperatePrivilege] = operatePrivilegeMsg.Unmarshal
return p
}