1
0
Fork 0
arangodb/arangod/RestServer/QueryRegistryFeature.h

86 lines
2.7 KiB
C++

////////////////////////////////////////////////////////////////////////////////
/// 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"
namespace arangodb {
namespace aql {
class QueryRegistry;
}
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 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; }
uint64_t queryMemoryLimit() const { return _queryMemoryLimit; }
uint64_t maxQueryPlans() const { return _maxQueryPlans; }
private:
bool _trackSlowQueries;
bool _trackBindVars;
bool _failOnWarning;
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