com.orientechnologies.enterprise.server.OEnterpriseServerImpl Maven / Gradle / Ivy
package com.orientechnologies.enterprise.server;
import com.orientechnologies.agent.OEnterpriseAgent;
import com.orientechnologies.agent.operation.NodesManager;
import com.orientechnologies.agent.profiler.metrics.OHistogram;
import com.orientechnologies.agent.profiler.metrics.OSnapshot;
import com.orientechnologies.agent.services.OEnterpriseService;
import com.orientechnologies.agent.services.metrics.OrientDBMetricsService;
import com.orientechnologies.agent.services.metrics.server.database.QueryInfo;
import com.orientechnologies.enterprise.server.listener.OEnterpriseConnectionListener;
import com.orientechnologies.enterprise.server.listener.OEnterpriseStorageListener;
import com.orientechnologies.orient.core.Orient;
import com.orientechnologies.orient.core.command.OCommandExecutor;
import com.orientechnologies.orient.core.command.OCommandRequestText;
import com.orientechnologies.orient.core.db.*;
import com.orientechnologies.orient.core.db.OSystemDatabase;
import com.orientechnologies.orient.core.db.document.OQueryDatabaseState;
import com.orientechnologies.orient.core.record.impl.ODocument;
import com.orientechnologies.orient.core.security.OSecuritySystem;
import com.orientechnologies.orient.core.sql.OSQLEngine;
import com.orientechnologies.orient.core.sql.executor.*;
import com.orientechnologies.orient.core.sql.functions.OSQLFunction;
import com.orientechnologies.orient.core.sql.parser.OLocalResultSet;
import com.orientechnologies.orient.core.sql.parser.OLocalResultSetLifecycleDecorator;
import com.orientechnologies.orient.core.storage.OStorage;
import com.orientechnologies.orient.core.storage.impl.local.paginated.OEnterpriseLocalPaginatedStorage;
import com.orientechnologies.orient.server.OClientConnection;
import com.orientechnologies.orient.server.OServer;
import com.orientechnologies.orient.server.OServerLifecycleListener;
import com.orientechnologies.orient.server.config.OServerParameterConfiguration;
import com.orientechnologies.orient.server.distributed.ODistributedServerManager;
import com.orientechnologies.orient.server.network.OServerNetworkListener;
import com.orientechnologies.orient.server.network.protocol.ONetworkProtocol;
import com.orientechnologies.orient.server.network.protocol.http.ONetworkProtocolHttpAbstract;
import com.orientechnologies.orient.server.network.protocol.http.command.OServerCommand;
import com.orientechnologies.orient.server.plugin.OServerPlugin;
import com.orientechnologies.orient.server.plugin.OServerPluginInfo;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Function;
import java.util.stream.Collectors;
/** Created by Enrico Risa on 16/07/2018. */
public class OEnterpriseServerImpl
implements OEnterpriseServer,
OServerPlugin,
ODatabaseLifecycleListener,
ODatabaseListener,
OServerLifecycleListener {
private final OEnterpriseAgent agent;
private OServer server;
private NodesManager nodesManager;
private List listeners = new ArrayList<>();
private List dbListeners = new ArrayList<>();
private Map storages = new ConcurrentHashMap<>();
public OEnterpriseServerImpl(final OServer server, final OEnterpriseAgent agent) {
this.server = server;
this.agent = agent;
server
.getPluginManager()
.registerPlugin(
new OServerPluginInfo("Enterprise Server", null, null, null, this, null, 0, null));
Orient.instance().addDbLifecycleListener(this);
this.server.registerLifecycleListener(this);
}
@Override
public void registerConnectionListener(OEnterpriseConnectionListener listener) {
listeners.add(listener);
}
@Override
public void unregisterConnectionListener(OEnterpriseConnectionListener listener) {
listeners.remove(listener);
}
@Override
public void registerDatabaseListener(OEnterpriseStorageListener listener) {
dbListeners.add(listener);
}
@Override
public void unRegisterDatabaseListener(OEnterpriseStorageListener listener) {
dbListeners.remove(listener);
}
@Override
public Map getAvailableStorageNames() {
return server.getAvailableStorageNames();
}
@Override
public String getName() {
return null;
}
@Override
public void startup() {}
@Override
public void shutdown() {
listeners.clear();
}
@Override
public OrientDBInternal getDatabases() {
return server.getDatabases();
}
@Override
public void interruptConnection(Integer connectionId) {
server.getClientConnectionManager().kill(connectionId);
}
@Override
public void onClientConnection(OClientConnection oClientConnection) {
this.listeners.forEach((l) -> l.onClientConnection(oClientConnection));
}
@Override
public void onSocketAccepted(ONetworkProtocol protocol) {
this.listeners.forEach((l) -> l.onSocketAccepted(protocol));
}
@Override
public void onSocketDestroyed(ONetworkProtocol protocol) {
this.listeners.forEach((l) -> l.onSocketDestroyed(protocol));
}
@Override
public OSystemDatabase getSystemDatabase() {
return server.getSystemDatabase();
}
public NodesManager getNodesManager() {
return nodesManager;
}
@Override
public ODistributedServerManager getDistributedManager() {
return server.getDistributedManager();
}
@Override
public boolean existsDatabase(String databaseName) {
return server.existsDatabase(databaseName);
}
@Override
public void restore(String databaseName, String path) {
server.restore(databaseName, path);
}
@Override
public void onClientDisconnection(OClientConnection oClientConnection) {
this.listeners.forEach((l) -> l.onClientDisconnection(oClientConnection));
}
@Override
public void onBeforeClientRequest(OClientConnection oClientConnection, byte b) {
this.listeners.forEach((l) -> l.onBeforeClientRequest(oClientConnection, b));
}
@Override
public void onAfterClientRequest(OClientConnection oClientConnection, byte b) {
this.listeners.forEach((l) -> l.onAfterClientRequest(oClientConnection, b));
}
@Override
public void onClientError(OClientConnection oClientConnection, Throwable throwable) {
this.listeners.forEach((l) -> l.onClientError(oClientConnection, throwable));
}
@Override
public void config(
OServer oServer, OServerParameterConfiguration[] oServerParameterConfigurations) {}
@Override
public void sendShutdown() {}
@Override
public Object getContent(String s) {
return null;
}
@Override
public PRIORITY getPriority() {
return PRIORITY.LAST;
}
@Override
public void onAfterActivate() {
if (server.getDistributedManager() != null) {
nodesManager = new NodesManager(server.getDistributedManager());
}
}
@Override
public void onCreate(ODatabaseInternal iDatabase) {
OStorage storage = iDatabase.getStorage();
if (storages.get(storage.getName()) == null) {
if (storage instanceof OEnterpriseLocalPaginatedStorage) {
OEnterpriseLocalPaginatedStorage s = (OEnterpriseLocalPaginatedStorage) storage;
storages.put(storage.getName(), s);
dbListeners.forEach((l) -> l.onOpen(s));
}
}
iDatabase.registerListener(this);
}
@Override
public void onOpen(final ODatabaseInternal iDatabase) {
final OStorage storage = iDatabase.getStorage();
if (storage.getUnderlying() instanceof OEnterpriseLocalPaginatedStorage) {
OEnterpriseLocalPaginatedStorage s =
(OEnterpriseLocalPaginatedStorage) storage.getUnderlying();
if (storages.putIfAbsent(storage.getName(), s) == null) {
storages.put(storage.getName(), s);
dbListeners.forEach((l) -> l.onOpen(s));
}
}
iDatabase.registerListener(this);
}
@Override
public void onClose(ODatabaseInternal iDatabase) {}
@Override
public void onDrop(final ODatabaseInternal iDatabase) {
final OStorage storage = iDatabase.getStorage();
if (storage instanceof OEnterpriseLocalPaginatedStorage) {
if (storages.remove(storage.getName()) != null) {
OEnterpriseLocalPaginatedStorage s = (OEnterpriseLocalPaginatedStorage) storage;
dbListeners.forEach((l) -> l.onDrop(s));
}
}
}
@Override
public void onLocalNodeConfigurationRequest(ODocument iConfiguration) {}
@Override
public void registerFunction(OSQLFunction function) {
OSQLEngine.getInstance().registerFunction(function.getName(), function);
}
@Override
public void registerStatelessCommand(OServerCommand iCommand) {
final OServerNetworkListener listener =
server.getListenerByProtocol(ONetworkProtocolHttpAbstract.class);
if (listener != null) {
listener.registerStatelessCommand(iCommand);
}
}
@Override
public void unregisterStatelessCommand(Class iCommandClass) {
final OServerNetworkListener listener =
server.getListenerByProtocol(ONetworkProtocolHttpAbstract.class);
if (listener != null) {
listener.unregisterStatelessCommand(iCommandClass);
}
}
@Override
public Collection listCommands() {
final OServerNetworkListener listener =
server.getListenerByProtocol(ONetworkProtocolHttpAbstract.class);
return listener.getStatelessCommands();
}
@Override
public List getConnections() {
return server.getClientConnectionManager().getConnections();
}
@Override
public void unregisterFunction(String function) {
OSQLEngine.getInstance().unregisterFunction(function);
}
@Override
public OSecuritySystem getSecurity() {
return server.getSecurity();
}
@Override
public List listQueries(Optional> filter) {
return getConnections().stream()
.filter((c) -> c.getDatabase() != null && filter.map(f -> f.apply(c)).orElse(true))
.flatMap(
(c) ->
new HashMap<>(c.getDatabase().getActiveQueries())
.entrySet().stream()
.map(
(k) -> {
OResultInternal internal = new OResultInternal();
internal.setProperty("queryId", k.getKey());
OQueryDatabaseState resultSet = k.getValue();
String user = "-";
if (c.getDatabase() != null && c.getDatabase().getUser() != null) {
user = c.getDatabase().getUser().getName();
}
internal.setProperty("sessionId", c.getId());
internal.setProperty("user", user);
internal.setProperty("database", c.getDatabase().getName());
Optional info = getQueryInfo(resultSet.getResultSet());
info.ifPresent(
(it) -> {
internal.setProperty("language", it.getLanguage());
internal.setProperty("query", it.getStatement());
internal.setProperty("startTime", it.getStartTime());
internal.setProperty(
"elapsedTimeMillis", it.getElapsedTimeMillis());
});
return internal;
}))
.collect(Collectors.toList());
}
@Override
public List getQueryStats(Optional database) {
return agent
.getServiceByClass(OrientDBMetricsService.class)
.map(
(service) -> {
SortedMap histograms =
service
.getRegistry()
.getHistograms((name, metric) -> name.matches("(?s)db.*.query.*"));
return histograms.entrySet().stream()
.sorted(
(v1, v2) -> {
OSnapshot snapshot1 = v1.getValue().getSnapshot();
OSnapshot snapshot2 = v2.getValue().getSnapshot();
return Double.compare(snapshot2.getMean(), snapshot1.getMean());
})
.map(
(e) -> {
OResultInternal result = new OResultInternal();
String key = e.getKey();
OHistogram h = e.getValue();
OSnapshot snapshot = h.getSnapshot();
String statement = key.substring(key.indexOf(".query.") + 7);
String language = statement.substring(0, statement.indexOf("."));
String query = statement.substring(statement.indexOf(".") + 1);
String db = key.substring(key.indexOf("db.") + 3, key.indexOf(".query."));
result.setProperty("database", db);
result.setProperty("language", language);
result.setProperty("query", query);
result.setProperty("count", h.getCount());
result.setProperty("max", snapshot.getMax());
result.setProperty("min", snapshot.getMin());
result.setProperty("mean", snapshot.getMean());
return (OResult) result;
})
.filter(
f ->
database
.map(e -> e.equalsIgnoreCase(f.getProperty("database")))
.orElse(true))
.collect(Collectors.toList());
})
.orElse(new ArrayList());
}
@Override
public void onCreate(ODatabase iDatabase) {}
@Override
public void onDelete(ODatabase iDatabase) {}
@Override
public void onOpen(ODatabase iDatabase) {}
@Override
public void onBeforeTxBegin(ODatabase iDatabase) {}
@Override
public void onBeforeTxRollback(ODatabase iDatabase) {}
@Override
public void onAfterTxRollback(ODatabase iDatabase) {}
@Override
public void onBeforeTxCommit(ODatabase iDatabase) {}
@Override
public void onAfterTxCommit(ODatabase iDatabase) {}
@Override
public void onClose(ODatabase iDatabase) {}
@Override
public void onBeforeCommand(OCommandRequestText iCommand, OCommandExecutor executor) {}
@Override
public void onAfterCommand(
OCommandRequestText iCommand, OCommandExecutor executor, Object result) {}
@Override
public void onCommandStart(ODatabase database, OResultSet result) {
this.dbListeners.forEach((c -> c.onCommandStart(database, result)));
}
@Override
public void onCommandEnd(ODatabase database, OResultSet result) {
this.dbListeners.forEach((c -> c.onCommandEnd(database, result)));
}
@Override
public Optional getQueryInfo(final OResultSet resultSet) {
Optional info = Optional.empty();
if (resultSet instanceof OLocalResultSetLifecycleDecorator) {
OResultSet oResultSet = ((OLocalResultSetLifecycleDecorator) resultSet).getInternal();
if (oResultSet instanceof OLocalResultSet) {
OLocalResultSet oLocalResultSet = (OLocalResultSet) oResultSet;
Optional plan = oLocalResultSet.getExecutionPlan();
info =
plan.map(
(p -> {
String q = "";
if (p instanceof OInternalExecutionPlan) {
String stm = ((OInternalExecutionPlan) p).getGenericStatement();
if (stm != null) {
q = stm;
}
} else {
q = p.toString();
}
return new QueryInfo(
q,
"sql",
oLocalResultSet.getStartTime(),
oLocalResultSet.getTotalExecutionTime());
}));
} else if (oResultSet instanceof OQueryMetrics) {
OQueryMetrics oQueryMetrics = (OQueryMetrics) oResultSet;
info =
Optional.of(
new QueryInfo(
oQueryMetrics.getStatement(),
oQueryMetrics.getLanguage(),
oQueryMetrics.getStartTime(),
oQueryMetrics.getElapsedTimeMillis()));
}
}
return info;
}
@Override
public Optional getServiceByClass(Class klass) {
return agent.getServiceByClass(klass);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy