milvus/proxy/src/message_client/PartitionPolicy.h
xige-16 d58f6ff817 Support write node send key2seg msg
Signed-off-by: xige-16 <xi.ge@zilliz.com>
2020-09-06 19:39:35 +08:00

17 lines
346 B
C++

#pragma once
#include <istream>
#include "pulsar/MessageRoutingPolicy.h"
#include "pulsar/Message.h"
#include "pulsar/TopicMetadata.h"
namespace milvus {
namespace message_client {
class PartitionPolicy : public pulsar::MessageRoutingPolicy {
int getPartition(const pulsar::Message& msg, const pulsar::TopicMetadata& topicMetadata);
};
}
}