//////////////////////////////////////////////////////////////////////////////// /// @brief V8-vocbase bridge /// /// @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 Copyright 2014, ArangoDB GmbH, Cologne, Germany /// @author Copyright 2011-2013, triAGENS GmbH, Cologne, Germany //////////////////////////////////////////////////////////////////////////////// #include "v8-collection.h" #include "v8-vocbaseprivate.h" #include "Basics/conversions.h" #include "V8/v8-conv.h" using namespace std; using namespace triagens::basics; using namespace triagens::arango; using namespace triagens::rest; //////////////////////////////////////////////////////////////////////////////// /// @brief slot for a "collection" //////////////////////////////////////////////////////////////////////////////// static int const SLOT_COLLECTION = 2; //////////////////////////////////////////////////////////////////////////////// /// @brief free a coordinator collection //////////////////////////////////////////////////////////////////////////////// void FreeCoordinatorCollection (TRI_vocbase_col_t* collection) { TRI_DestroyReadWriteLock(&collection->_lock); TRI_Free(TRI_UNKNOWN_MEM_ZONE, collection); } //////////////////////////////////////////////////////////////////////////////// /// @brief releases a collection //////////////////////////////////////////////////////////////////////////////// void ReleaseCollection (TRI_vocbase_col_t const* collection) { TRI_ReleaseCollectionVocBase(collection->_vocbase, const_cast(collection)); } //////////////////////////////////////////////////////////////////////////////// /// @brief convert a collection info into a TRI_vocbase_col_t //////////////////////////////////////////////////////////////////////////////// TRI_vocbase_col_t* CoordinatorCollection (TRI_vocbase_t* vocbase, CollectionInfo const& ci) { TRI_vocbase_col_t* c = static_cast(TRI_Allocate(TRI_UNKNOWN_MEM_ZONE, sizeof(TRI_vocbase_col_t), false)); if (c == nullptr) { return nullptr; } c->_internalVersion = 0; c->_isLocal = false; c->_vocbase = vocbase; c->_type = ci.type(); c->_cid = ci.id(); c->_planId = ci.id(); c->_status = ci.status(); c->_collection = nullptr; std::string const name = ci.name(); memset(c->_name, 0, TRI_COL_NAME_LENGTH + 1); memcpy(c->_name, name.c_str(), name.size()); memset(c->_path, 0, TRI_COL_PATH_LENGTH + 1); memset(c->_dbName, 0, TRI_COL_NAME_LENGTH + 1); memcpy(c->_dbName, vocbase->_name, strlen(vocbase->_name)); c->_canDrop = true; c->_canUnload = true; c->_canRename = true; if (TRI_IsSystemNameCollection(c->_name)) { // a few system collections have special behavior if (TRI_EqualString(c->_name, TRI_COL_NAME_USERS) || TRI_IsPrefixString(c->_name, TRI_COL_NAME_STATISTICS)) { // these collections cannot be dropped or renamed c->_canDrop = false; c->_canRename = false; } } TRI_InitReadWriteLock(&c->_lock); return c; } /////////////////////////////////////////////////////////////////////////////// /// @brief check if a name belongs to a collection //////////////////////////////////////////////////////////////////////////////// bool EqualCollection (CollectionNameResolver const* resolver, string const& collectionName, TRI_vocbase_col_t const* collection) { if (collectionName == StringUtils::itoa(collection->_cid)) { return true; } if (collectionName == string(collection->_name)) { return true; } if (ServerState::instance()->isCoordinator()) { if (collectionName == resolver->getCollectionNameCluster(collection->_cid)) { return true; } return false; } if (collectionName == resolver->getCollectionName(collection->_cid)) { return true; } return false; } //////////////////////////////////////////////////////////////////////////////// /// @brief create a v8 collection id value from the internal collection id //////////////////////////////////////////////////////////////////////////////// static inline v8::Handle V8CollectionId (v8::Isolate* isolate, TRI_voc_cid_t cid) { char buffer[21]; size_t len = TRI_StringUInt64InPlace((uint64_t) cid, (char*) &buffer); return TRI_V8_PAIR_STRING((const char*) buffer, (int) len); } //////////////////////////////////////////////////////////////////////////////// /// @brief weak reference callback for collections //////////////////////////////////////////////////////////////////////////////// static void WeakCollectionCallback (const v8::WeakCallbackData>& data) { auto isolate = data.GetIsolate(); auto persistent = data.GetParameter(); auto myCollection = v8::Local::New(isolate, *persistent); auto collection = static_cast(myCollection->Value()); TRI_GET_GLOBALS(); v8g->decreaseActiveExternals(); // decrease the reference-counter for the database TRI_ReleaseVocBase(collection->_vocbase); // find the persistent handle #if TRI_ENABLE_MAINTAINER_MODE auto const& it = v8g->JSCollections.find(collection); TRI_ASSERT(it != v8g->JSCollections.end()) #endif if (! collection->_isLocal) { FreeCoordinatorCollection(collection); } // dispose and clear the persistent handle v8g->JSCollections[collection].Reset(); v8g->JSCollections.erase(collection); } ///////////////////////////////////////////////////////////////////////////////// /// @brief wraps a TRI_vocbase_col_t //////////////////////////////////////////////////////////////////////////////// v8::Handle WrapCollection (v8::Isolate* isolate, TRI_vocbase_col_t const* collection) { v8::EscapableHandleScope scope(isolate); TRI_GET_GLOBALS(); TRI_GET_GLOBAL(VocbaseColTempl, v8::ObjectTemplate); v8::Handle result = VocbaseColTempl->NewInstance(); if (! result.IsEmpty()) { TRI_vocbase_col_t* nonconstCollection = const_cast(collection); result->SetInternalField(SLOT_CLASS_TYPE, v8::Integer::New(isolate, WRP_VOCBASE_COL_TYPE)); result->SetInternalField(SLOT_CLASS, v8::External::New(isolate, nonconstCollection)); auto const& it = v8g->JSCollections.find(nonconstCollection); if (it == v8g->JSCollections.end()) { // increase the reference-counter for the database TRI_UseVocBase(collection->_vocbase); auto externalCollection = v8::External::New(isolate, nonconstCollection); result->SetInternalField(SLOT_COLLECTION, externalCollection); v8g->JSCollections[nonconstCollection].Reset(isolate, externalCollection); v8g->JSCollections[nonconstCollection].SetWeak(&v8g->JSCollections[nonconstCollection], WeakCollectionCallback); v8g->increaseActiveExternals(); } else { auto myCollection = v8::Local::New(isolate, it->second); result->SetInternalField(SLOT_COLLECTION, myCollection); } TRI_GET_GLOBAL_STRING(_IdKey); TRI_GET_GLOBAL_STRING(_DbNameKey); TRI_GET_GLOBAL_STRING(VersionKeyHidden); result->ForceSet(_IdKey, V8CollectionId(isolate, collection->_cid), v8::ReadOnly); result->Set(_DbNameKey, TRI_V8_STRING(collection->_dbName)); result->ForceSet(VersionKeyHidden, v8::Integer::NewFromUnsigned(isolate, collection->_internalVersion), v8::DontEnum); } return scope.Escape(result); }