////////////////////////////////////////////////////////////////////////////////
/// DISCLAIMER
///
/// Copyright 2016 ArangoDB 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
////////////////////////////////////////////////////////////////////////////////

#ifndef APPLICATION_FEATURES_QUERY_REGISTRY_FEATUREx_H
#define APPLICATION_FEATURES_QUERY_REGISTRY_FEATUREx_H 1

#include "ApplicationFeatures/ApplicationFeature.h"
#include "Aql/QueryRegistry.h"

namespace arangodb {

class QueryRegistryFeature final : public application_features::ApplicationFeature {
 public:
  static aql::QueryRegistry* registry() {
    return QUERY_REGISTRY.load(std::memory_order_acquire);
  }

  explicit QueryRegistryFeature(application_features::ApplicationServer& server);

  void collectOptions(std::shared_ptr<options::ProgramOptions>) override final;
  void validateOptions(std::shared_ptr<options::ProgramOptions>) override final;
  void prepare() override final;
  void start() override final;
  void beginShutdown() override final;
  void stop() override final;
  void unprepare() override final;

  bool trackSlowQueries() const { return _trackSlowQueries; }
  bool trackBindVars() const { return _trackBindVars; }
  double slowQueryThreshold() const { return _slowQueryThreshold; }
  double slowStreamingQueryThreshold() const {
    return _slowStreamingQueryThreshold;
  }
  bool failOnWarning() const { return _failOnWarning; }
  bool smartJoins() const { return _smartJoins; }
  uint64_t queryMemoryLimit() const { return _queryMemoryLimit; }
  uint64_t maxQueryPlans() const { return _maxQueryPlans; }

 private:
  bool _trackSlowQueries;
  bool _trackBindVars;
  bool _failOnWarning;
  bool _smartJoins;
  uint64_t _queryMemoryLimit;
  uint64_t _maxQueryPlans;
  double _slowQueryThreshold;
  double _slowStreamingQueryThreshold;
  std::string _queryCacheMode;
  uint64_t _queryCacheMaxResultsCount;
  uint64_t _queryCacheMaxResultsSize;
  uint64_t _queryCacheMaxEntrySize;
  bool _queryCacheIncludeSystem;
  double _queryRegistryTTL;

 public:
  aql::QueryRegistry* queryRegistry() const { return _queryRegistry.get(); }

 private:
  static std::atomic<aql::QueryRegistry*> QUERY_REGISTRY;

  std::unique_ptr<aql::QueryRegistry> _queryRegistry;
};

}  // namespace arangodb

#endif