Skip to content

Commit

Permalink
Add query example
Browse files Browse the repository at this point in the history
  • Loading branch information
avsej committed Jun 26, 2024
1 parent 54cce4b commit a2e6798
Show file tree
Hide file tree
Showing 2 changed files with 169 additions and 0 deletions.
4 changes: 4 additions & 0 deletions examples/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,10 @@ add_executable(minimal minimal.cpp)
target_link_libraries(minimal
PRIVATE couchbase_cxx_client::couchbase_cxx_client taocpp::json)

add_executable(minimal_query minimal_query.cpp)
target_link_libraries(minimal_query
PRIVATE couchbase_cxx_client::couchbase_cxx_client taocpp::json)

add_executable(transactions_transfer_basic transactions_transfer_basic.cpp)
target_link_libraries(transactions_transfer_basic
PRIVATE couchbase_cxx_client::couchbase_cxx_client taocpp::json)
Expand Down
165 changes: 165 additions & 0 deletions examples/minimal_query.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,165 @@
#include <couchbase/cluster.hxx>
#include <couchbase/logger.hxx>

#include <tao/json.hpp>

#include <iostream>

struct program_config {
std::string connection_string{ "couchbase://127.0.0.1" };
std::string user_name{ "Administrator" };
std::string password{ "password" };
std::string bucket_name{ "travel-sample" };
std::string scope_name{ "inventory" };
std::optional<std::string> profile{};
bool verbose{ false };

static auto from_env() -> program_config;
static auto quote(std::string val) -> std::string;
void dump();
};

struct airline {
std::uint32_t id{ 0 };
std::string name;
std::string iata;
std::string icao;
std::string callsign;
std::string country;
};

std::ostream&
operator<<(std::ostream& os, const airline& a)
{
os << "airline(id: " << a.id << ", name: \"" << a.name << "\", iata: \"" << a.iata
<< "\", icao: \"" << a.icao << "\", callsign: \"" << a.callsign << "\", country: \""
<< a.country << "\")";
return os;
}

template<>
struct tao::json::traits<airline> {
template<template<typename...> class Traits>
static airline as(const tao::json::basic_value<Traits>& v)
{
if (!v.is_object()) {
return {};
}
if (const auto* airline_json = v.find("airline");
airline_json != nullptr && airline_json->is_object()) {
airline result{};
const auto& object = airline_json->get_object();
result.id = object.at("id").template optional<std::uint32_t>().value_or(0);
result.name = object.at("name").template optional<std::string>().value_or("");
result.iata = object.at("iata").template optional<std::string>().value_or("");
result.icao = object.at("icao").template optional<std::string>().value_or("");
result.callsign = object.at("callsign").template optional<std::string>().value_or("");
result.country = object.at("country").template optional<std::string>().value_or("");
return result;
}
return {};
}
};

int
main()
{
auto config = program_config::from_env();
config.dump();

if (config.verbose) {
couchbase::logger::initialize_console_logger();
couchbase::logger::set_level(couchbase::logger::log_level::trace);
}

auto options = couchbase::cluster_options(config.user_name, config.password);
if (config.profile) {
options.apply_profile(config.profile.value());
}

auto [connect_err, cluster] =
couchbase::cluster::connect(config.connection_string, options).get();
if (connect_err) {
std::cout << "Unable to connect to the cluster. ec: " << connect_err.message() << "\n";
return EXIT_FAILURE;
}
auto scope = cluster.bucket(config.bucket_name).scope(config.scope_name);

auto [err, resp] = scope.query("SELECT * FROM airline LIMIT 10").get();

std::cout << "--- Iterating as Binary data:\n";
for (auto row : resp.rows_as_binary()) {
std::cout << std::string(reinterpret_cast<const char*>(row.data()), row.size()) << "\n";
}

std::cout << "--- Iterating as C++ types:\n";
for (auto row : resp.rows_as<couchbase::codec::tao_json_serializer, airline>()) {
std::cout << row << "\n";
}

std::cout << "--- Iterating as JSON objects:\n";
for (auto row : resp.rows_as_json()) {
std::cout << "Airline(id: " << row["airline"]["id"] << ", name: \"" << row["airline"]["name"]
<< "\")\n";
}

cluster.close().get();

return 0;
}

auto
program_config::from_env() -> program_config
{
program_config config{};

if (const auto* val = getenv("CONNECTION_STRING"); val != nullptr) {
config.connection_string = val;
}
if (const auto* val = getenv("USER_NAME"); val != nullptr) {
config.user_name = val;
}
if (const auto* val = getenv("PASSWORD"); val != nullptr) {
config.password = val;
}
if (const auto* val = getenv("BUCKET_NAME"); val != nullptr) {
config.bucket_name = val;
}
if (const auto* val = getenv("SCOPE_NAME"); val != nullptr) {
config.scope_name = val;
}
if (const auto* val = getenv("PROFILE"); val != nullptr) {
config.profile = val; // e.g. "wan_development"
}
if (const auto* val = getenv("VERBOSE"); val != nullptr) {
const std::array<std::string, 5> truthy_values = {
"yes", "y", "on", "true", "1",
};
for (const auto& truth : truthy_values) {
if (val == truth) {
config.verbose = true;
break;
}
}
}

return config;
}

auto
program_config::quote(std::string val) -> std::string
{
return "\"" + val + "\"";
}

void
program_config::dump()
{
std::cout << " CONNECTION_STRING: " << quote(connection_string) << "\n";
std::cout << " USER_NAME: " << quote(user_name) << "\n";
std::cout << " PASSWORD: [HIDDEN]\n";
std::cout << " BUCKET_NAME: " << quote(bucket_name) << "\n";
std::cout << " SCOPE_NAME: " << quote(scope_name) << "\n";
std::cout << " VERBOSE: " << std::boolalpha << verbose << "\n";
std::cout << " PROFILE: " << (profile ? quote(*profile) : "[NONE]") << "\n\n";
}

0 comments on commit a2e6798

Please sign in to comment.