Static execute in JSON interface.
GitOrigin-RevId: 58748f011a6def1d81e11063e9a15f5a8750174e
This commit is contained in:
parent
4378f96948
commit
abbb1a42a7
@ -79,7 +79,7 @@ def td_receive():
|
|||||||
|
|
||||||
def td_execute(query):
|
def td_execute(query):
|
||||||
query = json.dumps(query).encode('utf-8')
|
query = json.dumps(query).encode('utf-8')
|
||||||
result = td_json_client_execute(client, query)
|
result = td_json_client_execute(None, query)
|
||||||
if result:
|
if result:
|
||||||
result = json.loads(result.decode('utf-8'))
|
result = json.loads(result.decode('utf-8'))
|
||||||
return result
|
return result
|
||||||
|
@ -14,55 +14,50 @@
|
|||||||
#include "td/utils/format.h"
|
#include "td/utils/format.h"
|
||||||
#include "td/utils/JsonBuilder.h"
|
#include "td/utils/JsonBuilder.h"
|
||||||
#include "td/utils/logging.h"
|
#include "td/utils/logging.h"
|
||||||
|
#include "td/utils/port/thread_local.h"
|
||||||
#include "td/utils/Status.h"
|
#include "td/utils/Status.h"
|
||||||
|
|
||||||
|
#include <utility>
|
||||||
|
|
||||||
namespace td {
|
namespace td {
|
||||||
|
|
||||||
Result<Client::Request> ClientJson::to_request(Slice request) {
|
static Result<std::pair<td_api::object_ptr<td_api::Function>, string>> to_request(Slice request) {
|
||||||
auto request_str = request.str();
|
auto request_str = request.str();
|
||||||
TRY_RESULT(json_value, json_decode(request_str));
|
TRY_RESULT(json_value, json_decode(request_str));
|
||||||
if (json_value.type() != JsonValue::Type::Object) {
|
if (json_value.type() != JsonValue::Type::Object) {
|
||||||
return Status::Error("Expected an Object");
|
return Status::Error("Expected an Object");
|
||||||
}
|
}
|
||||||
std::uint64_t extra_id = extra_id_.fetch_add(1, std::memory_order_relaxed);
|
|
||||||
|
string extra;
|
||||||
if (has_json_object_field(json_value.get_object(), "@extra")) {
|
if (has_json_object_field(json_value.get_object(), "@extra")) {
|
||||||
std::lock_guard<std::mutex> guard(mutex_);
|
extra = json_encode<string>(
|
||||||
extra_[extra_id] = json_encode<string>(
|
|
||||||
get_json_object_field(json_value.get_object(), "@extra", JsonValue::Type::Null).move_as_ok());
|
get_json_object_field(json_value.get_object(), "@extra", JsonValue::Type::Null).move_as_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
td_api::object_ptr<td_api::Function> func;
|
td_api::object_ptr<td_api::Function> func;
|
||||||
TRY_STATUS(from_json(func, json_value));
|
TRY_STATUS(from_json(func, json_value));
|
||||||
return Client::Request{extra_id, std::move(func)};
|
return std::make_pair(std::move(func), extra);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string ClientJson::from_response(Client::Response response) {
|
static std::string from_response(const td_api::Object &object, const string &extra) {
|
||||||
auto str = json_encode<string>(ToJson(*response.object));
|
auto str = json_encode<string>(ToJson(object));
|
||||||
CHECK(!str.empty() && str.back() == '}');
|
CHECK(!str.empty() && str.back() == '}');
|
||||||
std::string extra;
|
|
||||||
if (response.id != 0) {
|
|
||||||
std::lock_guard<std::mutex> guard(mutex_);
|
|
||||||
auto it = extra_.find(response.id);
|
|
||||||
if (it != extra_.end()) {
|
|
||||||
extra = std::move(it->second);
|
|
||||||
extra_.erase(it);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (!extra.empty()) {
|
if (!extra.empty()) {
|
||||||
str.pop_back();
|
str.pop_back();
|
||||||
str.reserve(str.size() + 10 + extra.size());
|
str.reserve(str.size() + 11 + extra.size());
|
||||||
str += ",\"@extra\":";
|
str += ",\"@extra\":";
|
||||||
str += extra;
|
str += extra;
|
||||||
str += "}";
|
str += '}';
|
||||||
}
|
}
|
||||||
return str;
|
return str;
|
||||||
}
|
}
|
||||||
|
|
||||||
TD_THREAD_LOCAL std::string *ClientJson::current_output_;
|
static TD_THREAD_LOCAL std::string *current_output;
|
||||||
CSlice ClientJson::store_string(std::string str) {
|
|
||||||
init_thread_local<std::string>(ClientJson::current_output_);
|
static CSlice store_string(std::string str) {
|
||||||
*current_output_ = std::move(str);
|
init_thread_local<std::string>(current_output);
|
||||||
return *current_output_;
|
*current_output = std::move(str);
|
||||||
|
return *current_output;
|
||||||
}
|
}
|
||||||
|
|
||||||
void ClientJson::send(Slice request) {
|
void ClientJson::send(Slice request) {
|
||||||
@ -72,7 +67,12 @@ void ClientJson::send(Slice request) {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
client_.send(r_request.move_as_ok());
|
std::uint64_t extra_id = extra_id_.fetch_add(1, std::memory_order_relaxed);
|
||||||
|
if (!r_request.ok_ref().second.empty()) {
|
||||||
|
std::lock_guard<std::mutex> guard(mutex_);
|
||||||
|
extra_[extra_id] = std::move(r_request.ok_ref().second);
|
||||||
|
}
|
||||||
|
client_.send(Client::Request{extra_id, std::move(r_request.ok_ref().first)});
|
||||||
}
|
}
|
||||||
|
|
||||||
CSlice ClientJson::receive(double timeout) {
|
CSlice ClientJson::receive(double timeout) {
|
||||||
@ -80,7 +80,17 @@ CSlice ClientJson::receive(double timeout) {
|
|||||||
if (!response.object) {
|
if (!response.object) {
|
||||||
return {};
|
return {};
|
||||||
}
|
}
|
||||||
return store_string(from_response(std::move(response)));
|
|
||||||
|
std::string extra;
|
||||||
|
if (response.id != 0) {
|
||||||
|
std::lock_guard<std::mutex> guard(mutex_);
|
||||||
|
auto it = extra_.find(response.id);
|
||||||
|
if (it != extra_.end()) {
|
||||||
|
extra = std::move(it->second);
|
||||||
|
extra_.erase(it);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return store_string(from_response(*response.object, extra));
|
||||||
}
|
}
|
||||||
|
|
||||||
CSlice ClientJson::execute(Slice request) {
|
CSlice ClientJson::execute(Slice request) {
|
||||||
@ -90,7 +100,8 @@ CSlice ClientJson::execute(Slice request) {
|
|||||||
return {};
|
return {};
|
||||||
}
|
}
|
||||||
|
|
||||||
return store_string(from_response(Client::execute(r_request.move_as_ok())));
|
return store_string(from_response(*Client::execute(Client::Request{0, std::move(r_request.ok_ref().first)}).object,
|
||||||
|
r_request.ok().second));
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace td
|
} // namespace td
|
||||||
|
@ -8,16 +8,13 @@
|
|||||||
|
|
||||||
#include "td/telegram/Client.h"
|
#include "td/telegram/Client.h"
|
||||||
|
|
||||||
#include "td/utils/port/thread_local.h"
|
|
||||||
#include "td/utils/Slice.h"
|
#include "td/utils/Slice.h"
|
||||||
#include "td/utils/Status.h"
|
|
||||||
|
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
namespace td {
|
namespace td {
|
||||||
|
|
||||||
@ -27,18 +24,13 @@ class ClientJson final {
|
|||||||
|
|
||||||
CSlice receive(double timeout);
|
CSlice receive(double timeout);
|
||||||
|
|
||||||
CSlice execute(Slice request);
|
static CSlice execute(Slice request);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Client client_;
|
Client client_;
|
||||||
std::mutex mutex_; // for extra_
|
std::mutex mutex_; // for extra_
|
||||||
std::unordered_map<std::int64_t, std::string> extra_;
|
std::unordered_map<std::int64_t, std::string> extra_;
|
||||||
std::atomic<std::uint64_t> extra_id_{1};
|
std::atomic<std::uint64_t> extra_id_{1};
|
||||||
static TD_THREAD_LOCAL std::string *current_output_;
|
|
||||||
|
|
||||||
CSlice store_string(std::string str);
|
|
||||||
|
|
||||||
Result<Client::Request> to_request(Slice request);
|
|
||||||
std::string from_response(Client::Response response);
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace td
|
} // namespace td
|
||||||
|
@ -36,7 +36,7 @@ const char *td_json_client_receive(void *client, double timeout) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const char *td_json_client_execute(void *client, const char *request) {
|
const char *td_json_client_execute(void *client, const char *request) {
|
||||||
auto slice = static_cast<td::ClientJson *>(client)->execute(td::Slice(request == nullptr ? "" : request));
|
auto slice = td::ClientJson::execute(td::Slice(request == nullptr ? "" : request));
|
||||||
if (slice.empty()) {
|
if (slice.empty()) {
|
||||||
return nullptr;
|
return nullptr;
|
||||||
} else {
|
} else {
|
||||||
|
@ -82,7 +82,7 @@ TDJSON_EXPORT const char *td_json_client_receive(void *client, double timeout);
|
|||||||
* Only a few requests can be executed synchronously.
|
* Only a few requests can be executed synchronously.
|
||||||
* Returned pointer will be deallocated by TDLib during next call to td_json_client_receive or td_json_client_execute
|
* Returned pointer will be deallocated by TDLib during next call to td_json_client_receive or td_json_client_execute
|
||||||
* in the same thread, so it can't be used after that.
|
* in the same thread, so it can't be used after that.
|
||||||
* \param[in] client The client.
|
* \param[in] client The client. Currently ignored for all requests, so NULL can be passed.
|
||||||
* \param[in] request JSON-serialized null-terminated request to TDLib.
|
* \param[in] request JSON-serialized null-terminated request to TDLib.
|
||||||
* \return JSON-serialized null-terminated request response. May be NULL if the request can't be parsed.
|
* \return JSON-serialized null-terminated request response. May be NULL if the request can't be parsed.
|
||||||
*/
|
*/
|
||||||
|
Reference in New Issue
Block a user