com.orientechnologies.agent.http.command.OServerCommandBackupManager Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016 OrientDB LTD (info(at)orientdb.com)
 *
 *   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.
 *
 *   For more information: http://www.orientdb.com
 */

package com.orientechnologies.agent.http.command;

import com.orientechnologies.agent.EnterprisePermissions;
import com.orientechnologies.agent.services.backup.OBackupService;
import com.orientechnologies.enterprise.server.OEnterpriseServer;
import com.orientechnologies.orient.core.record.impl.ODocument;
import com.orientechnologies.orient.server.network.protocol.http.OHttpRequest;
import com.orientechnologies.orient.server.network.protocol.http.OHttpResponse;
import com.orientechnologies.orient.server.network.protocol.http.OHttpUtils;
import java.io.IOException;

/** Created by Enrico Risa on 22/03/16. */
public class OServerCommandBackupManager extends OServerCommandDistributedScope {

  OBackupService backupManager;
  private static final String[] NAMES = {
    "GET|backupManager",
    "GET|backupManager/*",
    "POST|backupManager",
    "POST|backupManager/*",
    "PUT|backupManager/*",
    "DELETE|backupManager/*"
  };

  public OServerCommandBackupManager(OBackupService manager, OEnterpriseServer server) {
    super(EnterprisePermissions.SERVER_BACKUP.toString(), server);
    backupManager = manager;
  }

  @Override
  public boolean execute(OHttpRequest iRequest, OHttpResponse iResponse) throws Exception {
    return super.execute(iRequest, iResponse);
  }

  @Override
  protected void doPost(OHttpRequest iRequest, OHttpResponse iResponse) throws IOException {

    if (super.authenticate(
        iRequest, iResponse, true, EnterprisePermissions.SERVER_BACKUP_EDIT.toString())) {
      final String[] parts = checkSyntax(iRequest.getUrl(), 1, "Syntax error: backupManager");
      if (parts.length == 1) {
        ODocument body = new ODocument().fromJSON(iRequest.getContent(), "noMap");
        ODocument doc = backupManager.addBackupAndSchedule(body);
        iResponse.send(
            OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, doc.toJSON(""), null);
      } else if (parts.length == 3) {

        String uuid = parts[1];
        String command = parts[2];
        if (command.equals("restore")) {
          ODocument body = new ODocument().fromJSON(iRequest.getContent(), "noMap");
          backupManager.restoreBackup(uuid, body);
          iResponse.send(OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, null, null);
        } else {
          throw new IllegalArgumentException("cannot execute post request ");
        }
      } else {
        throw new IllegalArgumentException("cannot execute post request ");
      }
    }
  }

  @Override
  protected void doPut(final OHttpRequest iRequest, final OHttpResponse iResponse)
      throws IOException {
    if (super.authenticate(
        iRequest, iResponse, true, EnterprisePermissions.SERVER_BACKUP_EDIT.toString())) {
      final String[] parts = checkSyntax(iRequest.getUrl(), 2, "Syntax error: backupManager");
      ODocument body = new ODocument().fromJSON(iRequest.getContent(), "noMap");
      backupManager.changeBackup(parts[1], body);
      iResponse.send(
          OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, body.toJSON(""), null);
    } else {
      throw new IllegalArgumentException("cannot execute post put ");
    }
  }

  @Override
  protected void doGet(OHttpRequest iRequest, OHttpResponse iResponse) throws IOException {
    final String[] parts = checkSyntax(iRequest.getUrl(), 1, "Syntax error: backupManager");

    if (parts.length == 1) {
      ODocument doc = backupManager.getConfiguration();
      iResponse.send(
          OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, doc.toJSON(""), null);
    } else if (parts.length >= 3) {
      String uuid = parts[1];
      String command = parts[2];

      if (command.equalsIgnoreCase("status")) {
        ODocument status = backupManager.logs(uuid, 0, 1, iRequest.getParameters());
        iResponse.send(
            OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, status.toJSON(""), null);
      } else if (command.equalsIgnoreCase("log")) {

        String pagePize = iRequest.getParameter("pageSize");
        String page = iRequest.getParameter("page");
        int pSize = pagePize != null ? Integer.valueOf(pagePize) : -1;
        int p = page != null ? Integer.valueOf(page) : 0;
        ODocument history;
        if (parts.length == 4) {
          Long unitId = Long.parseLong(parts[3]);
          history = backupManager.logs(uuid, unitId, p, pSize, iRequest.getParameters());
        } else {
          history = backupManager.logs(uuid, p, pSize, iRequest.getParameters());
        }
        iResponse.send(
            OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, history.toJSON(""), null);
      } else {
        throw new IllegalArgumentException("cannot find executor for command:" + command);
      }
    }
  }

  @Override
  protected void doDelete(OHttpRequest iRequest, OHttpResponse iResponse) throws IOException {

    final String[] parts = checkSyntax(iRequest.getUrl(), 1, "Syntax error: backupManager");

    if (parts.length == 2) {
      String uuid = parts[1];
      backupManager.removeAndStopBackup(uuid);
      iResponse.send(OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, null, null);
    } else if (parts.length >= 3) {

      String uuid = parts[1];
      String command = parts[2];

      if (command.equalsIgnoreCase("remove")) {

        try {
          Long unitId = Long.parseLong(iRequest.getParameter("unitId"));
          Long timestamp = Long.parseLong(iRequest.getParameter("txId"));
          backupManager.deleteBackup(uuid, unitId, timestamp);
          iResponse.send(OHttpUtils.STATUS_OK_CODE, "OK", OHttpUtils.CONTENT_JSON, null, null);
        } catch (Exception e) {
          throw new RuntimeException(e);
        }

      } else {
        throw new IllegalArgumentException("cannot find executor for command:" + command);
      }
    } else {
      throw new IllegalArgumentException("cannot find executor for command:" + parts[0]);
    }
  }

  @Override
  public String[] getNames() {
    return NAMES;
  }
}