generic-library/poco
1
0
Fork 0
mirror of https://github.com/pocoproject/poco.git synced 2025-05-02 15:41:36 +02:00
Code Issues Projects Releases Wiki Activity

Add getInteger

Browse Source
This commit is contained in:
fbraem 2015-12-12 17:15:55 +01:00
parent cfdf3e4134
commit edc8d2abff
4 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
MongoDB/include/Poco/MongoDB/Document.h
View File

@ -153,6 +153,12 @@ public:
/// Returns the element with the given name. /// Returns the element with the given name.
/// An empty element will be returned when the element is not found. /// An empty element will be returned when the element is not found.
Int64 getInteger(const std::string& name) const;
/// Returns an integer. Useful when MongoDB returns int32, int64
/// or double for a number (count for example). This method will always
/// return an Int64. When the element is not found, a
/// NotFoundException will be thrown.
template<typename T> template<typename T>
bool isType(const std::string& name) const bool isType(const std::string& name) const
/// Returns true when the type of the element equals the TypeId of ElementTrait /// Returns true when the type of the element equals the TypeId of ElementTrait

11
MongoDB/src/Database.cpp
View File

@ -42,16 +42,7 @@ Int64 Database::count(Connection& connection, const std::string& collectionName)
if ( response.documents().size() > 0 ) if ( response.documents().size() > 0 )
{ {
Poco::MongoDB::Document::Ptr doc = response.documents()[0]; Poco::MongoDB::Document::Ptr doc = response.documents()[0];
return doc->getInteger("n");
if (doc->isType<double>("n")) {
return static_cast<Int64>(doc->get<double>("n"));
}
else if (doc->isType<Int32>("n")) {
return doc->get<Int32>("n");
}
else if (doc->isType<Int64>("n")) {
return doc->get<Int64>("n");
}
} }
return -1; return -1;

24
MongoDB/src/Document.cpp
View File

@ -52,6 +52,28 @@ Element::Ptr Document::get(const std::string& name) const
return element; return element;
} }
Int64 Document::getInteger(const std::string& name) const
{
Element::Ptr element = get(name);
if ( element.isNull() ) throw NotFoundException(name);
if ( ElementTraits<double>::TypeId == element->type() )
{
ConcreteElement<double>* concrete = dynamic_cast<ConcreteElement<double>* >(element.get());
if ( concrete != NULL ) return concrete->value();
}
else if ( ElementTraits<Int32>::TypeId == element->type() )
{
ConcreteElement<Int32>* concrete = dynamic_cast<ConcreteElement<Int32>* >(element.get());
if ( concrete != NULL ) return concrete->value();
}
else if ( ElementTraits<Int64>::TypeId == element->type() )
{
ConcreteElement<Int64>* concrete = dynamic_cast<ConcreteElement<Int64>* >(element.get());
if ( concrete != NULL ) return concrete->value();
}
throw BadCastException("Invalid type mismatch!");
}
void Document::read(BinaryReader& reader) void Document::read(BinaryReader& reader)
{ {
@ -184,7 +206,7 @@ void Document::write(BinaryWriter& writer)
element->write(tempWriter); element->write(tempWriter);
} }
tempWriter.flush(); tempWriter.flush();
Poco::Int32 len = static_cast<Poco::Int32>(5 + sstream.tellp()); /* 5 = sizeof(len) + 0-byte */ Poco::Int32 len = static_cast<Poco::Int32>(5 + sstream.tellp()); /* 5 = sizeof(len) + 0-byte */
writer << len; writer << len;
writer.writeRaw(sstream.str()); writer.writeRaw(sstream.str());

10
MongoDB/testsuite/src/MongoDBTest.cpp
View File

@ -174,8 +174,8 @@ void MongoDBTest::testCountCommand()
{ {
Poco::MongoDB::Document::Ptr doc = response.documents()[0]; Poco::MongoDB::Document::Ptr doc = response.documents()[0];
std::cout << doc->toString() << std::endl; std::cout << doc->toString() << std::endl;
double count = doc->get<double>("n");
assert(count == 1); assert(doc->getInteger("n") == 1);
} }
else else
{ {
@ -195,8 +195,7 @@ void MongoDBTest::testDBCountCommand()
if ( response.documents().size() > 0 ) if ( response.documents().size() > 0 )
{ {
Poco::MongoDB::Document::Ptr doc = response.documents()[0]; Poco::MongoDB::Document::Ptr doc = response.documents()[0];
double count = doc->get<double>("n"); assert(doc->getInteger("n") == 1);
assert(count == 1);
} }
else else
{ {
@ -314,8 +313,7 @@ void MongoDBTest::testConnectionPool()
if ( response.documents().size() > 0 ) if ( response.documents().size() > 0 )
{ {
Poco::MongoDB::Document::Ptr doc = response.documents()[0]; Poco::MongoDB::Document::Ptr doc = response.documents()[0];
double count = doc->get<double>("n"); assert(doc->getInteger("n") == 1);
assert(count == 1);
} }
else else
{ {