arangodb/arangod/VocBase/ExampleMatcher.cpp

////////////////////////////////////////////////////////////////////////////////
/// @brief Class to allow simple byExample matching of mptr.
///
/// @file
///
/// DISCLAIMER
///
/// Copyright 2014-2015 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 Michael Hackstein
/// @author Copyright 2014-2015, ArangoDB GmbH, Cologne, Germany
/// @author Copyright 2011-2013, triAGENS GmbH, Cologne, Germany
////////////////////////////////////////////////////////////////////////////////

#include "ExampleMatcher.h"
#include "voc-shaper.h"
#include "V8/v8-utils.h"
#include "V8/v8-conv.h"
#include "Basics/JsonHelper.h"

using namespace std;
using namespace triagens::arango;
using namespace triagens::basics;


////////////////////////////////////////////////////////////////////////////////
/// @brief cleans up the example object
////////////////////////////////////////////////////////////////////////////////

void ExampleMatcher::cleanup () {
  auto zone = _shaper->_memoryZone;
  // clean shaped json objects
  for (auto it : _values) {
    TRI_FreeShapedJson(zone, it);
  }
}

////////////////////////////////////////////////////////////////////////////////
/// @brief Constructor using a v8::Object example
////////////////////////////////////////////////////////////////////////////////
 
ExampleMatcher::ExampleMatcher (
  v8::Isolate* isolate,
  v8::Handle<v8::Object> const example,
  TRI_shaper_t* shaper,
  std::string& errorMessage
) : _shaper(shaper) {
  v8::Handle<v8::Array> names = example->GetOwnPropertyNames();
  size_t n = names->Length();

  _pids.reserve(n);
  _values.reserve(n);

  try { 
    for (size_t i = 0;  i < n;  ++i) {
      v8::Handle<v8::Value> key = names->Get((uint32_t) i);
      v8::Handle<v8::Value> val = example->Get(key);
      TRI_Utf8ValueNFC keyStr(TRI_UNKNOWN_MEM_ZONE, key);
      if (*keyStr != nullptr) {
        auto pid = _shaper->lookupAttributePathByName(_shaper, *keyStr);

        if (pid == 0) {
          // no attribute path found. this means the result will be empty
          ExampleMatcher::cleanup();
          throw TRI_RESULT_ELEMENT_NOT_FOUND;
        }
        _pids.push_back(pid);

        auto value = TRI_ShapedJsonV8Object(isolate, val, shaper, false);

        if (value == nullptr) {
          ExampleMatcher::cleanup();
          throw TRI_RESULT_ELEMENT_NOT_FOUND;
        }
        _values.push_back(value);
      }
      else {
        ExampleMatcher::cleanup();
        errorMessage = "cannot convert attribute path to UTF8";
        throw TRI_ERROR_BAD_PARAMETER;
      }
    }
  } catch (bad_alloc& e) {
    ExampleMatcher::cleanup();
    throw e;
  }

};


////////////////////////////////////////////////////////////////////////////////
/// @brief Constructor using a TRI_json_t object
////////////////////////////////////////////////////////////////////////////////

ExampleMatcher::ExampleMatcher (
  TRI_json_t* const example,
  TRI_shaper_t* shaper
) : _shaper(shaper) {
  // Just make sure we are talking about objects
  TRI_ASSERT(TRI_IsObjectJson(example));

  TRI_vector_t objects = example->_value._objects;

  // Trolololol std::vector in C... ;(
  size_t n = TRI_LengthVector(&objects); 

  _pids.reserve(n);
  _values.reserve(n);

  try { 
    for (size_t i = 0; i < n; i += 2) {
      TRI_json_t* keyobj = static_cast<TRI_json_t*>(TRI_AtVector(&objects, i));
      TRI_ASSERT(TRI_IsStringJson(keyobj));
      auto pid = _shaper->lookupAttributePathByName(_shaper, keyobj->_value._string.data);

      if (pid == 0) {
        // no attribute path found. this means the result will be empty
        ExampleMatcher::cleanup();
        throw TRI_RESULT_ELEMENT_NOT_FOUND;
      }

      _pids.push_back(pid);
      TRI_json_t* jsonvalue = static_cast<TRI_json_t*>(TRI_AtVector(&objects, i+1));
      if (TRI_IsStringJson(jsonvalue)) {
        cout << JsonHelper::toString(jsonvalue) << endl;
      }
      auto value = TRI_ShapedJsonJson(_shaper, jsonvalue, false);


      if (value == nullptr) {
        ExampleMatcher::cleanup();
        throw TRI_RESULT_ELEMENT_NOT_FOUND;
      }
      _values.push_back(value);
    }
  } catch (bad_alloc& e) {
    ExampleMatcher::cleanup();
    throw e;
  }
};




////////////////////////////////////////////////////////////////////////////////
/// @brief Checks if the given mptr matches the example in this class
////////////////////////////////////////////////////////////////////////////////

bool ExampleMatcher::matches (TRI_doc_mptr_t const* mptr) const {
  if (mptr == nullptr) {
    return false;
  }
  TRI_shaped_json_t document;
  TRI_shaped_json_t result;
  TRI_shape_t const* shape;

  TRI_EXTRACT_SHAPED_JSON_MARKER(document, mptr->getDataPtr());

  for (size_t i = 0;  i < _values.size();  ++i) {
    TRI_shaped_json_t* example = _values[i];

    bool ok = TRI_ExtractShapedJsonVocShaper(_shaper,
                                             &document,
                                             example->_sid,
                                             _pids[i],
                                             &result,
                                             &shape);

    if (! ok || shape == nullptr) {
      return false;
    }

    if (result._data.length != example->_data.length) {
      // suppress excessive log spam
      // LOG_TRACE("expecting length %lu, got length %lu for path %lu",
      //           (unsigned long) result._data.length,
      //           (unsigned long) example->_data.length,
      //           (unsigned long) pids[i]);

      return false;
    }

    if (memcmp(result._data.data, example->_data.data, example->_data.length) != 0) {
      // suppress excessive log spam
      // LOG_TRACE("data mismatch at path %lu", (unsigned long) pids[i]);
      return false;
    }
  }
  return true;
};