Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 41 additions & 1 deletion src/PythonWrapper.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,18 @@ int PySetProducerCompressLevel(void *producer, int level) {
int PySetProducerMaxMessageSize(void *producer, int size) {
return SetProducerMaxMessageSize((CProducer *)producer, size);
}

int PySetProducerLogPath(void *producer, const char *logPath) {
return SetProducerLogPath((CProducer *) producer, logPath);
}
int PySetProducerLogFileNumAndSize(void *producer, int fileNum, long fileSize) {
return SetProducerLogFileNumAndSize((CProducer *) producer, fileNum, fileSize);
}
int PySetProducerLogLevel(void *producer, CLogLevel level) {
return SetProducerLogLevel((CProducer *) producer, level);
}
int PySetProducerSendMsgTimeout(void *producer, int timeout) {
return SetProducerSendMsgTimeout((CProducer *) producer, timeout);
}

PySendResult PySendMessageSync(void *producer, void *msg) {
PySendResult ret;
Expand Down Expand Up @@ -337,6 +348,18 @@ int PySetPushConsumerMessageModel(void *consumer, CMessageModel messageModel) {
return SetPushConsumerMessageModel((CPushConsumer *) consumer, messageModel);
}

int PySetPushConsumerLogPath(void *consumer, const char *logPath) {
return SetPushConsumerLogPath((CPushConsumer *) consumer, logPath);
}

int PySetPushConsumerLogFileNumAndSize(void *consumer, int fileNum, long fileSize) {
return SetPushConsumerLogFileNumAndSize((CPushConsumer *) consumer, fileNum, fileSize);
}

int PySetPushConsumerLogLevel(void *consumer, CLogLevel level) {
return SetPushConsumerLogLevel((CPushConsumer *) consumer, level);
}

//push consumer
int PySetPullConsumerNameServerDomain(void *consumer, const char *domain) {
return SetPullConsumerNameServerDomain((CPullConsumer *) consumer, domain);
Expand Down Expand Up @@ -386,6 +409,15 @@ BOOST_PYTHON_MODULE (librocketmqclientpython) {
.value("BROADCASTING", BROADCASTING)
.value("CLUSTERING", CLUSTERING);

enum_<CLogLevel>("CLogLevel")
.value("E_LOG_LEVEL_FATAL", E_LOG_LEVEL_FATAL)
.value("E_LOG_LEVEL_ERROR", E_LOG_LEVEL_ERROR)
.value("E_LOG_LEVEL_WARN", E_LOG_LEVEL_WARN)
.value("E_LOG_LEVEL_INFO", E_LOG_LEVEL_INFO)
.value("E_LOG_LEVEL_DEBUG", E_LOG_LEVEL_DEBUG)
.value("E_LOG_LEVEL_TRACE", E_LOG_LEVEL_TRACE)
.value("E_LOG_LEVEL_LEVEL_NUM", E_LOG_LEVEL_LEVEL_NUM);


//For Message
def("CreateMessage", PyCreateMessage, return_value_policy<return_opaque_pointer>());
Expand Down Expand Up @@ -422,6 +454,11 @@ BOOST_PYTHON_MODULE (librocketmqclientpython) {
def("SetProducerSessionCredentials", PySetProducerSessionCredentials);
def("SetProducerCompressLevel", PySetProducerCompressLevel);
def("SetProducerMaxMessageSize", PySetProducerMaxMessageSize);
def("SetProducerSendMsgTimeout", PySetProducerSendMsgTimeout);

def("SetProducerLogPath", PySetProducerLogPath);
def("SetProducerLogFileNumAndSize", PySetProducerLogFileNumAndSize);
def("SetProducerLogLevel", PySetProducerLogLevel);

def("SendMessageSync", PySendMessageSync);
def("SendMessageAsync", PySendMessageAsync);
Expand All @@ -445,6 +482,9 @@ BOOST_PYTHON_MODULE (librocketmqclientpython) {
def("Subscribe", PySubscribe);
def("RegisterMessageCallback", PyRegisterMessageCallback);
def("RegisterMessageCallbackOrderly", PyRegisterMessageCallbackOrderly);
def("SetPushConsumerLogPath", PySetPushConsumerLogPath);
def("SetPushConsumerLogFileNumAndSize", PySetPushConsumerLogFileNumAndSize);
def("SetPushConsumerLogLevel", PySetPushConsumerLogLevel);

//pull consumer
def("SetPullConsumerNameServerDomain", PySetPullConsumerNameServerDomain);
Expand Down
10 changes: 8 additions & 2 deletions src/PythonWrapper.h
Original file line number Diff line number Diff line change
Expand Up @@ -115,6 +115,10 @@ int PySetProducerInstanceName(void *producer, const char *instanceName);
int PySetProducerSessionCredentials(void *producer, const char *accessKey, const char *secretKey, const char *channel);
int PySetProducerCompressLevel(void *producer, int level);
int PySetProducerMaxMessageSize(void *producer, int size);
int PySetProducerLogPath(void *producer, const char *logPath);
int PySetProducerLogFileNumAndSize(void *producer, int fileNum, long fileSize);
int PySetProducerLogLevel(void *producer, CLogLevel level);
int PySetProducerSendMsgTimeout(void *producer, int timeout);

PySendResult PySendMessageSync(void *producer, void *msg);
int PySendMessageOneway(void *producer, void *msg);
Expand Down Expand Up @@ -146,9 +150,11 @@ int PythonMessageCallBackInner(CPushConsumer *consumer, CMessageExt *msg);
int PySetPushConsumerThreadCount(void *consumer, int threadCount);
int PySetPushConsumerMessageBatchMaxSize(void *consumer, int batchSize);
int PySetPushConsumerInstanceName(void *consumer, const char *instanceName);
int PySetPushConsumerSessionCredentials(void *consumer, const char *accessKey, const char *secretKey,
const char *channel);
int PySetPushConsumerSessionCredentials(void *consumer, const char *accessKey, const char *secretKey, const char *channel);
int PySetPushConsumerMessageModel(void *consumer, CMessageModel messageModel);
int PySetPushConsumerLogPath(void *consumer, const char *logPath);
int PySetPushConsumerLogFileNumAndSize(void *consumer, int fileNum, long fileSize);
int PySetPushConsumerLogLevel(void *consumer, CLogLevel level);

//push consumer
int PySetPullConsumerNameServerDomain(void *consumer, const char *domain);
Expand Down
1 change: 1 addition & 0 deletions test/TestConsumeMessages.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,7 @@ def init_consumer(_group, _topic, _tag):
consumer = CreatePushConsumer(_group)
SetPushConsumerNameServerAddress(consumer, name_srv)
SetPushConsumerThreadCount(consumer, 1)
SetPushConsumerLogLevel(consumer, CLogLevel.E_LOG_LEVEL_INFO)
SetPushConsumerMessageModel(consumer, CMessageModel.CLUSTERING)
Subscribe(consumer, _topic, _tag)
RegisterMessageCallback(consumer, consumer_message, None)
Expand Down
1 change: 1 addition & 0 deletions test/TestSendMessages.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@

def init_producer():
producer = CreateProducer('TestProducer')
SetProducerLogLevel(producer, CLogLevel.E_LOG_LEVEL_INFO)
SetProducerNameServerAddress(producer, name_srv)
StartProducer(producer)
return producer
Expand Down