Skip to content

Commit

Permalink
Update proto again
Browse files Browse the repository at this point in the history
Signed-off-by: yangxuan <[email protected]>
  • Loading branch information
XuanYang-cn committed Oct 24, 2023
1 parent 73ab019 commit 2ba2703
Show file tree
Hide file tree
Showing 6 changed files with 461 additions and 384 deletions.
94 changes: 48 additions & 46 deletions pymilvus/grpc_gen/common_pb2.py

Large diffs are not rendered by default.

14 changes: 12 additions & 2 deletions pymilvus/grpc_gen/common_pb2.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -599,7 +599,7 @@ class Address(_message.Message):
def __init__(self, ip: _Optional[str] = ..., port: _Optional[int] = ...) -> None: ...

class MsgBase(_message.Message):
__slots__ = ["msg_type", "msgID", "timestamp", "sourceID", "targetID", "properties"]
__slots__ = ["msg_type", "msgID", "timestamp", "sourceID", "targetID", "properties", "replicateInfo"]
class PropertiesEntry(_message.Message):
__slots__ = ["key", "value"]
KEY_FIELD_NUMBER: _ClassVar[int]
Expand All @@ -613,13 +613,23 @@ class MsgBase(_message.Message):
SOURCEID_FIELD_NUMBER: _ClassVar[int]
TARGETID_FIELD_NUMBER: _ClassVar[int]
PROPERTIES_FIELD_NUMBER: _ClassVar[int]
REPLICATEINFO_FIELD_NUMBER: _ClassVar[int]
msg_type: MsgType
msgID: int
timestamp: int
sourceID: int
targetID: int
properties: _containers.ScalarMap[str, str]
def __init__(self, msg_type: _Optional[_Union[MsgType, str]] = ..., msgID: _Optional[int] = ..., timestamp: _Optional[int] = ..., sourceID: _Optional[int] = ..., targetID: _Optional[int] = ..., properties: _Optional[_Mapping[str, str]] = ...) -> None: ...
replicateInfo: ReplicateInfo
def __init__(self, msg_type: _Optional[_Union[MsgType, str]] = ..., msgID: _Optional[int] = ..., timestamp: _Optional[int] = ..., sourceID: _Optional[int] = ..., targetID: _Optional[int] = ..., properties: _Optional[_Mapping[str, str]] = ..., replicateInfo: _Optional[_Union[ReplicateInfo, _Mapping]] = ...) -> None: ...

class ReplicateInfo(_message.Message):
__slots__ = ["isReplicate", "msgTimestamp"]
ISREPLICATE_FIELD_NUMBER: _ClassVar[int]
MSGTIMESTAMP_FIELD_NUMBER: _ClassVar[int]
isReplicate: bool
msgTimestamp: int
def __init__(self, isReplicate: bool = ..., msgTimestamp: _Optional[int] = ...) -> None: ...

class MsgHeader(_message.Message):
__slots__ = ["base"]
Expand Down
2 changes: 1 addition & 1 deletion pymilvus/grpc_gen/milvus-proto
675 changes: 340 additions & 335 deletions pymilvus/grpc_gen/milvus_pb2.py

Large diffs are not rendered by default.

27 changes: 27 additions & 0 deletions pymilvus/grpc_gen/milvus_pb2.pyi
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from . import common_pb2 as _common_pb2
from . import schema_pb2 as _schema_pb2
from . import feder_pb2 as _feder_pb2
import msg_pb2 as _msg_pb2
from google.protobuf import descriptor_pb2 as _descriptor_pb2
from google.protobuf.internal import containers as _containers
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
Expand Down Expand Up @@ -1795,3 +1796,29 @@ class ListDatabasesResponse(_message.Message):
db_names: _containers.RepeatedScalarFieldContainer[str]
created_timestamp: _containers.RepeatedScalarFieldContainer[int]
def __init__(self, status: _Optional[_Union[_common_pb2.Status, _Mapping]] = ..., db_names: _Optional[_Iterable[str]] = ..., created_timestamp: _Optional[_Iterable[int]] = ...) -> None: ...

class ReplicateMessageRequest(_message.Message):
__slots__ = ["base", "channel_name", "BeginTs", "EndTs", "Msgs", "StartPositions", "EndPositions"]
BASE_FIELD_NUMBER: _ClassVar[int]
CHANNEL_NAME_FIELD_NUMBER: _ClassVar[int]
BEGINTS_FIELD_NUMBER: _ClassVar[int]
ENDTS_FIELD_NUMBER: _ClassVar[int]
MSGS_FIELD_NUMBER: _ClassVar[int]
STARTPOSITIONS_FIELD_NUMBER: _ClassVar[int]
ENDPOSITIONS_FIELD_NUMBER: _ClassVar[int]
base: _common_pb2.MsgBase
channel_name: str
BeginTs: int
EndTs: int
Msgs: _containers.RepeatedScalarFieldContainer[bytes]
StartPositions: _containers.RepeatedCompositeFieldContainer[_msg_pb2.MsgPosition]
EndPositions: _containers.RepeatedCompositeFieldContainer[_msg_pb2.MsgPosition]
def __init__(self, base: _Optional[_Union[_common_pb2.MsgBase, _Mapping]] = ..., channel_name: _Optional[str] = ..., BeginTs: _Optional[int] = ..., EndTs: _Optional[int] = ..., Msgs: _Optional[_Iterable[bytes]] = ..., StartPositions: _Optional[_Iterable[_Union[_msg_pb2.MsgPosition, _Mapping]]] = ..., EndPositions: _Optional[_Iterable[_Union[_msg_pb2.MsgPosition, _Mapping]]] = ...) -> None: ...

class ReplicateMessageResponse(_message.Message):
__slots__ = ["status", "position"]
STATUS_FIELD_NUMBER: _ClassVar[int]
POSITION_FIELD_NUMBER: _ClassVar[int]
status: _common_pb2.Status
position: str
def __init__(self, status: _Optional[_Union[_common_pb2.Status, _Mapping]] = ..., position: _Optional[str] = ...) -> None: ...
33 changes: 33 additions & 0 deletions pymilvus/grpc_gen/milvus_pb2_grpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -416,6 +416,11 @@ def __init__(self, channel):
request_serializer=milvus__pb2.ListDatabasesRequest.SerializeToString,
response_deserializer=milvus__pb2.ListDatabasesResponse.FromString,
)
self.ReplicateMessage = channel.unary_unary(
'/milvus.proto.milvus.MilvusService/ReplicateMessage',
request_serializer=milvus__pb2.ReplicateMessageRequest.SerializeToString,
response_deserializer=milvus__pb2.ReplicateMessageResponse.FromString,
)


class MilvusServiceServicer(object):
Expand Down Expand Up @@ -908,6 +913,12 @@ def ListDatabases(self, request, context):
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')

def ReplicateMessage(self, request, context):
"""Missing associated documentation comment in .proto file."""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')


def add_MilvusServiceServicer_to_server(servicer, server):
rpc_method_handlers = {
Expand Down Expand Up @@ -1311,6 +1322,11 @@ def add_MilvusServiceServicer_to_server(servicer, server):
request_deserializer=milvus__pb2.ListDatabasesRequest.FromString,
response_serializer=milvus__pb2.ListDatabasesResponse.SerializeToString,
),
'ReplicateMessage': grpc.unary_unary_rpc_method_handler(
servicer.ReplicateMessage,
request_deserializer=milvus__pb2.ReplicateMessageRequest.FromString,
response_serializer=milvus__pb2.ReplicateMessageResponse.SerializeToString,
),
}
generic_handler = grpc.method_handlers_generic_handler(
'milvus.proto.milvus.MilvusService', rpc_method_handlers)
Expand Down Expand Up @@ -2681,6 +2697,23 @@ def ListDatabases(request,
options, channel_credentials,
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

@staticmethod
def ReplicateMessage(request,
target,
options=(),
channel_credentials=None,
call_credentials=None,
insecure=False,
compression=None,
wait_for_ready=None,
timeout=None,
metadata=None):
return grpc.experimental.unary_unary(request, target, '/milvus.proto.milvus.MilvusService/ReplicateMessage',
milvus__pb2.ReplicateMessageRequest.SerializeToString,
milvus__pb2.ReplicateMessageResponse.FromString,
options, channel_credentials,
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)


class ProxyServiceStub(object):
"""Missing associated documentation comment in .proto file."""
Expand Down

0 comments on commit 2ba2703

Please sign in to comment.