1
0
Fork 0
arangodb/arangod/VocBase/VocShaper.h

336 lines
14 KiB
C++

////////////////////////////////////////////////////////////////////////////////
/// @brief json shaper used to compute the shape of an json object
///
/// @file
///
/// DISCLAIMER
///
/// Copyright 2014 ArangoDB GmbH, Cologne, Germany
/// Copyright 2004-2014 triAGENS GmbH, Cologne, Germany
///
/// 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.
///
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
///
/// @author Dr. Frank Celler
/// @author Martin Schoenert
/// @author Copyright 2014, ArangoDB GmbH, Cologne, Germany
/// @author Copyright 2006-2013, triAGENS GmbH, Cologne, Germany
////////////////////////////////////////////////////////////////////////////////
#ifndef ARANGODB_VOC_BASE_VOC_SHAPER_H
#define ARANGODB_VOC_BASE_VOC_SHAPER_H 1
#include "Basics/Common.h"
#include "VocBase/datafile.h"
#include "VocBase/document-collection.h"
#include "VocBase/shape-accessor.h"
#include "VocBase/shaped-json.h"
#include "VocBase/Shaper.h"
#include "Wal/Marker.h"
#define NUM_SHAPE_ACCESSORS 8
// -----------------------------------------------------------------------------
// --SECTION-- VocShaper
// -----------------------------------------------------------------------------
class VocShaper : public Shaper {
public:
// -----------------------------------------------------------------------------
// --SECTION-- constructors and destructors
// -----------------------------------------------------------------------------
VocShaper (VocShaper const&);
VocShaper& operator= (VocShaper const&);
VocShaper (TRI_memory_zone_t*,
TRI_document_collection_t*);
~VocShaper ();
// -----------------------------------------------------------------------------
// --SECTION-- public methods
// -----------------------------------------------------------------------------
public:
////////////////////////////////////////////////////////////////////////////////
/// @brief return the shaper's memory zone
////////////////////////////////////////////////////////////////////////////////
TRI_memory_zone_t* memoryZone () const {
return _memoryZone;
}
////////////////////////////////////////////////////////////////////////////////
/// @brief looks up a shape by identifier
////////////////////////////////////////////////////////////////////////////////
TRI_shape_t const* lookupShapeId (TRI_shape_sid_t) override final;
////////////////////////////////////////////////////////////////////////////////
/// @brief looks up an attribute name by identifier
////////////////////////////////////////////////////////////////////////////////
char const* lookupAttributeId (TRI_shape_aid_t) override final;
////////////////////////////////////////////////////////////////////////////////
/// @brief looks up an attribute path by identifier
////////////////////////////////////////////////////////////////////////////////
TRI_shape_path_t const* lookupAttributePathByPid (TRI_shape_pid_t);
////////////////////////////////////////////////////////////////////////////////
/// @brief finds an attribute path by identifier
////////////////////////////////////////////////////////////////////////////////
TRI_shape_pid_t findOrCreateAttributePathByName (char const*);
////////////////////////////////////////////////////////////////////////////////
/// @brief looks up an attribute path by identifier
////////////////////////////////////////////////////////////////////////////////
TRI_shape_pid_t lookupAttributePathByName (char const*);
////////////////////////////////////////////////////////////////////////////////
/// @brief returns the attribute name for an attribute path
////////////////////////////////////////////////////////////////////////////////
char const* attributeNameShapePid (TRI_shape_pid_t);
////////////////////////////////////////////////////////////////////////////////
/// @brief looks up an attribute identifier by name
////////////////////////////////////////////////////////////////////////////////
TRI_shape_aid_t lookupAttributeByName (char const*);
////////////////////////////////////////////////////////////////////////////////
/// @brief finds or creates an attribute identifier by name
////////////////////////////////////////////////////////////////////////////////
TRI_shape_aid_t findOrCreateAttributeByName (char const*);
////////////////////////////////////////////////////////////////////////////////
/// @brief finds a shape
/// if the function returns non-nullptr, the return value is a pointer to an
/// already existing shape and the value must not be freed
/// if the function returns nullptr, it has not found the shape and was not able
/// to create it. The value must then be freed by the caller
////////////////////////////////////////////////////////////////////////////////
TRI_shape_t const* findShape (TRI_shape_t*,
bool);
////////////////////////////////////////////////////////////////////////////////
/// @brief move a shape marker, called during compaction
////////////////////////////////////////////////////////////////////////////////
int moveMarker (TRI_df_marker_t*,
void*);
////////////////////////////////////////////////////////////////////////////////
/// @brief insert a shape, called when opening a collection
////////////////////////////////////////////////////////////////////////////////
int insertShape (TRI_df_marker_t const*,
bool);
////////////////////////////////////////////////////////////////////////////////
/// @brief insert an attribute, called when opening a collection
////////////////////////////////////////////////////////////////////////////////
int insertAttribute (TRI_df_marker_t const*,
bool);
////////////////////////////////////////////////////////////////////////////////
/// @brief finds an accessor
////////////////////////////////////////////////////////////////////////////////
TRI_shape_access_t const* findAccessor (TRI_shape_sid_t,
TRI_shape_pid_t);
////////////////////////////////////////////////////////////////////////////////
/// @brief extracts a sub-shape
////////////////////////////////////////////////////////////////////////////////
bool extractShapedJson (TRI_shaped_json_t const* document,
TRI_shape_sid_t sid,
TRI_shape_pid_t pid,
TRI_shaped_json_t* result,
TRI_shape_t const** shape);
// -----------------------------------------------------------------------------
// --SECTION-- private methods
// -----------------------------------------------------------------------------
private:
////////////////////////////////////////////////////////////////////////////////
/// @brief looks up a shape path by identifier
////////////////////////////////////////////////////////////////////////////////
TRI_shape_path_t const* findShapePathByName (char const* name,
bool create);
// -----------------------------------------------------------------------------
// --SECTION-- private variables
// -----------------------------------------------------------------------------
private:
TRI_memory_zone_t* _memoryZone;
TRI_document_collection_t* _collection;
// attribute paths
triagens::basics::Mutex _attributePathsCreateLock;
triagens::basics::ReadWriteLock _attributePathsByNameLock;
TRI_associative_pointer_t _attributePathsByName;
triagens::basics::ReadWriteLock _attributePathsByPidLock;
TRI_associative_pointer_t _attributePathsByPid;
// attributes
triagens::basics::Mutex _attributeCreateLock;
triagens::basics::ReadWriteLock _attributeNamesLock;
TRI_associative_pointer_t _attributeNames;
triagens::basics::ReadWriteLock _attributeIdsLock;
TRI_associative_pointer_t _attributeIds;
// shapes
triagens::basics::Mutex _shapeCreateLock;
triagens::basics::ReadWriteLock _shapeDictionaryLock;
TRI_associative_pointer_t _shapeDictionary;
triagens::basics::ReadWriteLock _shapeIdsLock;
TRI_associative_pointer_t _shapeIds;
// accessors
triagens::basics::ReadWriteLock _accessorLock[NUM_SHAPE_ACCESSORS];
TRI_associative_pointer_t _accessors[NUM_SHAPE_ACCESSORS];
TRI_shape_pid_t _nextPid;
std::atomic<TRI_shape_aid_t> _nextAid;
std::atomic<TRI_shape_sid_t> _nextSid;
};
////////////////////////////////////////////////////////////////////////////////
/// @brief helper method for recursive shapes comparison
///
/// You must either supply (leftDocument, leftObject) or leftShaped.
/// You must either supply (rightDocument, rightObject) or rightShaped.
////////////////////////////////////////////////////////////////////////////////
int TRI_CompareShapeTypes (char const* leftDocument,
TRI_shaped_sub_t const* leftObject,
TRI_shaped_json_t const* leftShaped,
VocShaper* leftShaper,
char const* rightDocument,
TRI_shaped_sub_t const* rightObject,
TRI_shaped_json_t const* rightShaped,
VocShaper* rightShaper);
////////////////////////////////////////////////////////////////////////////////
/// @brief extracts the shape identifier pointer from a marker
////////////////////////////////////////////////////////////////////////////////
static inline void TRI_EXTRACT_SHAPE_IDENTIFIER_MARKER (TRI_shape_sid_t& dst,
void const* src) {
auto type = static_cast<TRI_df_marker_t const*>(src)->_type;
if (type == TRI_DOC_MARKER_KEY_DOCUMENT) {
dst = static_cast<TRI_doc_document_key_marker_t const*>(src)->_shape;
}
else if (type == TRI_DOC_MARKER_KEY_EDGE) {
dst = static_cast<TRI_doc_edge_key_marker_t const*>(src)->base._shape;
}
else if (type == TRI_WAL_MARKER_DOCUMENT) {
dst = static_cast<triagens::wal::document_marker_t const*>(src)->_shape;
}
else if (type == TRI_WAL_MARKER_EDGE) {
dst = static_cast<triagens::wal::edge_marker_t const*>(src)->_shape;
}
else {
dst = 0;
}
}
////////////////////////////////////////////////////////////////////////////////
/// @brief extracts the shaped JSON pointer from a marker
////////////////////////////////////////////////////////////////////////////////
static inline void TRI_EXTRACT_SHAPED_JSON_MARKER (TRI_shaped_json_t& dst,
void const* src) {
auto type = static_cast<TRI_df_marker_t const*>(src)->_type;
if (type == TRI_DOC_MARKER_KEY_DOCUMENT ||
type == TRI_DOC_MARKER_KEY_EDGE) {
dst._sid = static_cast<TRI_doc_document_key_marker_t const*>(src)->_shape;
dst._data.length = static_cast<TRI_df_marker_t const*>(src)->_size
- static_cast<TRI_doc_document_key_marker_t const*>(src)->_offsetJson;
dst._data.data = const_cast<char*>(static_cast<char const*>(src))
+ static_cast<TRI_doc_document_key_marker_t const*>(src)->_offsetJson;
}
else if (type == TRI_WAL_MARKER_DOCUMENT) {
dst._sid = static_cast<triagens::wal::document_marker_t const*>(src)->_shape;
dst._data.length = static_cast<TRI_df_marker_t const*>(src)->_size
- static_cast<triagens::wal::document_marker_t const*>(src)->_offsetJson;
dst._data.data = const_cast<char*>(static_cast<char const*>(src))
+ static_cast<triagens::wal::document_marker_t const*>(src)->_offsetJson;
}
else if (type == TRI_WAL_MARKER_EDGE) {
dst._sid = static_cast<triagens::wal::edge_marker_t const*>(src)->_shape;
dst._data.length = static_cast<TRI_df_marker_t const*>(src)->_size
- static_cast<triagens::wal::edge_marker_t const*>(src)->_offsetJson;
dst._data.data = const_cast<char*>(static_cast<char const*>(src))
+ static_cast<triagens::wal::edge_marker_t const*>(src)->_offsetJson;
}
else {
dst._sid = 0;
dst._data.length = 0;
dst._data.data = nullptr;
}
}
void TRI_InspectShapedSub (TRI_shaped_sub_t const*,
char const*,
TRI_shaped_json_t&);
void TRI_InspectShapedSub (TRI_shaped_sub_t const*,
TRI_doc_mptr_t const*,
char const*&,
size_t&);
void TRI_FillShapedSub (TRI_shaped_sub_t*,
TRI_shaped_json_t const*,
char const*);
#endif
// -----------------------------------------------------------------------------
// --SECTION-- END-OF-FILE
// -----------------------------------------------------------------------------
// Local Variables:
// mode: outline-minor
// outline-regexp: "/// @brief\\|/// {@inheritDoc}\\|/// @page\\|// --SECTION--\\|/// @\\}"
// End: