Skip to content

Commit 89bb75f

Browse files
tasty-gumitasty-gumi
committed
fix: gen grpc for test
1 parent 2376f6a commit 89bb75f

File tree

7 files changed

+592
-401
lines changed

7 files changed

+592
-401
lines changed

Diff for: ‎pymilvus/grpc_gen/common_pb2.py

+18-18
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Diff for: ‎pymilvus/grpc_gen/common_pb2.pyi

+14
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,8 @@ class MsgType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
151151
Flush: _ClassVar[MsgType]
152152
ResendSegmentStats: _ClassVar[MsgType]
153153
Upsert: _ClassVar[MsgType]
154+
ManualFlush: _ClassVar[MsgType]
155+
FlushSegment: _ClassVar[MsgType]
154156
Search: _ClassVar[MsgType]
155157
SearchResult: _ClassVar[MsgType]
156158
GetIndexState: _ClassVar[MsgType]
@@ -302,6 +304,11 @@ class ObjectPrivilege(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
302304
PrivilegeUpdateResourceGroups: _ClassVar[ObjectPrivilege]
303305
PrivilegeAlterDatabase: _ClassVar[ObjectPrivilege]
304306
PrivilegeDescribeDatabase: _ClassVar[ObjectPrivilege]
307+
PrivilegeBackupRBAC: _ClassVar[ObjectPrivilege]
308+
PrivilegeRestoreRBAC: _ClassVar[ObjectPrivilege]
309+
PrivilegeGroupReadOnly: _ClassVar[ObjectPrivilege]
310+
PrivilegeGroupReadWrite: _ClassVar[ObjectPrivilege]
311+
PrivilegeGroupAdmin: _ClassVar[ObjectPrivilege]
305312

306313
class StateCode(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
307314
__slots__ = ()
@@ -444,6 +451,8 @@ Delete: MsgType
444451
Flush: MsgType
445452
ResendSegmentStats: MsgType
446453
Upsert: MsgType
454+
ManualFlush: MsgType
455+
FlushSegment: MsgType
447456
Search: MsgType
448457
SearchResult: MsgType
449458
GetIndexState: MsgType
@@ -577,6 +586,11 @@ PrivilegeListAliases: ObjectPrivilege
577586
PrivilegeUpdateResourceGroups: ObjectPrivilege
578587
PrivilegeAlterDatabase: ObjectPrivilege
579588
PrivilegeDescribeDatabase: ObjectPrivilege
589+
PrivilegeBackupRBAC: ObjectPrivilege
590+
PrivilegeRestoreRBAC: ObjectPrivilege
591+
PrivilegeGroupReadOnly: ObjectPrivilege
592+
PrivilegeGroupReadWrite: ObjectPrivilege
593+
PrivilegeGroupAdmin: ObjectPrivilege
580594
Initializing: StateCode
581595
Healthy: StateCode
582596
Abnormal: StateCode

Diff for: ‎pymilvus/grpc_gen/milvus_pb2.py

+346-332
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Diff for: ‎pymilvus/grpc_gen/milvus_pb2.pyi

+46-2
Original file line numberDiff line numberDiff line change
@@ -1461,12 +1461,14 @@ class CreateRoleRequest(_message.Message):
14611461
def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., entity: _Optional[_Union[RoleEntity, _Mapping]] = ...) -> None: ...
14621462

14631463
class DropRoleRequest(_message.Message):
1464-
__slots__ = ("base", "role_name")
1464+
__slots__ = ("base", "role_name", "force_drop")
14651465
BASE_FIELD_NUMBER: _ClassVar[int]
14661466
ROLE_NAME_FIELD_NUMBER: _ClassVar[int]
1467+
FORCE_DROP_FIELD_NUMBER: _ClassVar[int]
14671468
base: _common_pb2.MsgBase
14681469
role_name: str
1469-
def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., role_name: _Optional[str] = ...) -> None: ...
1470+
force_drop: bool
1471+
def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., role_name: _Optional[str] = ..., force_drop: bool = ...) -> None: ...
14701472

14711473
class OperateUserRoleRequest(_message.Message):
14721474
__slots__ = ("base", "username", "role_name", "type")
@@ -1598,6 +1600,48 @@ class OperatePrivilegeRequest(_message.Message):
15981600
type: OperatePrivilegeType
15991601
def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., entity: _Optional[_Union[GrantEntity, _Mapping]] = ..., type: _Optional[_Union[OperatePrivilegeType, str]] = ...) -> None: ...
16001602

1603+
class UserInfo(_message.Message):
1604+
__slots__ = ("user", "password", "roles")
1605+
USER_FIELD_NUMBER: _ClassVar[int]
1606+
PASSWORD_FIELD_NUMBER: _ClassVar[int]
1607+
ROLES_FIELD_NUMBER: _ClassVar[int]
1608+
user: str
1609+
password: str
1610+
roles: _containers.RepeatedCompositeFieldContainer[RoleEntity]
1611+
def __init__(self, user: _Optional[str] = ..., password: _Optional[str] = ..., roles: _Optional[_Iterable[_Union[RoleEntity, _Mapping]]] = ...) -> None: ...
1612+
1613+
class RBACMeta(_message.Message):
1614+
__slots__ = ("users", "roles", "grants")
1615+
USERS_FIELD_NUMBER: _ClassVar[int]
1616+
ROLES_FIELD_NUMBER: _ClassVar[int]
1617+
GRANTS_FIELD_NUMBER: _ClassVar[int]
1618+
users: _containers.RepeatedCompositeFieldContainer[UserInfo]
1619+
roles: _containers.RepeatedCompositeFieldContainer[RoleEntity]
1620+
grants: _containers.RepeatedCompositeFieldContainer[GrantEntity]
1621+
def __init__(self, users: _Optional[_Iterable[_Union[UserInfo, _Mapping]]] = ..., roles: _Optional[_Iterable[_Union[RoleEntity, _Mapping]]] = ..., grants: _Optional[_Iterable[_Union[GrantEntity, _Mapping]]] = ...) -> None: ...
1622+
1623+
class BackupRBACMetaRequest(_message.Message):
1624+
__slots__ = ("base",)
1625+
BASE_FIELD_NUMBER: _ClassVar[int]
1626+
base: _common_pb2.MsgBase
1627+
def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ...) -> None: ...
1628+
1629+
class BackupRBACMetaResponse(_message.Message):
1630+
__slots__ = ("status", "RBAC_meta")
1631+
STATUS_FIELD_NUMBER: _ClassVar[int]
1632+
RBAC_META_FIELD_NUMBER: _ClassVar[int]
1633+
status: _common_pb2.Status
1634+
RBAC_meta: RBACMeta
1635+
def __init__(self, status: _Optional[_Union[_common_pb2.Status, _Mapping]] = ..., RBAC_meta: _Optional[_Union[RBACMeta, _Mapping]] = ...) -> None: ...
1636+
1637+
class RestoreRBACMetaRequest(_message.Message):
1638+
__slots__ = ("base", "RBAC_meta")
1639+
BASE_FIELD_NUMBER: _ClassVar[int]
1640+
RBAC_META_FIELD_NUMBER: _ClassVar[int]
1641+
base: _common_pb2.MsgBase
1642+
RBAC_meta: RBACMeta
1643+
def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., RBAC_meta: _Optional[_Union[RBACMeta, _Mapping]] = ...) -> None: ...
1644+
16011645
class GetLoadingProgressRequest(_message.Message):
16021646
__slots__ = ("base", "collection_name", "partition_names", "db_name")
16031647
BASE_FIELD_NUMBER: _ClassVar[int]

Diff for: ‎pymilvus/grpc_gen/milvus_pb2_grpc.py

+86
Original file line numberDiff line numberDiff line change
@@ -471,6 +471,16 @@ def __init__(self, channel):
471471
request_serializer=milvus__pb2.ReplicateMessageRequest.SerializeToString,
472472
response_deserializer=milvus__pb2.ReplicateMessageResponse.FromString,
473473
_registered_method=True)
474+
self.BackupRBAC = channel.unary_unary(
475+
'/milvus.proto.milvus.MilvusService/BackupRBAC',
476+
request_serializer=milvus__pb2.BackupRBACMetaRequest.SerializeToString,
477+
response_deserializer=milvus__pb2.BackupRBACMetaResponse.FromString,
478+
_registered_method=True)
479+
self.RestoreRBAC = channel.unary_unary(
480+
'/milvus.proto.milvus.MilvusService/RestoreRBAC',
481+
request_serializer=milvus__pb2.RestoreRBACMetaRequest.SerializeToString,
482+
response_deserializer=common__pb2.Status.FromString,
483+
_registered_method=True)
474484

475485

476486
class MilvusServiceServicer(object):
@@ -999,6 +1009,18 @@ def ReplicateMessage(self, request, context):
9991009
context.set_details('Method not implemented!')
10001010
raise NotImplementedError('Method not implemented!')
10011011

1012+
def BackupRBAC(self, request, context):
1013+
"""Missing associated documentation comment in .proto file."""
1014+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
1015+
context.set_details('Method not implemented!')
1016+
raise NotImplementedError('Method not implemented!')
1017+
1018+
def RestoreRBAC(self, request, context):
1019+
"""Missing associated documentation comment in .proto file."""
1020+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
1021+
context.set_details('Method not implemented!')
1022+
raise NotImplementedError('Method not implemented!')
1023+
10021024

10031025
def add_MilvusServiceServicer_to_server(servicer, server):
10041026
rpc_method_handlers = {
@@ -1432,6 +1454,16 @@ def add_MilvusServiceServicer_to_server(servicer, server):
14321454
request_deserializer=milvus__pb2.ReplicateMessageRequest.FromString,
14331455
response_serializer=milvus__pb2.ReplicateMessageResponse.SerializeToString,
14341456
),
1457+
'BackupRBAC': grpc.unary_unary_rpc_method_handler(
1458+
servicer.BackupRBAC,
1459+
request_deserializer=milvus__pb2.BackupRBACMetaRequest.FromString,
1460+
response_serializer=milvus__pb2.BackupRBACMetaResponse.SerializeToString,
1461+
),
1462+
'RestoreRBAC': grpc.unary_unary_rpc_method_handler(
1463+
servicer.RestoreRBAC,
1464+
request_deserializer=milvus__pb2.RestoreRBACMetaRequest.FromString,
1465+
response_serializer=common__pb2.Status.SerializeToString,
1466+
),
14351467
}
14361468
generic_handler = grpc.method_handlers_generic_handler(
14371469
'milvus.proto.milvus.MilvusService', rpc_method_handlers)
@@ -3765,6 +3797,60 @@ def ReplicateMessage(request,
37653797
metadata,
37663798
_registered_method=True)
37673799

3800+
@staticmethod
3801+
def BackupRBAC(request,
3802+
target,
3803+
options=(),
3804+
channel_credentials=None,
3805+
call_credentials=None,
3806+
insecure=False,
3807+
compression=None,
3808+
wait_for_ready=None,
3809+
timeout=None,
3810+
metadata=None):
3811+
return grpc.experimental.unary_unary(
3812+
request,
3813+
target,
3814+
'/milvus.proto.milvus.MilvusService/BackupRBAC',
3815+
milvus__pb2.BackupRBACMetaRequest.SerializeToString,
3816+
milvus__pb2.BackupRBACMetaResponse.FromString,
3817+
options,
3818+
channel_credentials,
3819+
insecure,
3820+
call_credentials,
3821+
compression,
3822+
wait_for_ready,
3823+
timeout,
3824+
metadata,
3825+
_registered_method=True)
3826+
3827+
@staticmethod
3828+
def RestoreRBAC(request,
3829+
target,
3830+
options=(),
3831+
channel_credentials=None,
3832+
call_credentials=None,
3833+
insecure=False,
3834+
compression=None,
3835+
wait_for_ready=None,
3836+
timeout=None,
3837+
metadata=None):
3838+
return grpc.experimental.unary_unary(
3839+
request,
3840+
target,
3841+
'/milvus.proto.milvus.MilvusService/RestoreRBAC',
3842+
milvus__pb2.RestoreRBACMetaRequest.SerializeToString,
3843+
common__pb2.Status.FromString,
3844+
options,
3845+
channel_credentials,
3846+
insecure,
3847+
call_credentials,
3848+
compression,
3849+
wait_for_ready,
3850+
timeout,
3851+
metadata,
3852+
_registered_method=True)
3853+
37683854

37693855
class ProxyServiceStub(object):
37703856
"""Missing associated documentation comment in .proto file."""

0 commit comments

Comments
 (0)