mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-02 12:32:04 +00:00
159 lines
3.5 KiB
C++
159 lines
3.5 KiB
C++
//
|
|
// Binary.h
|
|
//
|
|
// Library: MongoDB
|
|
// Package: MongoDB
|
|
// Module: Binary
|
|
//
|
|
// Definition of the Binary class.
|
|
//
|
|
// Copyright (c) 2012, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#ifndef MongoDB_Binary_INCLUDED
|
|
#define MongoDB_Binary_INCLUDED
|
|
|
|
|
|
#include <sstream>
|
|
#include "Poco/Base64Encoder.h"
|
|
#include "Poco/Buffer.h"
|
|
#include "Poco/MemoryStream.h"
|
|
#include "Poco/MongoDB/Element.h"
|
|
#include "Poco/MongoDB/MongoDB.h"
|
|
#include "Poco/StreamCopier.h"
|
|
#include "Poco/UUID.h"
|
|
|
|
|
|
namespace Poco
|
|
{
|
|
namespace MongoDB
|
|
{
|
|
|
|
|
|
class MongoDB_API Binary
|
|
/// Implements BSON Binary.
|
|
///
|
|
/// A Binary stores its data in a Poco::Buffer<unsigned char>.
|
|
{
|
|
public:
|
|
typedef SharedPtr<Binary> Ptr;
|
|
|
|
Binary();
|
|
/// Creates an empty Binary with subtype 0.
|
|
|
|
Binary(Poco::Int32 size, unsigned char subtype);
|
|
/// Creates a Binary with a buffer of the given size and the given subtype.
|
|
|
|
Binary(const UUID & uuid);
|
|
/// Creates a Binary containing an UUID.
|
|
|
|
Binary(const std::string & data, unsigned char subtype = 0);
|
|
/// Creates a Binary with the contents of the given string and the given subtype.
|
|
|
|
Binary(const void * data, Poco::Int32 size, unsigned char subtype = 0);
|
|
/// Creates a Binary with the contents of the given buffer and the given subtype.
|
|
|
|
virtual ~Binary();
|
|
/// Destroys the Binary.
|
|
|
|
Buffer<unsigned char> & buffer();
|
|
/// Returns a reference to the internal buffer
|
|
|
|
unsigned char subtype() const;
|
|
/// Returns the subtype.
|
|
|
|
void subtype(unsigned char type);
|
|
/// Sets the subtype.
|
|
|
|
std::string toString(int indent = 0) const;
|
|
/// Returns the contents of the Binary as Base64-encoded string.
|
|
|
|
std::string toRawString() const;
|
|
/// Returns the raw content of the Binary as a string.
|
|
|
|
UUID uuid() const;
|
|
/// Returns the UUID when the binary subtype is 0x04.
|
|
/// Otherwise, throws a Poco::BadCastException.
|
|
|
|
private:
|
|
Buffer<unsigned char> _buffer;
|
|
unsigned char _subtype;
|
|
};
|
|
|
|
|
|
//
|
|
// inlines
|
|
//
|
|
inline unsigned char Binary::subtype() const
|
|
{
|
|
return _subtype;
|
|
}
|
|
|
|
|
|
inline void Binary::subtype(unsigned char type)
|
|
{
|
|
_subtype = type;
|
|
}
|
|
|
|
|
|
inline Buffer<unsigned char> & Binary::buffer()
|
|
{
|
|
return _buffer;
|
|
}
|
|
|
|
|
|
inline std::string Binary::toRawString() const
|
|
{
|
|
return std::string(reinterpret_cast<const char *>(_buffer.begin()), _buffer.size());
|
|
}
|
|
|
|
|
|
// BSON Embedded Document
|
|
// spec: binary
|
|
template <>
|
|
struct ElementTraits<Binary::Ptr>
|
|
{
|
|
enum
|
|
{
|
|
TypeId = 0x05
|
|
};
|
|
|
|
static std::string toString(const Binary::Ptr & value, int indent = 0) { return value.isNull() ? "" : value->toString(); }
|
|
};
|
|
|
|
|
|
template <>
|
|
inline void BSONReader::read<Binary::Ptr>(Binary::Ptr & to)
|
|
{
|
|
Poco::Int32 size;
|
|
_reader >> size;
|
|
|
|
to->buffer().resize(size);
|
|
|
|
unsigned char subtype;
|
|
_reader >> subtype;
|
|
to->subtype(subtype);
|
|
|
|
_reader.readRaw((char *)to->buffer().begin(), size);
|
|
}
|
|
|
|
|
|
template <>
|
|
inline void BSONWriter::write<Binary::Ptr>(Binary::Ptr & from)
|
|
{
|
|
_writer << (Poco::Int32)from->buffer().size();
|
|
_writer << from->subtype();
|
|
_writer.writeRaw((char *)from->buffer().begin(), from->buffer().size());
|
|
}
|
|
|
|
|
|
}
|
|
} // namespace Poco::MongoDB
|
|
|
|
|
|
#endif // MongoDB_Binary_INCLUDED
|