generic-library/poco
1
0
Fork 0
mirror of https://github.com/pocoproject/poco.git synced 2025-10-30 21:50:47 +01:00
Code Issues Projects Releases Wiki Activity

Database::count() can parse different types of 'n' attribute in result.

Browse Source
This commit is contained in:
fbraem
2015-06-03 20:34:05 +02:00
parent 0a68bbfbea
commit aaabc3c9e9
3 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
MongoDB/src/Database.cpp
View File

@@ -23,7 +23,7 @@ namespace Poco {
namespace MongoDB {
Database::Database( const std::string& db) : _dbname(db)
Database::Database(const std::string& db) : _dbname(db)
{
}
@@ -32,7 +32,7 @@ Database::~Database()
}
int Database::count(Connection& connection, const std::string& collectionName) const
Int64 Database::count(Connection& connection, const std::string& collectionName) const
{
Poco::SharedPtr<Poco::MongoDB::QueryRequest> countRequest = createCountRequest(collectionName);
@@ -42,7 +42,16 @@ int Database::count(Connection& connection, const std::string& collectionName) c
if ( response.documents().size() > 0 )
{
Poco::MongoDB::Document::Ptr doc = response.documents()[0];
return doc->get<int>("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;