oxen-io/oxen-pyoxenmq
1
0
Fork 0
mirror of https://github.com/oxen-io/oxen-pyoxenmq.git synced 2023-12-13 20:50:34 +01:00
Code Issues Projects Releases Wiki Activity

use py::function

Browse source
This commit is contained in:
Jeff 2020-09-24 11:59:36 -04:00
parent 6ea7941862
commit f568f911d1
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
pylokimq/lokimq.cpp
View file

@ -4,6 +4,7 @@
#include <lokimq/lokimq.h> #include <lokimq/lokimq.h>
#include <lokimq/address.h> #include <lokimq/address.h>
#include <pybind11/chrono.h> #include <pybind11/chrono.h>
#include <pybind11/functional.h>
#include <future> #include <future>
#include <memory> #include <memory>
@ -145,9 +146,9 @@ namespace lokimq
}) })
.def("add_request_command", .def("add_request_command",
[](LokiMQ &self, [](LokiMQ &self,
const std::string & category, std::string category,
const std::string & name, std::string name,
std::function<std::string(std::vector<py::bytes>)> handler) py::function handler)
{ {
self.add_request_command(category, name, self.add_request_command(category, name,
[handler](Message & msg) { [handler](Message & msg) {
@ -163,7 +164,8 @@ namespace lokimq
try try
{ {
result = handler(data); const auto obj = handler(data);
result = py::str(obj);
} }
catch(std::exception & ex) catch(std::exception & ex)
{ {
@ -175,9 +177,9 @@ namespace lokimq
}) })
.def("add_request_command_ex", .def("add_request_command_ex",
[](LokiMQ &self, [](LokiMQ &self,
const std::string & category, std::string category,
const std::string & name, std::string name,
std::function<std::string(std::vector<py::bytes>, std::string_view, ConnectionID)> handler) py::function handler)
{ {
self.add_request_command(category, name, self.add_request_command(category, name,
[handler](Message & msg) { [handler](Message & msg) {
@ -193,7 +195,8 @@ namespace lokimq
try try
{ {
result = handler(data, msg.remote, msg.conn); const auto obj = handler(data, msg);
result = py::str(obj);
} }
catch(std::exception & ex) catch(std::exception & ex)
{ {