1500 lines
52 KiB
C
1500 lines
52 KiB
C
|
// Copyright 2020 The Abseil Authors.
|
||
|
//
|
||
|
// 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
|
||
|
//
|
||
|
// https://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.
|
||
|
//
|
||
|
// -----------------------------------------------------------------------------
|
||
|
// File: cord.h
|
||
|
// -----------------------------------------------------------------------------
|
||
|
//
|
||
|
// This file defines the `absl::Cord` data structure and operations on that data
|
||
|
// structure. A Cord is a string-like sequence of characters optimized for
|
||
|
// specific use cases. Unlike a `std::string`, which stores an array of
|
||
|
// contiguous characters, Cord data is stored in a structure consisting of
|
||
|
// separate, reference-counted "chunks." (Currently, this implementation is a
|
||
|
// tree structure, though that implementation may change.)
|
||
|
//
|
||
|
// Because a Cord consists of these chunks, data can be added to or removed from
|
||
|
// a Cord during its lifetime. Chunks may also be shared between Cords. Unlike a
|
||
|
// `std::string`, a Cord can therefore accommodate data that changes over its
|
||
|
// lifetime, though it's not quite "mutable"; it can change only in the
|
||
|
// attachment, detachment, or rearrangement of chunks of its constituent data.
|
||
|
//
|
||
|
// A Cord provides some benefit over `std::string` under the following (albeit
|
||
|
// narrow) circumstances:
|
||
|
//
|
||
|
// * Cord data is designed to grow and shrink over a Cord's lifetime. Cord
|
||
|
// provides efficient insertions and deletions at the start and end of the
|
||
|
// character sequences, avoiding copies in those cases. Static data should
|
||
|
// generally be stored as strings.
|
||
|
// * External memory consisting of string-like data can be directly added to
|
||
|
// a Cord without requiring copies or allocations.
|
||
|
// * Cord data may be shared and copied cheaply. Cord provides a copy-on-write
|
||
|
// implementation and cheap sub-Cord operations. Copying a Cord is an O(1)
|
||
|
// operation.
|
||
|
//
|
||
|
// As a consequence to the above, Cord data is generally large. Small data
|
||
|
// should generally use strings, as construction of a Cord requires some
|
||
|
// overhead. Small Cords (<= 15 bytes) are represented inline, but most small
|
||
|
// Cords are expected to grow over their lifetimes.
|
||
|
//
|
||
|
// Note that because a Cord is made up of separate chunked data, random access
|
||
|
// to character data within a Cord is slower than within a `std::string`.
|
||
|
//
|
||
|
// Thread Safety
|
||
|
//
|
||
|
// Cord has the same thread-safety properties as many other types like
|
||
|
// std::string, std::vector<>, int, etc -- it is thread-compatible. In
|
||
|
// particular, if threads do not call non-const methods, then it is safe to call
|
||
|
// const methods without synchronization. Copying a Cord produces a new instance
|
||
|
// that can be used concurrently with the original in arbitrary ways.
|
||
|
|
||
|
#ifndef ABSL_STRINGS_CORD_H_
|
||
|
#define ABSL_STRINGS_CORD_H_
|
||
|
|
||
|
#include <algorithm>
|
||
|
#include <cstddef>
|
||
|
#include <cstdint>
|
||
|
#include <cstring>
|
||
|
#include <iosfwd>
|
||
|
#include <iterator>
|
||
|
#include <string>
|
||
|
#include <type_traits>
|
||
|
|
||
|
#include "absl/base/config.h"
|
||
|
#include "absl/base/internal/endian.h"
|
||
|
#include "absl/base/internal/per_thread_tls.h"
|
||
|
#include "absl/base/macros.h"
|
||
|
#include "absl/base/port.h"
|
||
|
#include "absl/container/inlined_vector.h"
|
||
|
#include "absl/functional/function_ref.h"
|
||
|
#include "absl/meta/type_traits.h"
|
||
|
#include "absl/strings/internal/cord_internal.h"
|
||
|
#include "absl/strings/internal/cord_rep_btree.h"
|
||
|
#include "absl/strings/internal/cord_rep_btree_reader.h"
|
||
|
#include "absl/strings/internal/cord_rep_ring.h"
|
||
|
#include "absl/strings/internal/cordz_functions.h"
|
||
|
#include "absl/strings/internal/cordz_info.h"
|
||
|
#include "absl/strings/internal/cordz_statistics.h"
|
||
|
#include "absl/strings/internal/cordz_update_scope.h"
|
||
|
#include "absl/strings/internal/cordz_update_tracker.h"
|
||
|
#include "absl/strings/internal/resize_uninitialized.h"
|
||
|
#include "absl/strings/internal/string_constant.h"
|
||
|
#include "absl/strings/string_view.h"
|
||
|
#include "absl/types/optional.h"
|
||
|
|
||
|
namespace absl {
|
||
|
ABSL_NAMESPACE_BEGIN
|
||
|
class Cord;
|
||
|
class CordTestPeer;
|
||
|
template <typename Releaser>
|
||
|
Cord MakeCordFromExternal(absl::string_view, Releaser&&);
|
||
|
void CopyCordToString(const Cord& src, std::string* dst);
|
||
|
|
||
|
// Cord
|
||
|
//
|
||
|
// A Cord is a sequence of characters, designed to be more efficient than a
|
||
|
// `std::string` in certain circumstances: namely, large string data that needs
|
||
|
// to change over its lifetime or shared, especially when such data is shared
|
||
|
// across API boundaries.
|
||
|
//
|
||
|
// A Cord stores its character data in a structure that allows efficient prepend
|
||
|
// and append operations. This makes a Cord useful for large string data sent
|
||
|
// over in a wire format that may need to be prepended or appended at some point
|
||
|
// during the data exchange (e.g. HTTP, protocol buffers). For example, a
|
||
|
// Cord is useful for storing an HTTP request, and prepending an HTTP header to
|
||
|
// such a request.
|
||
|
//
|
||
|
// Cords should not be used for storing general string data, however. They
|
||
|
// require overhead to construct and are slower than strings for random access.
|
||
|
//
|
||
|
// The Cord API provides the following common API operations:
|
||
|
//
|
||
|
// * Create or assign Cords out of existing string data, memory, or other Cords
|
||
|
// * Append and prepend data to an existing Cord
|
||
|
// * Create new Sub-Cords from existing Cord data
|
||
|
// * Swap Cord data and compare Cord equality
|
||
|
// * Write out Cord data by constructing a `std::string`
|
||
|
//
|
||
|
// Additionally, the API provides iterator utilities to iterate through Cord
|
||
|
// data via chunks or character bytes.
|
||
|
//
|
||
|
class Cord {
|
||
|
private:
|
||
|
template <typename T>
|
||
|
using EnableIfString =
|
||
|
absl::enable_if_t<std::is_same<T, std::string>::value, int>;
|
||
|
|
||
|
public:
|
||
|
// Cord::Cord() Constructors.
|
||
|
|
||
|
// Creates an empty Cord.
|
||
|
constexpr Cord() noexcept;
|
||
|
|
||
|
// Creates a Cord from an existing Cord. Cord is copyable and efficiently
|
||
|
// movable. The moved-from state is valid but unspecified.
|
||
|
Cord(const Cord& src);
|
||
|
Cord(Cord&& src) noexcept;
|
||
|
Cord& operator=(const Cord& x);
|
||
|
Cord& operator=(Cord&& x) noexcept;
|
||
|
|
||
|
// Creates a Cord from a `src` string. This constructor is marked explicit to
|
||
|
// prevent implicit Cord constructions from arguments convertible to an
|
||
|
// `absl::string_view`.
|
||
|
explicit Cord(absl::string_view src);
|
||
|
Cord& operator=(absl::string_view src);
|
||
|
|
||
|
// Creates a Cord from a `std::string&&` rvalue. These constructors are
|
||
|
// templated to avoid ambiguities for types that are convertible to both
|
||
|
// `absl::string_view` and `std::string`, such as `const char*`.
|
||
|
template <typename T, EnableIfString<T> = 0>
|
||
|
explicit Cord(T&& src);
|
||
|
template <typename T, EnableIfString<T> = 0>
|
||
|
Cord& operator=(T&& src);
|
||
|
|
||
|
// Cord::~Cord()
|
||
|
//
|
||
|
// Destructs the Cord.
|
||
|
~Cord() {
|
||
|
if (contents_.is_tree()) DestroyCordSlow();
|
||
|
}
|
||
|
|
||
|
// MakeCordFromExternal()
|
||
|
//
|
||
|
// Creates a Cord that takes ownership of external string memory. The
|
||
|
// contents of `data` are not copied to the Cord; instead, the external
|
||
|
// memory is added to the Cord and reference-counted. This data may not be
|
||
|
// changed for the life of the Cord, though it may be prepended or appended
|
||
|
// to.
|
||
|
//
|
||
|
// `MakeCordFromExternal()` takes a callable "releaser" that is invoked when
|
||
|
// the reference count for `data` reaches zero. As noted above, this data must
|
||
|
// remain live until the releaser is invoked. The callable releaser also must:
|
||
|
//
|
||
|
// * be move constructible
|
||
|
// * support `void operator()(absl::string_view) const` or `void operator()`
|
||
|
//
|
||
|
// Example:
|
||
|
//
|
||
|
// Cord MakeCord(BlockPool* pool) {
|
||
|
// Block* block = pool->NewBlock();
|
||
|
// FillBlock(block);
|
||
|
// return absl::MakeCordFromExternal(
|
||
|
// block->ToStringView(),
|
||
|
// [pool, block](absl::string_view v) {
|
||
|
// pool->FreeBlock(block, v);
|
||
|
// });
|
||
|
// }
|
||
|
//
|
||
|
// WARNING: Because a Cord can be reference-counted, it's likely a bug if your
|
||
|
// releaser doesn't do anything. For example, consider the following:
|
||
|
//
|
||
|
// void Foo(const char* buffer, int len) {
|
||
|
// auto c = absl::MakeCordFromExternal(absl::string_view(buffer, len),
|
||
|
// [](absl::string_view) {});
|
||
|
//
|
||
|
// // BUG: If Bar() copies its cord for any reason, including keeping a
|
||
|
// // substring of it, the lifetime of buffer might be extended beyond
|
||
|
// // when Foo() returns.
|
||
|
// Bar(c);
|
||
|
// }
|
||
|
template <typename Releaser>
|
||
|
friend Cord MakeCordFromExternal(absl::string_view data, Releaser&& releaser);
|
||
|
|
||
|
// Cord::Clear()
|
||
|
//
|
||
|
// Releases the Cord data. Any nodes that share data with other Cords, if
|
||
|
// applicable, will have their reference counts reduced by 1.
|
||
|
void Clear();
|
||
|
|
||
|
// Cord::Append()
|
||
|
//
|
||
|
// Appends data to the Cord, which may come from another Cord or other string
|
||
|
// data.
|
||
|
void Append(const Cord& src);
|
||
|
void Append(Cord&& src);
|
||
|
void Append(absl::string_view src);
|
||
|
template <typename T, EnableIfString<T> = 0>
|
||
|
void Append(T&& src);
|
||
|
|
||
|
// Cord::Prepend()
|
||
|
//
|
||
|
// Prepends data to the Cord, which may come from another Cord or other string
|
||
|
// data.
|
||
|
void Prepend(const Cord& src);
|
||
|
void Prepend(absl::string_view src);
|
||
|
template <typename T, EnableIfString<T> = 0>
|
||
|
void Prepend(T&& src);
|
||
|
|
||
|
// Cord::RemovePrefix()
|
||
|
//
|
||
|
// Removes the first `n` bytes of a Cord.
|
||
|
void RemovePrefix(size_t n);
|
||
|
void RemoveSuffix(size_t n);
|
||
|
|
||
|
// Cord::Subcord()
|
||
|
//
|
||
|
// Returns a new Cord representing the subrange [pos, pos + new_size) of
|
||
|
// *this. If pos >= size(), the result is empty(). If
|
||
|
// (pos + new_size) >= size(), the result is the subrange [pos, size()).
|
||
|
Cord Subcord(size_t pos, size_t new_size) const;
|
||
|
|
||
|
// Cord::swap()
|
||
|
//
|
||
|
// Swaps the contents of the Cord with `other`.
|
||
|
void swap(Cord& other) noexcept;
|
||
|
|
||
|
// swap()
|
||
|
//
|
||
|
// Swaps the contents of two Cords.
|
||
|
friend void swap(Cord& x, Cord& y) noexcept {
|
||
|
x.swap(y);
|
||
|
}
|
||
|
|
||
|
// Cord::size()
|
||
|
//
|
||
|
// Returns the size of the Cord.
|
||
|
size_t size() const;
|
||
|
|
||
|
// Cord::empty()
|
||
|
//
|
||
|
// Determines whether the given Cord is empty, returning `true` is so.
|
||
|
bool empty() const;
|
||
|
|
||
|
// Cord::EstimatedMemoryUsage()
|
||
|
//
|
||
|
// Returns the *approximate* number of bytes held in full or in part by this
|
||
|
// Cord (which may not remain the same between invocations). Note that Cords
|
||
|
// that share memory could each be "charged" independently for the same shared
|
||
|
// memory.
|
||
|
size_t EstimatedMemoryUsage() const;
|
||
|
|
||
|
// Cord::Compare()
|
||
|
//
|
||
|
// Compares 'this' Cord with rhs. This function and its relatives treat Cords
|
||
|
// as sequences of unsigned bytes. The comparison is a straightforward
|
||
|
// lexicographic comparison. `Cord::Compare()` returns values as follows:
|
||
|
//
|
||
|
// -1 'this' Cord is smaller
|
||
|
// 0 two Cords are equal
|
||
|
// 1 'this' Cord is larger
|
||
|
int Compare(absl::string_view rhs) const;
|
||
|
int Compare(const Cord& rhs) const;
|
||
|
|
||
|
// Cord::StartsWith()
|
||
|
//
|
||
|
// Determines whether the Cord starts with the passed string data `rhs`.
|
||
|
bool StartsWith(const Cord& rhs) const;
|
||
|
bool StartsWith(absl::string_view rhs) const;
|
||
|
|
||
|
// Cord::EndsWith()
|
||
|
//
|
||
|
// Determines whether the Cord ends with the passed string data `rhs`.
|
||
|
bool EndsWith(absl::string_view rhs) const;
|
||
|
bool EndsWith(const Cord& rhs) const;
|
||
|
|
||
|
// Cord::operator std::string()
|
||
|
//
|
||
|
// Converts a Cord into a `std::string()`. This operator is marked explicit to
|
||
|
// prevent unintended Cord usage in functions that take a string.
|
||
|
explicit operator std::string() const;
|
||
|
|
||
|
// CopyCordToString()
|
||
|
//
|
||
|
// Copies the contents of a `src` Cord into a `*dst` string.
|
||
|
//
|
||
|
// This function optimizes the case of reusing the destination string since it
|
||
|
// can reuse previously allocated capacity. However, this function does not
|
||
|
// guarantee that pointers previously returned by `dst->data()` remain valid
|
||
|
// even if `*dst` had enough capacity to hold `src`. If `*dst` is a new
|
||
|
// object, prefer to simply use the conversion operator to `std::string`.
|
||
|
friend void CopyCordToString(const Cord& src, std::string* dst);
|
||
|
|
||
|
class CharIterator;
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Cord::ChunkIterator
|
||
|
//----------------------------------------------------------------------------
|
||
|
//
|
||
|
// A `Cord::ChunkIterator` allows iteration over the constituent chunks of its
|
||
|
// Cord. Such iteration allows you to perform non-const operatons on the data
|
||
|
// of a Cord without modifying it.
|
||
|
//
|
||
|
// Generally, you do not instantiate a `Cord::ChunkIterator` directly;
|
||
|
// instead, you create one implicitly through use of the `Cord::Chunks()`
|
||
|
// member function.
|
||
|
//
|
||
|
// The `Cord::ChunkIterator` has the following properties:
|
||
|
//
|
||
|
// * The iterator is invalidated after any non-const operation on the
|
||
|
// Cord object over which it iterates.
|
||
|
// * The `string_view` returned by dereferencing a valid, non-`end()`
|
||
|
// iterator is guaranteed to be non-empty.
|
||
|
// * Two `ChunkIterator` objects can be compared equal if and only if they
|
||
|
// remain valid and iterate over the same Cord.
|
||
|
// * The iterator in this case is a proxy iterator; the `string_view`
|
||
|
// returned by the iterator does not live inside the Cord, and its
|
||
|
// lifetime is limited to the lifetime of the iterator itself. To help
|
||
|
// prevent lifetime issues, `ChunkIterator::reference` is not a true
|
||
|
// reference type and is equivalent to `value_type`.
|
||
|
// * The iterator keeps state that can grow for Cords that contain many
|
||
|
// nodes and are imbalanced due to sharing. Prefer to pass this type by
|
||
|
// const reference instead of by value.
|
||
|
class ChunkIterator {
|
||
|
public:
|
||
|
using iterator_category = std::input_iterator_tag;
|
||
|
using value_type = absl::string_view;
|
||
|
using difference_type = ptrdiff_t;
|
||
|
using pointer = const value_type*;
|
||
|
using reference = value_type;
|
||
|
|
||
|
ChunkIterator() = default;
|
||
|
|
||
|
ChunkIterator& operator++();
|
||
|
ChunkIterator operator++(int);
|
||
|
bool operator==(const ChunkIterator& other) const;
|
||
|
bool operator!=(const ChunkIterator& other) const;
|
||
|
reference operator*() const;
|
||
|
pointer operator->() const;
|
||
|
|
||
|
friend class Cord;
|
||
|
friend class CharIterator;
|
||
|
|
||
|
private:
|
||
|
using CordRep = absl::cord_internal::CordRep;
|
||
|
using CordRepBtree = absl::cord_internal::CordRepBtree;
|
||
|
using CordRepBtreeReader = absl::cord_internal::CordRepBtreeReader;
|
||
|
|
||
|
// Stack of right children of concat nodes that we have to visit.
|
||
|
// Keep this at the end of the structure to avoid cache-thrashing.
|
||
|
// TODO(jgm): Benchmark to see if there's a more optimal value than 47 for
|
||
|
// the inlined vector size (47 exists for backward compatibility).
|
||
|
using Stack = absl::InlinedVector<absl::cord_internal::CordRep*, 47>;
|
||
|
|
||
|
// Constructs a `begin()` iterator from `tree`. `tree` must not be null.
|
||
|
explicit ChunkIterator(cord_internal::CordRep* tree);
|
||
|
|
||
|
// Constructs a `begin()` iterator from `cord`.
|
||
|
explicit ChunkIterator(const Cord* cord);
|
||
|
|
||
|
// Initializes this instance from a tree. Invoked by constructors.
|
||
|
void InitTree(cord_internal::CordRep* tree);
|
||
|
|
||
|
// Removes `n` bytes from `current_chunk_`. Expects `n` to be smaller than
|
||
|
// `current_chunk_.size()`.
|
||
|
void RemoveChunkPrefix(size_t n);
|
||
|
Cord AdvanceAndReadBytes(size_t n);
|
||
|
void AdvanceBytes(size_t n);
|
||
|
|
||
|
// Stack specific operator++
|
||
|
ChunkIterator& AdvanceStack();
|
||
|
|
||
|
// Btree specific operator++
|
||
|
ChunkIterator& AdvanceBtree();
|
||
|
void AdvanceBytesBtree(size_t n);
|
||
|
|
||
|
// Iterates `n` bytes, where `n` is expected to be greater than or equal to
|
||
|
// `current_chunk_.size()`.
|
||
|
void AdvanceBytesSlowPath(size_t n);
|
||
|
|
||
|
// A view into bytes of the current `CordRep`. It may only be a view to a
|
||
|
// suffix of bytes if this is being used by `CharIterator`.
|
||
|
absl::string_view current_chunk_;
|
||
|
// The current leaf, or `nullptr` if the iterator points to short data.
|
||
|
// If the current chunk is a substring node, current_leaf_ points to the
|
||
|
// underlying flat or external node.
|
||
|
absl::cord_internal::CordRep* current_leaf_ = nullptr;
|
||
|
// The number of bytes left in the `Cord` over which we are iterating.
|
||
|
size_t bytes_remaining_ = 0;
|
||
|
|
||
|
// Cord reader for cord btrees. Empty if not traversing a btree.
|
||
|
CordRepBtreeReader btree_reader_;
|
||
|
|
||
|
// See 'Stack' alias definition.
|
||
|
Stack stack_of_right_children_;
|
||
|
};
|
||
|
|
||
|
// Cord::ChunkIterator::chunk_begin()
|
||
|
//
|
||
|
// Returns an iterator to the first chunk of the `Cord`.
|
||
|
//
|
||
|
// Generally, prefer using `Cord::Chunks()` within a range-based for loop for
|
||
|
// iterating over the chunks of a Cord. This method may be useful for getting
|
||
|
// a `ChunkIterator` where range-based for-loops are not useful.
|
||
|
//
|
||
|
// Example:
|
||
|
//
|
||
|
// absl::Cord::ChunkIterator FindAsChunk(const absl::Cord& c,
|
||
|
// absl::string_view s) {
|
||
|
// return std::find(c.chunk_begin(), c.chunk_end(), s);
|
||
|
// }
|
||
|
ChunkIterator chunk_begin() const;
|
||
|
|
||
|
// Cord::ChunkItertator::chunk_end()
|
||
|
//
|
||
|
// Returns an iterator one increment past the last chunk of the `Cord`.
|
||
|
//
|
||
|
// Generally, prefer using `Cord::Chunks()` within a range-based for loop for
|
||
|
// iterating over the chunks of a Cord. This method may be useful for getting
|
||
|
// a `ChunkIterator` where range-based for-loops may not be available.
|
||
|
ChunkIterator chunk_end() const;
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Cord::ChunkIterator::ChunkRange
|
||
|
//----------------------------------------------------------------------------
|
||
|
//
|
||
|
// `ChunkRange` is a helper class for iterating over the chunks of the `Cord`,
|
||
|
// producing an iterator which can be used within a range-based for loop.
|
||
|
// Construction of a `ChunkRange` will return an iterator pointing to the
|
||
|
// first chunk of the Cord. Generally, do not construct a `ChunkRange`
|
||
|
// directly; instead, prefer to use the `Cord::Chunks()` method.
|
||
|
//
|
||
|
// Implementation note: `ChunkRange` is simply a convenience wrapper over
|
||
|
// `Cord::chunk_begin()` and `Cord::chunk_end()`.
|
||
|
class ChunkRange {
|
||
|
public:
|
||
|
explicit ChunkRange(const Cord* cord) : cord_(cord) {}
|
||
|
|
||
|
ChunkIterator begin() const;
|
||
|
ChunkIterator end() const;
|
||
|
|
||
|
private:
|
||
|
const Cord* cord_;
|
||
|
};
|
||
|
|
||
|
// Cord::Chunks()
|
||
|
//
|
||
|
// Returns a `Cord::ChunkIterator::ChunkRange` for iterating over the chunks
|
||
|
// of a `Cord` with a range-based for-loop. For most iteration tasks on a
|
||
|
// Cord, use `Cord::Chunks()` to retrieve this iterator.
|
||
|
//
|
||
|
// Example:
|
||
|
//
|
||
|
// void ProcessChunks(const Cord& cord) {
|
||
|
// for (absl::string_view chunk : cord.Chunks()) { ... }
|
||
|
// }
|
||
|
//
|
||
|
// Note that the ordinary caveats of temporary lifetime extension apply:
|
||
|
//
|
||
|
// void Process() {
|
||
|
// for (absl::string_view chunk : CordFactory().Chunks()) {
|
||
|
// // The temporary Cord returned by CordFactory has been destroyed!
|
||
|
// }
|
||
|
// }
|
||
|
ChunkRange Chunks() const;
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Cord::CharIterator
|
||
|
//----------------------------------------------------------------------------
|
||
|
//
|
||
|
// A `Cord::CharIterator` allows iteration over the constituent characters of
|
||
|
// a `Cord`.
|
||
|
//
|
||
|
// Generally, you do not instantiate a `Cord::CharIterator` directly; instead,
|
||
|
// you create one implicitly through use of the `Cord::Chars()` member
|
||
|
// function.
|
||
|
//
|
||
|
// A `Cord::CharIterator` has the following properties:
|
||
|
//
|
||
|
// * The iterator is invalidated after any non-const operation on the
|
||
|
// Cord object over which it iterates.
|
||
|
// * Two `CharIterator` objects can be compared equal if and only if they
|
||
|
// remain valid and iterate over the same Cord.
|
||
|
// * The iterator keeps state that can grow for Cords that contain many
|
||
|
// nodes and are imbalanced due to sharing. Prefer to pass this type by
|
||
|
// const reference instead of by value.
|
||
|
// * This type cannot act as a forward iterator because a `Cord` can reuse
|
||
|
// sections of memory. This fact violates the requirement for forward
|
||
|
// iterators to compare equal if dereferencing them returns the same
|
||
|
// object.
|
||
|
class CharIterator {
|
||
|
public:
|
||
|
using iterator_category = std::input_iterator_tag;
|
||
|
using value_type = char;
|
||
|
using difference_type = ptrdiff_t;
|
||
|
using pointer = const char*;
|
||
|
using reference = const char&;
|
||
|
|
||
|
CharIterator() = default;
|
||
|
|
||
|
CharIterator& operator++();
|
||
|
CharIterator operator++(int);
|
||
|
bool operator==(const CharIterator& other) const;
|
||
|
bool operator!=(const CharIterator& other) const;
|
||
|
reference operator*() const;
|
||
|
pointer operator->() const;
|
||
|
|
||
|
friend Cord;
|
||
|
|
||
|
private:
|
||
|
explicit CharIterator(const Cord* cord) : chunk_iterator_(cord) {}
|
||
|
|
||
|
ChunkIterator chunk_iterator_;
|
||
|
};
|
||
|
|
||
|
// Cord::CharIterator::AdvanceAndRead()
|
||
|
//
|
||
|
// Advances the `Cord::CharIterator` by `n_bytes` and returns the bytes
|
||
|
// advanced as a separate `Cord`. `n_bytes` must be less than or equal to the
|
||
|
// number of bytes within the Cord; otherwise, behavior is undefined. It is
|
||
|
// valid to pass `char_end()` and `0`.
|
||
|
static Cord AdvanceAndRead(CharIterator* it, size_t n_bytes);
|
||
|
|
||
|
// Cord::CharIterator::Advance()
|
||
|
//
|
||
|
// Advances the `Cord::CharIterator` by `n_bytes`. `n_bytes` must be less than
|
||
|
// or equal to the number of bytes remaining within the Cord; otherwise,
|
||
|
// behavior is undefined. It is valid to pass `char_end()` and `0`.
|
||
|
static void Advance(CharIterator* it, size_t n_bytes);
|
||
|
|
||
|
// Cord::CharIterator::ChunkRemaining()
|
||
|
//
|
||
|
// Returns the longest contiguous view starting at the iterator's position.
|
||
|
//
|
||
|
// `it` must be dereferenceable.
|
||
|
static absl::string_view ChunkRemaining(const CharIterator& it);
|
||
|
|
||
|
// Cord::CharIterator::char_begin()
|
||
|
//
|
||
|
// Returns an iterator to the first character of the `Cord`.
|
||
|
//
|
||
|
// Generally, prefer using `Cord::Chars()` within a range-based for loop for
|
||
|
// iterating over the chunks of a Cord. This method may be useful for getting
|
||
|
// a `CharIterator` where range-based for-loops may not be available.
|
||
|
CharIterator char_begin() const;
|
||
|
|
||
|
// Cord::CharIterator::char_end()
|
||
|
//
|
||
|
// Returns an iterator to one past the last character of the `Cord`.
|
||
|
//
|
||
|
// Generally, prefer using `Cord::Chars()` within a range-based for loop for
|
||
|
// iterating over the chunks of a Cord. This method may be useful for getting
|
||
|
// a `CharIterator` where range-based for-loops are not useful.
|
||
|
CharIterator char_end() const;
|
||
|
|
||
|
// Cord::CharIterator::CharRange
|
||
|
//
|
||
|
// `CharRange` is a helper class for iterating over the characters of a
|
||
|
// producing an iterator which can be used within a range-based for loop.
|
||
|
// Construction of a `CharRange` will return an iterator pointing to the first
|
||
|
// character of the Cord. Generally, do not construct a `CharRange` directly;
|
||
|
// instead, prefer to use the `Cord::Chars()` method show below.
|
||
|
//
|
||
|
// Implementation note: `CharRange` is simply a convenience wrapper over
|
||
|
// `Cord::char_begin()` and `Cord::char_end()`.
|
||
|
class CharRange {
|
||
|
public:
|
||
|
explicit CharRange(const Cord* cord) : cord_(cord) {}
|
||
|
|
||
|
CharIterator begin() const;
|
||
|
CharIterator end() const;
|
||
|
|
||
|
private:
|
||
|
const Cord* cord_;
|
||
|
};
|
||
|
|
||
|
// Cord::CharIterator::Chars()
|
||
|
//
|
||
|
// Returns a `Cord::CharIterator` for iterating over the characters of a
|
||
|
// `Cord` with a range-based for-loop. For most character-based iteration
|
||
|
// tasks on a Cord, use `Cord::Chars()` to retrieve this iterator.
|
||
|
//
|
||
|
// Example:
|
||
|
//
|
||
|
// void ProcessCord(const Cord& cord) {
|
||
|
// for (char c : cord.Chars()) { ... }
|
||
|
// }
|
||
|
//
|
||
|
// Note that the ordinary caveats of temporary lifetime extension apply:
|
||
|
//
|
||
|
// void Process() {
|
||
|
// for (char c : CordFactory().Chars()) {
|
||
|
// // The temporary Cord returned by CordFactory has been destroyed!
|
||
|
// }
|
||
|
// }
|
||
|
CharRange Chars() const;
|
||
|
|
||
|
// Cord::operator[]
|
||
|
//
|
||
|
// Gets the "i"th character of the Cord and returns it, provided that
|
||
|
// 0 <= i < Cord.size().
|
||
|
//
|
||
|
// NOTE: This routine is reasonably efficient. It is roughly
|
||
|
// logarithmic based on the number of chunks that make up the cord. Still,
|
||
|
// if you need to iterate over the contents of a cord, you should
|
||
|
// use a CharIterator/ChunkIterator rather than call operator[] or Get()
|
||
|
// repeatedly in a loop.
|
||
|
char operator[](size_t i) const;
|
||
|
|
||
|
// Cord::TryFlat()
|
||
|
//
|
||
|
// If this cord's representation is a single flat array, returns a
|
||
|
// string_view referencing that array. Otherwise returns nullopt.
|
||
|
absl::optional<absl::string_view> TryFlat() const;
|
||
|
|
||
|
// Cord::Flatten()
|
||
|
//
|
||
|
// Flattens the cord into a single array and returns a view of the data.
|
||
|
//
|
||
|
// If the cord was already flat, the contents are not modified.
|
||
|
absl::string_view Flatten();
|
||
|
|
||
|
// Supports absl::Cord as a sink object for absl::Format().
|
||
|
friend void AbslFormatFlush(absl::Cord* cord, absl::string_view part) {
|
||
|
cord->Append(part);
|
||
|
}
|
||
|
|
||
|
template <typename H>
|
||
|
friend H AbslHashValue(H hash_state, const absl::Cord& c) {
|
||
|
absl::optional<absl::string_view> maybe_flat = c.TryFlat();
|
||
|
if (maybe_flat.has_value()) {
|
||
|
return H::combine(std::move(hash_state), *maybe_flat);
|
||
|
}
|
||
|
return c.HashFragmented(std::move(hash_state));
|
||
|
}
|
||
|
|
||
|
// Create a Cord with the contents of StringConstant<T>::value.
|
||
|
// No allocations will be done and no data will be copied.
|
||
|
// This is an INTERNAL API and subject to change or removal. This API can only
|
||
|
// be used by spelling absl::strings_internal::MakeStringConstant, which is
|
||
|
// also an internal API.
|
||
|
template <typename T>
|
||
|
explicit constexpr Cord(strings_internal::StringConstant<T>);
|
||
|
|
||
|
private:
|
||
|
using CordRep = absl::cord_internal::CordRep;
|
||
|
using CordRepFlat = absl::cord_internal::CordRepFlat;
|
||
|
using CordzInfo = cord_internal::CordzInfo;
|
||
|
using CordzUpdateScope = cord_internal::CordzUpdateScope;
|
||
|
using CordzUpdateTracker = cord_internal::CordzUpdateTracker;
|
||
|
using InlineData = cord_internal::InlineData;
|
||
|
using MethodIdentifier = CordzUpdateTracker::MethodIdentifier;
|
||
|
|
||
|
// Creates a cord instance with `method` representing the originating
|
||
|
// public API call causing the cord to be created.
|
||
|
explicit Cord(absl::string_view src, MethodIdentifier method);
|
||
|
|
||
|
friend class CordTestPeer;
|
||
|
friend bool operator==(const Cord& lhs, const Cord& rhs);
|
||
|
friend bool operator==(const Cord& lhs, absl::string_view rhs);
|
||
|
|
||
|
friend const CordzInfo* GetCordzInfoForTesting(const Cord& cord);
|
||
|
|
||
|
// Calls the provided function once for each cord chunk, in order. Unlike
|
||
|
// Chunks(), this API will not allocate memory.
|
||
|
void ForEachChunk(absl::FunctionRef<void(absl::string_view)>) const;
|
||
|
|
||
|
// Allocates new contiguous storage for the contents of the cord. This is
|
||
|
// called by Flatten() when the cord was not already flat.
|
||
|
absl::string_view FlattenSlowPath();
|
||
|
|
||
|
// Actual cord contents are hidden inside the following simple
|
||
|
// class so that we can isolate the bulk of cord.cc from changes
|
||
|
// to the representation.
|
||
|
//
|
||
|
// InlineRep holds either a tree pointer, or an array of kMaxInline bytes.
|
||
|
class InlineRep {
|
||
|
public:
|
||
|
static constexpr unsigned char kMaxInline = cord_internal::kMaxInline;
|
||
|
static_assert(kMaxInline >= sizeof(absl::cord_internal::CordRep*), "");
|
||
|
|
||
|
constexpr InlineRep() : data_() {}
|
||
|
explicit InlineRep(InlineData::DefaultInitType init) : data_(init) {}
|
||
|
InlineRep(const InlineRep& src);
|
||
|
InlineRep(InlineRep&& src);
|
||
|
InlineRep& operator=(const InlineRep& src);
|
||
|
InlineRep& operator=(InlineRep&& src) noexcept;
|
||
|
|
||
|
explicit constexpr InlineRep(cord_internal::InlineData data);
|
||
|
|
||
|
void Swap(InlineRep* rhs);
|
||
|
bool empty() const;
|
||
|
size_t size() const;
|
||
|
const char* data() const; // Returns nullptr if holding pointer
|
||
|
void set_data(const char* data, size_t n,
|
||
|
bool nullify_tail); // Discards pointer, if any
|
||
|
char* set_data(size_t n); // Write data to the result
|
||
|
// Returns nullptr if holding bytes
|
||
|
absl::cord_internal::CordRep* tree() const;
|
||
|
absl::cord_internal::CordRep* as_tree() const;
|
||
|
// Returns non-null iff was holding a pointer
|
||
|
absl::cord_internal::CordRep* clear();
|
||
|
// Converts to pointer if necessary.
|
||
|
void reduce_size(size_t n); // REQUIRES: holding data
|
||
|
void remove_prefix(size_t n); // REQUIRES: holding data
|
||
|
void AppendArray(absl::string_view src, MethodIdentifier method);
|
||
|
absl::string_view FindFlatStartPiece() const;
|
||
|
|
||
|
// Creates a CordRepFlat instance from the current inlined data with `extra'
|
||
|
// bytes of desired additional capacity.
|
||
|
CordRepFlat* MakeFlatWithExtraCapacity(size_t extra);
|
||
|
|
||
|
// Sets the tree value for this instance. `rep` must not be null.
|
||
|
// Requires the current instance to hold a tree, and a lock to be held on
|
||
|
// any CordzInfo referenced by this instance. The latter is enforced through
|
||
|
// the CordzUpdateScope argument. If the current instance is sampled, then
|
||
|
// the CordzInfo instance is updated to reference the new `rep` value.
|
||
|
void SetTree(CordRep* rep, const CordzUpdateScope& scope);
|
||
|
|
||
|
// Identical to SetTree(), except that `rep` is allowed to be null, in
|
||
|
// which case the current instance is reset to an empty value.
|
||
|
void SetTreeOrEmpty(CordRep* rep, const CordzUpdateScope& scope);
|
||
|
|
||
|
// Sets the tree value for this instance, and randomly samples this cord.
|
||
|
// This function disregards existing contents in `data_`, and should be
|
||
|
// called when a Cord is 'promoted' from an 'uninitialized' or 'inlined'
|
||
|
// value to a non-inlined (tree / ring) value.
|
||
|
void EmplaceTree(CordRep* rep, MethodIdentifier method);
|
||
|
|
||
|
// Identical to EmplaceTree, except that it copies the parent stack from
|
||
|
// the provided `parent` data if the parent is sampled.
|
||
|
void EmplaceTree(CordRep* rep, const InlineData& parent,
|
||
|
MethodIdentifier method);
|
||
|
|
||
|
// Commits the change of a newly created, or updated `rep` root value into
|
||
|
// this cord. `old_rep` indicates the old (inlined or tree) value of the
|
||
|
// cord, and determines if the commit invokes SetTree() or EmplaceTree().
|
||
|
void CommitTree(const CordRep* old_rep, CordRep* rep,
|
||
|
const CordzUpdateScope& scope, MethodIdentifier method);
|
||
|
|
||
|
void AppendTreeToInlined(CordRep* tree, MethodIdentifier method);
|
||
|
void AppendTreeToTree(CordRep* tree, MethodIdentifier method);
|
||
|
void AppendTree(CordRep* tree, MethodIdentifier method);
|
||
|
void PrependTreeToInlined(CordRep* tree, MethodIdentifier method);
|
||
|
void PrependTreeToTree(CordRep* tree, MethodIdentifier method);
|
||
|
void PrependTree(CordRep* tree, MethodIdentifier method);
|
||
|
|
||
|
template <bool has_length>
|
||
|
void GetAppendRegion(char** region, size_t* size, size_t length);
|
||
|
|
||
|
bool IsSame(const InlineRep& other) const {
|
||
|
return memcmp(&data_, &other.data_, sizeof(data_)) == 0;
|
||
|
}
|
||
|
int BitwiseCompare(const InlineRep& other) const {
|
||
|
uint64_t x, y;
|
||
|
// Use memcpy to avoid aliasing issues.
|
||
|
memcpy(&x, &data_, sizeof(x));
|
||
|
memcpy(&y, &other.data_, sizeof(y));
|
||
|
if (x == y) {
|
||
|
memcpy(&x, reinterpret_cast<const char*>(&data_) + 8, sizeof(x));
|
||
|
memcpy(&y, reinterpret_cast<const char*>(&other.data_) + 8, sizeof(y));
|
||
|
if (x == y) return 0;
|
||
|
}
|
||
|
return absl::big_endian::FromHost64(x) < absl::big_endian::FromHost64(y)
|
||
|
? -1
|
||
|
: 1;
|
||
|
}
|
||
|
void CopyTo(std::string* dst) const {
|
||
|
// memcpy is much faster when operating on a known size. On most supported
|
||
|
// platforms, the small string optimization is large enough that resizing
|
||
|
// to 15 bytes does not cause a memory allocation.
|
||
|
absl::strings_internal::STLStringResizeUninitialized(dst,
|
||
|
sizeof(data_) - 1);
|
||
|
memcpy(&(*dst)[0], &data_, sizeof(data_) - 1);
|
||
|
// erase is faster than resize because the logic for memory allocation is
|
||
|
// not needed.
|
||
|
dst->erase(inline_size());
|
||
|
}
|
||
|
|
||
|
// Copies the inline contents into `dst`. Assumes the cord is not empty.
|
||
|
void CopyToArray(char* dst) const;
|
||
|
|
||
|
bool is_tree() const { return data_.is_tree(); }
|
||
|
|
||
|
// Returns true if the Cord is being profiled by cordz.
|
||
|
bool is_profiled() const { return data_.is_tree() && data_.is_profiled(); }
|
||
|
|
||
|
// Returns the profiled CordzInfo, or nullptr if not sampled.
|
||
|
absl::cord_internal::CordzInfo* cordz_info() const {
|
||
|
return data_.cordz_info();
|
||
|
}
|
||
|
|
||
|
// Sets the profiled CordzInfo. `cordz_info` must not be null.
|
||
|
void set_cordz_info(cord_internal::CordzInfo* cordz_info) {
|
||
|
assert(cordz_info != nullptr);
|
||
|
data_.set_cordz_info(cordz_info);
|
||
|
}
|
||
|
|
||
|
// Resets the current cordz_info to null / empty.
|
||
|
void clear_cordz_info() { data_.clear_cordz_info(); }
|
||
|
|
||
|
private:
|
||
|
friend class Cord;
|
||
|
|
||
|
void AssignSlow(const InlineRep& src);
|
||
|
// Unrefs the tree and stops profiling.
|
||
|
void UnrefTree();
|
||
|
|
||
|
void ResetToEmpty() { data_ = {}; }
|
||
|
|
||
|
void set_inline_size(size_t size) { data_.set_inline_size(size); }
|
||
|
size_t inline_size() const { return data_.inline_size(); }
|
||
|
|
||
|
cord_internal::InlineData data_;
|
||
|
};
|
||
|
InlineRep contents_;
|
||
|
|
||
|
// Helper for MemoryUsage().
|
||
|
static size_t MemoryUsageAux(const absl::cord_internal::CordRep* rep);
|
||
|
|
||
|
// Helper for GetFlat() and TryFlat().
|
||
|
static bool GetFlatAux(absl::cord_internal::CordRep* rep,
|
||
|
absl::string_view* fragment);
|
||
|
|
||
|
// Helper for ForEachChunk().
|
||
|
static void ForEachChunkAux(
|
||
|
absl::cord_internal::CordRep* rep,
|
||
|
absl::FunctionRef<void(absl::string_view)> callback);
|
||
|
|
||
|
// The destructor for non-empty Cords.
|
||
|
void DestroyCordSlow();
|
||
|
|
||
|
// Out-of-line implementation of slower parts of logic.
|
||
|
void CopyToArraySlowPath(char* dst) const;
|
||
|
int CompareSlowPath(absl::string_view rhs, size_t compared_size,
|
||
|
size_t size_to_compare) const;
|
||
|
int CompareSlowPath(const Cord& rhs, size_t compared_size,
|
||
|
size_t size_to_compare) const;
|
||
|
bool EqualsImpl(absl::string_view rhs, size_t size_to_compare) const;
|
||
|
bool EqualsImpl(const Cord& rhs, size_t size_to_compare) const;
|
||
|
int CompareImpl(const Cord& rhs) const;
|
||
|
|
||
|
template <typename ResultType, typename RHS>
|
||
|
friend ResultType GenericCompare(const Cord& lhs, const RHS& rhs,
|
||
|
size_t size_to_compare);
|
||
|
static absl::string_view GetFirstChunk(const Cord& c);
|
||
|
static absl::string_view GetFirstChunk(absl::string_view sv);
|
||
|
|
||
|
// Returns a new reference to contents_.tree(), or steals an existing
|
||
|
// reference if called on an rvalue.
|
||
|
absl::cord_internal::CordRep* TakeRep() const&;
|
||
|
absl::cord_internal::CordRep* TakeRep() &&;
|
||
|
|
||
|
// Helper for Append().
|
||
|
template <typename C>
|
||
|
void AppendImpl(C&& src);
|
||
|
|
||
|
// Assigns the value in 'src' to this instance, 'stealing' its contents.
|
||
|
// Requires src.length() > kMaxBytesToCopy.
|
||
|
Cord& AssignLargeString(std::string&& src);
|
||
|
|
||
|
// Helper for AbslHashValue().
|
||
|
template <typename H>
|
||
|
H HashFragmented(H hash_state) const {
|
||
|
typename H::AbslInternalPiecewiseCombiner combiner;
|
||
|
ForEachChunk([&combiner, &hash_state](absl::string_view chunk) {
|
||
|
hash_state = combiner.add_buffer(std::move(hash_state), chunk.data(),
|
||
|
chunk.size());
|
||
|
});
|
||
|
return H::combine(combiner.finalize(std::move(hash_state)), size());
|
||
|
}
|
||
|
};
|
||
|
|
||
|
ABSL_NAMESPACE_END
|
||
|
} // namespace absl
|
||
|
|
||
|
namespace absl {
|
||
|
ABSL_NAMESPACE_BEGIN
|
||
|
|
||
|
// allow a Cord to be logged
|
||
|
extern std::ostream& operator<<(std::ostream& out, const Cord& cord);
|
||
|
|
||
|
// ------------------------------------------------------------------
|
||
|
// Internal details follow. Clients should ignore.
|
||
|
|
||
|
namespace cord_internal {
|
||
|
|
||
|
// Fast implementation of memmove for up to 15 bytes. This implementation is
|
||
|
// safe for overlapping regions. If nullify_tail is true, the destination is
|
||
|
// padded with '\0' up to 16 bytes.
|
||
|
inline void SmallMemmove(char* dst, const char* src, size_t n,
|
||
|
bool nullify_tail = false) {
|
||
|
if (n >= 8) {
|
||
|
assert(n <= 16);
|
||
|
uint64_t buf1;
|
||
|
uint64_t buf2;
|
||
|
memcpy(&buf1, src, 8);
|
||
|
memcpy(&buf2, src + n - 8, 8);
|
||
|
if (nullify_tail) {
|
||
|
memset(dst + 8, 0, 8);
|
||
|
}
|
||
|
memcpy(dst, &buf1, 8);
|
||
|
memcpy(dst + n - 8, &buf2, 8);
|
||
|
} else if (n >= 4) {
|
||
|
uint32_t buf1;
|
||
|
uint32_t buf2;
|
||
|
memcpy(&buf1, src, 4);
|
||
|
memcpy(&buf2, src + n - 4, 4);
|
||
|
if (nullify_tail) {
|
||
|
memset(dst + 4, 0, 4);
|
||
|
memset(dst + 8, 0, 8);
|
||
|
}
|
||
|
memcpy(dst, &buf1, 4);
|
||
|
memcpy(dst + n - 4, &buf2, 4);
|
||
|
} else {
|
||
|
if (n != 0) {
|
||
|
dst[0] = src[0];
|
||
|
dst[n / 2] = src[n / 2];
|
||
|
dst[n - 1] = src[n - 1];
|
||
|
}
|
||
|
if (nullify_tail) {
|
||
|
memset(dst + 8, 0, 8);
|
||
|
memset(dst + n, 0, 8);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Does non-template-specific `CordRepExternal` initialization.
|
||
|
// Expects `data` to be non-empty.
|
||
|
void InitializeCordRepExternal(absl::string_view data, CordRepExternal* rep);
|
||
|
|
||
|
// Creates a new `CordRep` that owns `data` and `releaser` and returns a pointer
|
||
|
// to it, or `nullptr` if `data` was empty.
|
||
|
template <typename Releaser>
|
||
|
// NOLINTNEXTLINE - suppress clang-tidy raw pointer return.
|
||
|
CordRep* NewExternalRep(absl::string_view data, Releaser&& releaser) {
|
||
|
using ReleaserType = absl::decay_t<Releaser>;
|
||
|
if (data.empty()) {
|
||
|
// Never create empty external nodes.
|
||
|
InvokeReleaser(Rank0{}, ReleaserType(std::forward<Releaser>(releaser)),
|
||
|
data);
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
CordRepExternal* rep = new CordRepExternalImpl<ReleaserType>(
|
||
|
std::forward<Releaser>(releaser), 0);
|
||
|
InitializeCordRepExternal(data, rep);
|
||
|
return rep;
|
||
|
}
|
||
|
|
||
|
// Overload for function reference types that dispatches using a function
|
||
|
// pointer because there are no `alignof()` or `sizeof()` a function reference.
|
||
|
// NOLINTNEXTLINE - suppress clang-tidy raw pointer return.
|
||
|
inline CordRep* NewExternalRep(absl::string_view data,
|
||
|
void (&releaser)(absl::string_view)) {
|
||
|
return NewExternalRep(data, &releaser);
|
||
|
}
|
||
|
|
||
|
} // namespace cord_internal
|
||
|
|
||
|
template <typename Releaser>
|
||
|
Cord MakeCordFromExternal(absl::string_view data, Releaser&& releaser) {
|
||
|
Cord cord;
|
||
|
if (auto* rep = ::absl::cord_internal::NewExternalRep(
|
||
|
data, std::forward<Releaser>(releaser))) {
|
||
|
cord.contents_.EmplaceTree(rep,
|
||
|
Cord::MethodIdentifier::kMakeCordFromExternal);
|
||
|
}
|
||
|
return cord;
|
||
|
}
|
||
|
|
||
|
constexpr Cord::InlineRep::InlineRep(cord_internal::InlineData data)
|
||
|
: data_(data) {}
|
||
|
|
||
|
inline Cord::InlineRep::InlineRep(const Cord::InlineRep& src)
|
||
|
: data_(InlineData::kDefaultInit) {
|
||
|
if (CordRep* tree = src.tree()) {
|
||
|
EmplaceTree(CordRep::Ref(tree), src.data_,
|
||
|
CordzUpdateTracker::kConstructorCord);
|
||
|
} else {
|
||
|
data_ = src.data_;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline Cord::InlineRep::InlineRep(Cord::InlineRep&& src) : data_(src.data_) {
|
||
|
src.ResetToEmpty();
|
||
|
}
|
||
|
|
||
|
inline Cord::InlineRep& Cord::InlineRep::operator=(const Cord::InlineRep& src) {
|
||
|
if (this == &src) {
|
||
|
return *this;
|
||
|
}
|
||
|
if (!is_tree() && !src.is_tree()) {
|
||
|
data_ = src.data_;
|
||
|
return *this;
|
||
|
}
|
||
|
AssignSlow(src);
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
inline Cord::InlineRep& Cord::InlineRep::operator=(
|
||
|
Cord::InlineRep&& src) noexcept {
|
||
|
if (is_tree()) {
|
||
|
UnrefTree();
|
||
|
}
|
||
|
data_ = src.data_;
|
||
|
src.ResetToEmpty();
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
inline void Cord::InlineRep::Swap(Cord::InlineRep* rhs) {
|
||
|
if (rhs == this) {
|
||
|
return;
|
||
|
}
|
||
|
std::swap(data_, rhs->data_);
|
||
|
}
|
||
|
|
||
|
inline const char* Cord::InlineRep::data() const {
|
||
|
return is_tree() ? nullptr : data_.as_chars();
|
||
|
}
|
||
|
|
||
|
inline absl::cord_internal::CordRep* Cord::InlineRep::as_tree() const {
|
||
|
assert(data_.is_tree());
|
||
|
return data_.as_tree();
|
||
|
}
|
||
|
|
||
|
inline absl::cord_internal::CordRep* Cord::InlineRep::tree() const {
|
||
|
if (is_tree()) {
|
||
|
return as_tree();
|
||
|
} else {
|
||
|
return nullptr;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline bool Cord::InlineRep::empty() const { return data_.is_empty(); }
|
||
|
|
||
|
inline size_t Cord::InlineRep::size() const {
|
||
|
return is_tree() ? as_tree()->length : inline_size();
|
||
|
}
|
||
|
|
||
|
inline cord_internal::CordRepFlat* Cord::InlineRep::MakeFlatWithExtraCapacity(
|
||
|
size_t extra) {
|
||
|
static_assert(cord_internal::kMinFlatLength >= sizeof(data_), "");
|
||
|
size_t len = data_.inline_size();
|
||
|
auto* result = CordRepFlat::New(len + extra);
|
||
|
result->length = len;
|
||
|
memcpy(result->Data(), data_.as_chars(), sizeof(data_));
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
inline void Cord::InlineRep::EmplaceTree(CordRep* rep,
|
||
|
MethodIdentifier method) {
|
||
|
assert(rep);
|
||
|
data_.make_tree(rep);
|
||
|
CordzInfo::MaybeTrackCord(data_, method);
|
||
|
}
|
||
|
|
||
|
inline void Cord::InlineRep::EmplaceTree(CordRep* rep, const InlineData& parent,
|
||
|
MethodIdentifier method) {
|
||
|
data_.make_tree(rep);
|
||
|
CordzInfo::MaybeTrackCord(data_, parent, method);
|
||
|
}
|
||
|
|
||
|
inline void Cord::InlineRep::SetTree(CordRep* rep,
|
||
|
const CordzUpdateScope& scope) {
|
||
|
assert(rep);
|
||
|
assert(data_.is_tree());
|
||
|
data_.set_tree(rep);
|
||
|
scope.SetCordRep(rep);
|
||
|
}
|
||
|
|
||
|
inline void Cord::InlineRep::SetTreeOrEmpty(CordRep* rep,
|
||
|
const CordzUpdateScope& scope) {
|
||
|
assert(data_.is_tree());
|
||
|
if (rep) {
|
||
|
data_.set_tree(rep);
|
||
|
} else {
|
||
|
data_ = {};
|
||
|
}
|
||
|
scope.SetCordRep(rep);
|
||
|
}
|
||
|
|
||
|
inline void Cord::InlineRep::CommitTree(const CordRep* old_rep, CordRep* rep,
|
||
|
const CordzUpdateScope& scope,
|
||
|
MethodIdentifier method) {
|
||
|
if (old_rep) {
|
||
|
SetTree(rep, scope);
|
||
|
} else {
|
||
|
EmplaceTree(rep, method);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline absl::cord_internal::CordRep* Cord::InlineRep::clear() {
|
||
|
if (is_tree()) {
|
||
|
CordzInfo::MaybeUntrackCord(cordz_info());
|
||
|
}
|
||
|
absl::cord_internal::CordRep* result = tree();
|
||
|
ResetToEmpty();
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
inline void Cord::InlineRep::CopyToArray(char* dst) const {
|
||
|
assert(!is_tree());
|
||
|
size_t n = inline_size();
|
||
|
assert(n != 0);
|
||
|
cord_internal::SmallMemmove(dst, data_.as_chars(), n);
|
||
|
}
|
||
|
|
||
|
constexpr inline Cord::Cord() noexcept {}
|
||
|
|
||
|
inline Cord::Cord(absl::string_view src)
|
||
|
: Cord(src, CordzUpdateTracker::kConstructorString) {}
|
||
|
|
||
|
template <typename T>
|
||
|
constexpr Cord::Cord(strings_internal::StringConstant<T>)
|
||
|
: contents_(strings_internal::StringConstant<T>::value.size() <=
|
||
|
cord_internal::kMaxInline
|
||
|
? cord_internal::InlineData(
|
||
|
strings_internal::StringConstant<T>::value)
|
||
|
: cord_internal::InlineData(
|
||
|
&cord_internal::ConstInitExternalStorage<
|
||
|
strings_internal::StringConstant<T>>::value)) {}
|
||
|
|
||
|
inline Cord& Cord::operator=(const Cord& x) {
|
||
|
contents_ = x.contents_;
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
template <typename T, Cord::EnableIfString<T>>
|
||
|
Cord& Cord::operator=(T&& src) {
|
||
|
if (src.size() <= cord_internal::kMaxBytesToCopy) {
|
||
|
return operator=(absl::string_view(src));
|
||
|
} else {
|
||
|
return AssignLargeString(std::forward<T>(src));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline Cord::Cord(const Cord& src) : contents_(src.contents_) {}
|
||
|
|
||
|
inline Cord::Cord(Cord&& src) noexcept : contents_(std::move(src.contents_)) {}
|
||
|
|
||
|
inline void Cord::swap(Cord& other) noexcept {
|
||
|
contents_.Swap(&other.contents_);
|
||
|
}
|
||
|
|
||
|
inline Cord& Cord::operator=(Cord&& x) noexcept {
|
||
|
contents_ = std::move(x.contents_);
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
extern template Cord::Cord(std::string&& src);
|
||
|
|
||
|
inline size_t Cord::size() const {
|
||
|
// Length is 1st field in str.rep_
|
||
|
return contents_.size();
|
||
|
}
|
||
|
|
||
|
inline bool Cord::empty() const { return contents_.empty(); }
|
||
|
|
||
|
inline size_t Cord::EstimatedMemoryUsage() const {
|
||
|
size_t result = sizeof(Cord);
|
||
|
if (const absl::cord_internal::CordRep* rep = contents_.tree()) {
|
||
|
result += MemoryUsageAux(rep);
|
||
|
}
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
inline absl::optional<absl::string_view> Cord::TryFlat() const {
|
||
|
absl::cord_internal::CordRep* rep = contents_.tree();
|
||
|
if (rep == nullptr) {
|
||
|
return absl::string_view(contents_.data(), contents_.size());
|
||
|
}
|
||
|
absl::string_view fragment;
|
||
|
if (GetFlatAux(rep, &fragment)) {
|
||
|
return fragment;
|
||
|
}
|
||
|
return absl::nullopt;
|
||
|
}
|
||
|
|
||
|
inline absl::string_view Cord::Flatten() {
|
||
|
absl::cord_internal::CordRep* rep = contents_.tree();
|
||
|
if (rep == nullptr) {
|
||
|
return absl::string_view(contents_.data(), contents_.size());
|
||
|
} else {
|
||
|
absl::string_view already_flat_contents;
|
||
|
if (GetFlatAux(rep, &already_flat_contents)) {
|
||
|
return already_flat_contents;
|
||
|
}
|
||
|
}
|
||
|
return FlattenSlowPath();
|
||
|
}
|
||
|
|
||
|
inline void Cord::Append(absl::string_view src) {
|
||
|
contents_.AppendArray(src, CordzUpdateTracker::kAppendString);
|
||
|
}
|
||
|
|
||
|
extern template void Cord::Append(std::string&& src);
|
||
|
extern template void Cord::Prepend(std::string&& src);
|
||
|
|
||
|
inline int Cord::Compare(const Cord& rhs) const {
|
||
|
if (!contents_.is_tree() && !rhs.contents_.is_tree()) {
|
||
|
return contents_.BitwiseCompare(rhs.contents_);
|
||
|
}
|
||
|
|
||
|
return CompareImpl(rhs);
|
||
|
}
|
||
|
|
||
|
// Does 'this' cord start/end with rhs
|
||
|
inline bool Cord::StartsWith(const Cord& rhs) const {
|
||
|
if (contents_.IsSame(rhs.contents_)) return true;
|
||
|
size_t rhs_size = rhs.size();
|
||
|
if (size() < rhs_size) return false;
|
||
|
return EqualsImpl(rhs, rhs_size);
|
||
|
}
|
||
|
|
||
|
inline bool Cord::StartsWith(absl::string_view rhs) const {
|
||
|
size_t rhs_size = rhs.size();
|
||
|
if (size() < rhs_size) return false;
|
||
|
return EqualsImpl(rhs, rhs_size);
|
||
|
}
|
||
|
|
||
|
inline void Cord::ChunkIterator::InitTree(cord_internal::CordRep* tree) {
|
||
|
if (tree->tag == cord_internal::BTREE) {
|
||
|
current_chunk_ = btree_reader_.Init(tree->btree());
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
stack_of_right_children_.push_back(tree);
|
||
|
operator++();
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator::ChunkIterator(cord_internal::CordRep* tree)
|
||
|
: bytes_remaining_(tree->length) {
|
||
|
InitTree(tree);
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator::ChunkIterator(const Cord* cord)
|
||
|
: bytes_remaining_(cord->size()) {
|
||
|
if (cord->contents_.is_tree()) {
|
||
|
InitTree(cord->contents_.as_tree());
|
||
|
} else {
|
||
|
current_chunk_ =
|
||
|
absl::string_view(cord->contents_.data(), bytes_remaining_);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator& Cord::ChunkIterator::AdvanceBtree() {
|
||
|
current_chunk_ = btree_reader_.Next();
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
inline void Cord::ChunkIterator::AdvanceBytesBtree(size_t n) {
|
||
|
assert(n >= current_chunk_.size());
|
||
|
bytes_remaining_ -= n;
|
||
|
if (bytes_remaining_) {
|
||
|
if (n == current_chunk_.size()) {
|
||
|
current_chunk_ = btree_reader_.Next();
|
||
|
} else {
|
||
|
size_t offset = btree_reader_.length() - bytes_remaining_;
|
||
|
current_chunk_ = btree_reader_.Seek(offset);
|
||
|
}
|
||
|
} else {
|
||
|
current_chunk_ = {};
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator& Cord::ChunkIterator::operator++() {
|
||
|
ABSL_HARDENING_ASSERT(bytes_remaining_ > 0 &&
|
||
|
"Attempted to iterate past `end()`");
|
||
|
assert(bytes_remaining_ >= current_chunk_.size());
|
||
|
bytes_remaining_ -= current_chunk_.size();
|
||
|
if (bytes_remaining_ > 0) {
|
||
|
return btree_reader_ ? AdvanceBtree() : AdvanceStack();
|
||
|
} else {
|
||
|
current_chunk_ = {};
|
||
|
}
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator Cord::ChunkIterator::operator++(int) {
|
||
|
ChunkIterator tmp(*this);
|
||
|
operator++();
|
||
|
return tmp;
|
||
|
}
|
||
|
|
||
|
inline bool Cord::ChunkIterator::operator==(const ChunkIterator& other) const {
|
||
|
return bytes_remaining_ == other.bytes_remaining_;
|
||
|
}
|
||
|
|
||
|
inline bool Cord::ChunkIterator::operator!=(const ChunkIterator& other) const {
|
||
|
return !(*this == other);
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator::reference Cord::ChunkIterator::operator*() const {
|
||
|
ABSL_HARDENING_ASSERT(bytes_remaining_ != 0);
|
||
|
return current_chunk_;
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator::pointer Cord::ChunkIterator::operator->() const {
|
||
|
ABSL_HARDENING_ASSERT(bytes_remaining_ != 0);
|
||
|
return ¤t_chunk_;
|
||
|
}
|
||
|
|
||
|
inline void Cord::ChunkIterator::RemoveChunkPrefix(size_t n) {
|
||
|
assert(n < current_chunk_.size());
|
||
|
current_chunk_.remove_prefix(n);
|
||
|
bytes_remaining_ -= n;
|
||
|
}
|
||
|
|
||
|
inline void Cord::ChunkIterator::AdvanceBytes(size_t n) {
|
||
|
assert(bytes_remaining_ >= n);
|
||
|
if (ABSL_PREDICT_TRUE(n < current_chunk_.size())) {
|
||
|
RemoveChunkPrefix(n);
|
||
|
} else if (n != 0) {
|
||
|
btree_reader_ ? AdvanceBytesBtree(n) : AdvanceBytesSlowPath(n);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator Cord::chunk_begin() const {
|
||
|
return ChunkIterator(this);
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator Cord::chunk_end() const { return ChunkIterator(); }
|
||
|
|
||
|
inline Cord::ChunkIterator Cord::ChunkRange::begin() const {
|
||
|
return cord_->chunk_begin();
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkIterator Cord::ChunkRange::end() const {
|
||
|
return cord_->chunk_end();
|
||
|
}
|
||
|
|
||
|
inline Cord::ChunkRange Cord::Chunks() const { return ChunkRange(this); }
|
||
|
|
||
|
inline Cord::CharIterator& Cord::CharIterator::operator++() {
|
||
|
if (ABSL_PREDICT_TRUE(chunk_iterator_->size() > 1)) {
|
||
|
chunk_iterator_.RemoveChunkPrefix(1);
|
||
|
} else {
|
||
|
++chunk_iterator_;
|
||
|
}
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
inline Cord::CharIterator Cord::CharIterator::operator++(int) {
|
||
|
CharIterator tmp(*this);
|
||
|
operator++();
|
||
|
return tmp;
|
||
|
}
|
||
|
|
||
|
inline bool Cord::CharIterator::operator==(const CharIterator& other) const {
|
||
|
return chunk_iterator_ == other.chunk_iterator_;
|
||
|
}
|
||
|
|
||
|
inline bool Cord::CharIterator::operator!=(const CharIterator& other) const {
|
||
|
return !(*this == other);
|
||
|
}
|
||
|
|
||
|
inline Cord::CharIterator::reference Cord::CharIterator::operator*() const {
|
||
|
return *chunk_iterator_->data();
|
||
|
}
|
||
|
|
||
|
inline Cord::CharIterator::pointer Cord::CharIterator::operator->() const {
|
||
|
return chunk_iterator_->data();
|
||
|
}
|
||
|
|
||
|
inline Cord Cord::AdvanceAndRead(CharIterator* it, size_t n_bytes) {
|
||
|
assert(it != nullptr);
|
||
|
return it->chunk_iterator_.AdvanceAndReadBytes(n_bytes);
|
||
|
}
|
||
|
|
||
|
inline void Cord::Advance(CharIterator* it, size_t n_bytes) {
|
||
|
assert(it != nullptr);
|
||
|
it->chunk_iterator_.AdvanceBytes(n_bytes);
|
||
|
}
|
||
|
|
||
|
inline absl::string_view Cord::ChunkRemaining(const CharIterator& it) {
|
||
|
return *it.chunk_iterator_;
|
||
|
}
|
||
|
|
||
|
inline Cord::CharIterator Cord::char_begin() const {
|
||
|
return CharIterator(this);
|
||
|
}
|
||
|
|
||
|
inline Cord::CharIterator Cord::char_end() const { return CharIterator(); }
|
||
|
|
||
|
inline Cord::CharIterator Cord::CharRange::begin() const {
|
||
|
return cord_->char_begin();
|
||
|
}
|
||
|
|
||
|
inline Cord::CharIterator Cord::CharRange::end() const {
|
||
|
return cord_->char_end();
|
||
|
}
|
||
|
|
||
|
inline Cord::CharRange Cord::Chars() const { return CharRange(this); }
|
||
|
|
||
|
inline void Cord::ForEachChunk(
|
||
|
absl::FunctionRef<void(absl::string_view)> callback) const {
|
||
|
absl::cord_internal::CordRep* rep = contents_.tree();
|
||
|
if (rep == nullptr) {
|
||
|
callback(absl::string_view(contents_.data(), contents_.size()));
|
||
|
} else {
|
||
|
return ForEachChunkAux(rep, callback);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Nonmember Cord-to-Cord relational operarators.
|
||
|
inline bool operator==(const Cord& lhs, const Cord& rhs) {
|
||
|
if (lhs.contents_.IsSame(rhs.contents_)) return true;
|
||
|
size_t rhs_size = rhs.size();
|
||
|
if (lhs.size() != rhs_size) return false;
|
||
|
return lhs.EqualsImpl(rhs, rhs_size);
|
||
|
}
|
||
|
|
||
|
inline bool operator!=(const Cord& x, const Cord& y) { return !(x == y); }
|
||
|
inline bool operator<(const Cord& x, const Cord& y) {
|
||
|
return x.Compare(y) < 0;
|
||
|
}
|
||
|
inline bool operator>(const Cord& x, const Cord& y) {
|
||
|
return x.Compare(y) > 0;
|
||
|
}
|
||
|
inline bool operator<=(const Cord& x, const Cord& y) {
|
||
|
return x.Compare(y) <= 0;
|
||
|
}
|
||
|
inline bool operator>=(const Cord& x, const Cord& y) {
|
||
|
return x.Compare(y) >= 0;
|
||
|
}
|
||
|
|
||
|
// Nonmember Cord-to-absl::string_view relational operators.
|
||
|
//
|
||
|
// Due to implicit conversions, these also enable comparisons of Cord with
|
||
|
// with std::string, ::string, and const char*.
|
||
|
inline bool operator==(const Cord& lhs, absl::string_view rhs) {
|
||
|
size_t lhs_size = lhs.size();
|
||
|
size_t rhs_size = rhs.size();
|
||
|
if (lhs_size != rhs_size) return false;
|
||
|
return lhs.EqualsImpl(rhs, rhs_size);
|
||
|
}
|
||
|
|
||
|
inline bool operator==(absl::string_view x, const Cord& y) { return y == x; }
|
||
|
inline bool operator!=(const Cord& x, absl::string_view y) { return !(x == y); }
|
||
|
inline bool operator!=(absl::string_view x, const Cord& y) { return !(x == y); }
|
||
|
inline bool operator<(const Cord& x, absl::string_view y) {
|
||
|
return x.Compare(y) < 0;
|
||
|
}
|
||
|
inline bool operator<(absl::string_view x, const Cord& y) {
|
||
|
return y.Compare(x) > 0;
|
||
|
}
|
||
|
inline bool operator>(const Cord& x, absl::string_view y) { return y < x; }
|
||
|
inline bool operator>(absl::string_view x, const Cord& y) { return y < x; }
|
||
|
inline bool operator<=(const Cord& x, absl::string_view y) { return !(y < x); }
|
||
|
inline bool operator<=(absl::string_view x, const Cord& y) { return !(y < x); }
|
||
|
inline bool operator>=(const Cord& x, absl::string_view y) { return !(x < y); }
|
||
|
inline bool operator>=(absl::string_view x, const Cord& y) { return !(x < y); }
|
||
|
|
||
|
// Some internals exposed to test code.
|
||
|
namespace strings_internal {
|
||
|
class CordTestAccess {
|
||
|
public:
|
||
|
static size_t FlatOverhead();
|
||
|
static size_t MaxFlatLength();
|
||
|
static size_t SizeofCordRepConcat();
|
||
|
static size_t SizeofCordRepExternal();
|
||
|
static size_t SizeofCordRepSubstring();
|
||
|
static size_t FlatTagToLength(uint8_t tag);
|
||
|
static uint8_t LengthToTag(size_t s);
|
||
|
};
|
||
|
} // namespace strings_internal
|
||
|
ABSL_NAMESPACE_END
|
||
|
} // namespace absl
|
||
|
|
||
|
#endif // ABSL_STRINGS_CORD_H_
|