mirror of
https://github.com/pocoproject/poco.git
synced 2025-04-26 01:36:18 +02:00
971a7cc670
* * Fix: MongoDB::OpMsgCursor did not handle zero batch size properly: cursor requests failed. * Improvement: Add emptyFirstBatch to indicate that the size of the first batch shall be zero for performance to get potential error ASAP from the server. * Poco::MongoDB: Some database commands do not need collection as an argument. An integer "1" is passed instead.
164 lines
4.2 KiB
C++
164 lines
4.2 KiB
C++
//
|
|
// OpMsgMessage.h
|
|
//
|
|
// Library: MongoDB
|
|
// Package: MongoDB
|
|
// Module: OpMsgMessage
|
|
//
|
|
// Definition of the OpMsgMessage class.
|
|
//
|
|
// Copyright (c) 2022, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#ifndef MongoDB_OpMsgMessage_INCLUDED
|
|
#define MongoDB_OpMsgMessage_INCLUDED
|
|
|
|
|
|
#include "Poco/MongoDB/MongoDB.h"
|
|
#include "Poco/MongoDB/Message.h"
|
|
#include "Poco/MongoDB/Document.h"
|
|
|
|
#include <string>
|
|
|
|
namespace Poco {
|
|
namespace MongoDB {
|
|
|
|
|
|
class MongoDB_API OpMsgMessage: public Message
|
|
/// This class represents a request/response (OP_MSG) to send requests and receive responses to/from MongoDB.
|
|
{
|
|
public:
|
|
|
|
// Constants for most often used MongoDB commands that can be sent using OP_MSG
|
|
// For complete list see: https://www.mongodb.com/docs/manual/reference/command/
|
|
|
|
// Query and write
|
|
static const std::string CMD_INSERT;
|
|
static const std::string CMD_DELETE;
|
|
static const std::string CMD_UPDATE;
|
|
static const std::string CMD_FIND;
|
|
static const std::string CMD_FIND_AND_MODIFY;
|
|
static const std::string CMD_GET_MORE;
|
|
|
|
// Aggregation
|
|
static const std::string CMD_AGGREGATE;
|
|
static const std::string CMD_COUNT;
|
|
static const std::string CMD_DISTINCT;
|
|
static const std::string CMD_MAP_REDUCE;
|
|
|
|
// Replication and administration
|
|
static const std::string CMD_HELLO;
|
|
static const std::string CMD_REPL_SET_GET_STATUS;
|
|
static const std::string CMD_REPL_SET_GET_CONFIG;
|
|
|
|
static const std::string CMD_CREATE;
|
|
static const std::string CMD_CREATE_INDEXES;
|
|
static const std::string CMD_DROP;
|
|
static const std::string CMD_DROP_DATABASE;
|
|
static const std::string CMD_KILL_CURSORS;
|
|
static const std::string CMD_LIST_DATABASES;
|
|
static const std::string CMD_LIST_INDEXES;
|
|
|
|
// Diagnostic
|
|
static const std::string CMD_BUILD_INFO;
|
|
static const std::string CMD_COLL_STATS;
|
|
static const std::string CMD_DB_STATS;
|
|
static const std::string CMD_HOST_INFO;
|
|
|
|
|
|
enum Flags : UInt32
|
|
{
|
|
MSG_FLAGS_DEFAULT = 0,
|
|
|
|
MSG_CHECKSUM_PRESENT = (1 << 0),
|
|
|
|
MSG_MORE_TO_COME = (1 << 1),
|
|
/// Sender will send another message and is not prepared for overlapping messages
|
|
|
|
MSG_EXHAUST_ALLOWED = (1 << 16)
|
|
/// Client is prepared for multiple replies (using the moreToCome bit) to this request
|
|
};
|
|
|
|
OpMsgMessage();
|
|
/// Creates an OpMsgMessage for response.
|
|
|
|
OpMsgMessage(const std::string& databaseName, const std::string& collectionName, UInt32 flags = MSG_FLAGS_DEFAULT);
|
|
/// Creates an OpMsgMessage for requests.
|
|
|
|
virtual ~OpMsgMessage();
|
|
|
|
const std::string& databaseName() const;
|
|
|
|
const std::string& collectionName() const;
|
|
|
|
void setCommandName(const std::string& command);
|
|
/// Sets the command name and clears the command document
|
|
|
|
void setCursor(Poco::Int64 cursorID, Poco::Int32 batchSize = -1);
|
|
/// Sets the command "getMore" for the cursor id with batch size (if it is not negative).
|
|
|
|
const std::string& commandName() const;
|
|
/// Current command name.
|
|
|
|
void setAcknowledgedRequest(bool ack);
|
|
/// Set false to create request that does not return response.
|
|
/// It has effect only for commands that write or delete documents.
|
|
/// Default is true (request returns acknowledge response).
|
|
|
|
bool acknowledgedRequest() const;
|
|
|
|
UInt32 flags() const;
|
|
|
|
Document& body();
|
|
/// Access to body document.
|
|
/// Additional query arguments shall be added after setting the command name.
|
|
|
|
const Document& body() const;
|
|
|
|
Document::Vector& documents();
|
|
/// Documents prepared for request or retrieved in response.
|
|
|
|
const Document::Vector& documents() const;
|
|
/// Documents prepared for request or retrieved in response.
|
|
|
|
bool responseOk() const;
|
|
/// Reads "ok" status from the response message.
|
|
|
|
void clear();
|
|
/// Clears the message.
|
|
|
|
void send(std::ostream& ostr);
|
|
/// Writes the request to stream.
|
|
|
|
void read(std::istream& istr);
|
|
/// Reads the response from the stream.
|
|
|
|
private:
|
|
|
|
enum PayloadType : UInt8
|
|
{
|
|
PAYLOAD_TYPE_0 = 0,
|
|
PAYLOAD_TYPE_1 = 1
|
|
};
|
|
|
|
std::string _databaseName;
|
|
std::string _collectionName;
|
|
UInt32 _flags { MSG_FLAGS_DEFAULT };
|
|
std::string _commandName;
|
|
bool _acknowledged {true};
|
|
|
|
Document _body;
|
|
Document::Vector _documents;
|
|
|
|
};
|
|
|
|
|
|
} } // namespace Poco::MongoDB
|
|
|
|
|
|
#endif // MongoDB_OpMsgMessage_INCLUDED
|