lookup_in_result.hxx

Go to the documentation of this file.

/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
 *   Copyright 2020-Present Couchbase, Inc.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */
 
#pragma once
 
#include <couchbase/codec/encoded_value.hxx>
#include <couchbase/codec/serializer_traits.hxx>
#include <couchbase/error_codes.hxx>
#include <couchbase/result.hxx>
#include <couchbase/subdoc/lookup_in_macro.hxx>
 
#include <optional>
 
namespace couchbase
{
 
namespace codec
{
class tao_json_serializer;
} // namespace codec
 
class lookup_in_result : public result
{
public:
  struct entry {
    std::string path;
    codec::binary value;
    std::size_t original_index;
    bool exists;
    std::error_code ec;
  };
 
  lookup_in_result() = default;
 
  lookup_in_result(couchbase::cas cas, std::vector<entry> entries, bool is_deleted)
    : result{ cas }
    , entries_{ std::move(entries) }
    , is_deleted_{ is_deleted }
  {
  }
 
  template<typename Document,
           typename Serializer = codec::tao_json_serializer,
           std::enable_if_t<codec::is_serializer_v<Serializer>, bool> = true>
  [[nodiscard]] auto content_as(std::size_t index) const -> Document
  {
    for (const entry& e : entries_) {
      if (e.original_index == index) {
        if (e.ec) {
          throw std::system_error(e.ec,
                                  "error getting result for spec at index " +
                                    std::to_string(index) + ", path \"" + e.path + "\"");
        }
 
        return Serializer::template deserialize<Document>(e.value);
      }
    }
    throw std::system_error(errc::key_value::path_invalid,
                            "invalid index for lookup_in result: {}" + std::to_string(index));
  }
 
  template<typename Document,
           typename Serializer = codec::tao_json_serializer,
           std::enable_if_t<codec::is_serializer_v<Serializer>, bool> = true>
  [[nodiscard]] auto content_as(const std::string& path) const -> Document
  {
    for (const entry& e : entries_) {
      if (e.path == path) {
        if (e.ec) {
          throw std::system_error(e.ec, "error getting result for path \"" + e.path + "\"");
        }
 
        return Serializer::template deserialize<Document>(e.value);
      }
    }
    throw std::system_error(errc::key_value::path_invalid,
                            "invalid path for lookup_in result: " + path);
  }
 
  template<typename Document,
           typename Serializer = codec::tao_json_serializer,
           std::enable_if_t<codec::is_serializer_v<Serializer>, bool> = true>
  [[nodiscard]] auto content_as(subdoc::lookup_in_macro macro) const -> Document
  {
    const auto& macro_string = subdoc::to_string(macro);
    for (const entry& e : entries_) {
      if (e.path == macro_string) {
        if (e.ec) {
          throw std::system_error(e.ec, "error getting result for macro \"" + macro_string + "\"");
        }
 
        return Serializer::template deserialize<Document>(e.value);
      }
    }
    throw std::system_error(errc::key_value::path_invalid,
                            "invalid path for lookup_in result: macro#" +
                              std::to_string(static_cast<std::uint32_t>(macro)));
  }
 
  [[nodiscard]] auto exists(std::size_t index) const -> bool
  {
    for (const entry& e : entries_) {
      if (e.original_index == index) {
        if (e.ec && e.ec != couchbase::errc::key_value::path_not_found) {
          throw std::system_error(e.ec, "error getting result for path \"" + e.path + "\"");
        }
 
        return e.exists;
      }
    }
    return false;
  }
 
  [[nodiscard]] auto exists(subdoc::lookup_in_macro macro) const -> bool
  {
    const auto& macro_string = subdoc::to_string(macro);
    for (const entry& e : entries_) {
      if (e.path == macro_string) {
        if (e.ec && e.ec != couchbase::errc::key_value::path_not_found) {
          throw std::system_error(e.ec, "error getting result for macro \"" + macro_string + "\"");
        }
 
        return e.exists;
      }
    }
    return false;
  }
 
  [[nodiscard]] auto exists(const std::string& path) const -> bool
  {
    for (const entry& e : entries_) {
      if (e.path == path) {
        if (e.ec && e.ec != couchbase::errc::key_value::path_not_found) {
          throw std::system_error(e.ec, "error getting result for path \"" + e.path + "\"");
        }
 
        return e.exists;
      }
    }
    return false;
  }
 
  [[nodiscard]] auto is_deleted() const -> bool
  {
    return is_deleted_;
  }
 
  [[nodiscard]] auto has_value(std::size_t index) const -> bool
  {
    for (const entry& e : entries_) {
      if (e.original_index == index) {
        return !e.value.empty();
      }
    }
    throw std::system_error(errc::key_value::path_invalid,
                            "invalid index for mutate_in result: " + std::to_string(index));
  }
 
  [[nodiscard]] auto has_value(const std::string& path) const -> bool
  {
    for (const entry& e : entries_) {
      if (e.path == path) {
        return !e.value.empty();
      }
    }
    throw std::system_error(errc::key_value::path_invalid,
                            "invalid path for mutate_in result: " + path);
  }
 
private:
  std::vector<entry> entries_{};
  bool is_deleted_{ false };
};
 
} // namespace couchbase