Пример #1
0
  protected <T> T networkAdminOperation(
      final OStorageRemoteOperation<T> operation, final String errorMessage) {

    OChannelBinaryAsynchClient network = null;
    try {
      // TODO:replace this api with one that get connection for only the specified url.
      String serverUrl = storage.getNextAvailableServerURL(false, session);
      do {
        try {
          network = storage.getNetwork(serverUrl);
        } catch (OException e) {
          serverUrl = storage.useNewServerURL(serverUrl);
          if (serverUrl == null) throw e;
        }
      } while (network == null);

      T res = operation.execute(network, storage.getCurrentSession());
      storage.connectionManager.release(network);
      return res;
    } catch (Exception e) {
      if (network != null) storage.connectionManager.release(network);
      storage.close(true, false);
      throw OException.wrapException(new OStorageException(errorMessage), e);
    }
  }
Пример #2
0
  /**
   * Drops a database from a remote server instance.
   *
   * @param iDatabaseName The database name
   * @param storageType Storage type between "plocal" or "memory".
   * @return The instance itself. Useful to execute method in chain
   * @throws IOException
   */
  public synchronized OServerAdmin dropDatabase(
      final String iDatabaseName, final String storageType) throws IOException {

    boolean retry = true;
    while (retry) {
      retry =
          networkAdminOperation(
              new OStorageRemoteOperation<Boolean>() {
                @Override
                public Boolean execute(
                    final OChannelBinaryAsynchClient network, OStorageRemoteSession session)
                    throws IOException {
                  try {
                    try {
                      storage.beginRequest(
                          network, OChannelBinaryProtocol.REQUEST_DB_DROP, session);
                      network.writeString(iDatabaseName);
                      network.writeString(storageType);
                    } finally {
                      storage.endRequest(network);
                    }

                    storage.getResponse(network, session);
                    return false;
                  } catch (OModificationOperationProhibitedException oope) {
                    return handleDBFreeze();
                  }
                }
              },
              "Cannot delete the remote storage: " + storage.getName());
    }

    final Set<OStorage> underlyingStorages = new HashSet<OStorage>();

    for (OStorage s : Orient.instance().getStorages()) {
      if (s.getType().equals(storage.getType()) && s.getName().equals(storage.getName())) {
        underlyingStorages.add(s.getUnderlying());
      }
    }

    for (OStorage s : underlyingStorages) {
      s.close(true, true);
    }

    ODatabaseRecordThreadLocal.INSTANCE.remove();

    return this;
  }
Пример #3
0
  protected ODocument sendRequest(
      final byte iRequest, final ODocument iPayLoad, final String iActivity) {
    // Using here networkOperation because the original retry logic was lik networkOperation
    return storage.networkOperation(
        new OStorageRemoteOperation<ODocument>() {
          @Override
          public ODocument execute(
              OChannelBinaryAsynchClient network, OStorageRemoteSession session)
              throws IOException {
            try {
              storage.beginRequest(network, iRequest, session);
              network.writeBytes(iPayLoad.toStream());
            } finally {
              storage.endRequest(network);
            }

            try {
              storage.beginResponse(network, session);
              return new ODocument(network.readBytes());
            } finally {
              storage.endResponse(network);
            }
          }
        },
        "Error on executing  '" + iActivity + "'");
  }
Пример #4
0
  /**
   * Releases a frozen cluster.
   *
   * @param clusterId Id of cluster to freeze
   * @param storageType Storage type between "plocal" or "memory".
   * @return
   * @throws IOException
   * @see #freezeCluster(int, String)
   */
  public synchronized OServerAdmin releaseCluster(final int clusterId, final String storageType)
      throws IOException {

    networkAdminOperation(
        new OStorageRemoteOperation<Void>() {
          @Override
          public Void execute(
              final OChannelBinaryAsynchClient network, OStorageRemoteSession session)
              throws IOException {
            try {
              storage.beginRequest(
                  network, OChannelBinaryProtocol.REQUEST_DATACLUSTER_RELEASE, session);
              network.writeString(storage.getName());
              network.writeShort((short) clusterId);
              network.writeString(storageType);
            } finally {
              storage.endRequest(network);
            }

            storage.getResponse(network, session);
            return null;
          }
        },
        "Cannot release the remote cluster " + clusterId + " on storage: " + storage.getName());

    return this;
  }
Пример #5
0
  /**
   * Checks if a database exists in the remote server.
   *
   * @param iDatabaseName The database name
   * @param storageType Storage type between "plocal" or "memory".
   * @return true if exists, otherwise false
   * @throws IOException
   */
  public synchronized boolean existsDatabase(final String iDatabaseName, final String storageType)
      throws IOException {

    return networkAdminOperation(
        new OStorageRemoteOperation<Boolean>() {
          @Override
          public Boolean execute(
              final OChannelBinaryAsynchClient network, OStorageRemoteSession session)
              throws IOException {

            try {
              storage.beginRequest(network, OChannelBinaryProtocol.REQUEST_DB_EXIST, session);
              network.writeString(iDatabaseName);
              network.writeString(storageType);
            } finally {
              storage.endRequest(network);
            }

            try {
              storage.beginResponse(network, session);
              return network.readByte() == 1;
            } finally {
              storage.endResponse(network);
            }
          }
        },
        "Error on checking existence of the remote storage: " + storage.getName());
  }
Пример #6
0
  /**
   * Connects to a remote server.
   *
   * @param iUserName Server's user name
   * @param iUserPassword Server's password for the user name used
   * @return The instance itself. Useful to execute method in chain
   * @throws IOException
   */
  public synchronized OServerAdmin connect(final String iUserName, final String iUserPassword)
      throws IOException {
    networkAdminOperation(
        new OStorageRemoteOperation<Void>() {
          @Override
          public Void execute(OChannelBinaryAsynchClient network, OStorageRemoteSession session)
              throws IOException {
            OStorageRemoteNodeSession nodeSession =
                storage.getCurrentSession().getOrCreate(network.getServerURL());
            try {
              storage.beginRequest(network, OChannelBinaryProtocol.REQUEST_CONNECT, session);

              storage.sendClientInfo(network, clientType, false, collectStats);

              String username = iUserName;
              String password = iUserPassword;

              OCredentialInterceptor ci = OSecurityManager.instance().newCredentialInterceptor();

              if (ci != null) {
                ci.intercept(storage.getURL(), iUserName, iUserPassword);
                username = ci.getUsername();
                password = ci.getPassword();
              }

              network.writeString(username);
              network.writeString(password);
            } finally {
              storage.endRequest(network);
            }

            try {
              network.beginResponse(nodeSession.getSessionId(), false);
              int sessionId = network.readInt();
              byte[] sessionToken = network.readBytes();
              if (sessionToken.length == 0) {
                sessionToken = null;
              }
              nodeSession.setSession(sessionId, sessionToken);
            } finally {
              storage.endResponse(network);
            }

            return null;
          }
        },
        "Cannot connect to the remote server/database '" + storage.getURL() + "'");
    return this;
  }
Пример #7
0
  /**
   * Creates a database in a remote server.
   *
   * @param iDatabaseName The database name
   * @param iDatabaseType 'document' or 'graph'
   * @param iStorageMode local or memory
   * @param backupPath path to incremental backup which will be used to create database (optional)
   * @return The instance itself. Useful to execute method in chain
   * @throws IOException
   */
  public synchronized OServerAdmin createDatabase(
      final String iDatabaseName,
      final String iDatabaseType,
      final String iStorageMode,
      final String backupPath)
      throws IOException {

    if (iDatabaseName == null || iDatabaseName.length() <= 0) {
      final String message = "Cannot create unnamed remote storage. Check your syntax";
      OLogManager.instance().error(this, message);
      throw new OStorageException(message);
    } else {
      networkAdminOperation(
          new OStorageRemoteOperation<Void>() {
            @Override
            public Void execute(
                final OChannelBinaryAsynchClient network, OStorageRemoteSession session)
                throws IOException {
              String storageMode;
              if (iStorageMode == null) storageMode = "csv";
              else storageMode = iStorageMode;

              try {
                storage.beginRequest(network, OChannelBinaryProtocol.REQUEST_DB_CREATE, session);
                network.writeString(iDatabaseName);
                if (network.getSrvProtocolVersion() >= 8) network.writeString(iDatabaseType);
                network.writeString(storageMode);

                if (network.getSrvProtocolVersion() > 35) network.writeString(backupPath);
              } finally {
                storage.endRequest(network);
              }

              storage.getResponse(network, session);
              return null;
            }
          },
          "Cannot create the remote storage: " + storage.getName());
    }

    return this;
  }
Пример #8
0
  /**
   * Freezes the database by locking it in exclusive mode.
   *
   * @param storageType Storage type between "plocal" or "memory".
   * @return
   * @throws IOException
   * @see #releaseDatabase(String)
   */
  public synchronized OServerAdmin freezeDatabase(final String storageType) throws IOException {
    networkAdminOperation(
        new OStorageRemoteOperation<Void>() {
          @Override
          public Void execute(OChannelBinaryAsynchClient network, OStorageRemoteSession session)
              throws IOException {
            try {
              storage.beginRequest(network, OChannelBinaryProtocol.REQUEST_DB_FREEZE, session);
              network.writeString(storage.getName());
              network.writeString(storageType);
            } finally {
              storage.endRequest(network);
            }

            storage.getResponse(network, session);
            return null;
          }
        },
        "Cannot freeze the remote storage: " + storage.getName());

    return this;
  }
Пример #9
0
 public boolean isConnected() {
   return storage != null && !storage.isClosed();
 }
Пример #10
0
 public synchronized String getURL() {
   return storage != null ? storage.getURL() : null;
 }
Пример #11
0
 public synchronized void close(boolean iForce) {
   storage.close(iForce, false);
 }
Пример #12
0
 /** Close the connection if open. */
 public synchronized void close() {
   storage.close();
 }
Пример #13
0
 /**
  * Drops a database from a remote server instance.
  *
  * @param storageType Storage type between "plocal" or "memory".
  * @return The instance itself. Useful to execute method in chain
  * @throws IOException
  */
 public synchronized OServerAdmin dropDatabase(final String storageType) throws IOException {
   return dropDatabase(storage.getName(), storageType);
 }
Пример #14
0
 /**
  * Checks if a database exists in the remote server.
  *
  * @param storageType Storage type between "plocal" or "memory".
  * @return true if exists, otherwise false
  * @throws IOException
  */
 public synchronized boolean existsDatabase(final String storageType) throws IOException {
   return existsDatabase(storage.getName(), storageType);
 }
Пример #15
0
 public synchronized String getStorageName() {
   return storage.getName();
 }
Пример #16
0
 /**
  * Creates a database in a remote server.
  *
  * @param iDatabaseType 'document' or 'graph'
  * @param iStorageMode local or memory
  * @return The instance itself. Useful to execute method in chain
  * @throws IOException
  */
 public synchronized OServerAdmin createDatabase(final String iDatabaseType, String iStorageMode)
     throws IOException {
   return createDatabase(storage.getName(), iDatabaseType, iStorageMode);
 }