com.orientechnologies.agent.services.backup.strategy.OBackupStrategyMixBackup Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 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.services.backup.strategy;

import com.orientechnologies.agent.services.backup.OBackupConfig;
import com.orientechnologies.agent.services.backup.OBackupListener;
import com.orientechnologies.agent.services.backup.log.*;
import com.orientechnologies.common.log.OLogManager;
import com.orientechnologies.orient.core.db.ODatabaseInternal;
import com.orientechnologies.orient.core.db.document.ODatabaseDocument;
import com.orientechnologies.orient.core.record.impl.ODocument;
import com.orientechnologies.orient.core.schedule.OCronExpression;
import com.orientechnologies.orient.core.storage.impl.local.paginated.OEnterpriseLocalPaginatedStorage;
import com.orientechnologies.orient.server.handler.OAutomaticBackup;
import java.io.File;
import java.io.IOException;
import java.text.ParseException;
import java.util.Date;

/** Created by Enrico Risa on 25/03/16. */
public class OBackupStrategyMixBackup extends OBackupStrategy {
  // Make sure not to displace / skip a full backup (e.g. in case an incremental backup took
  // longer). Hence (1) store last (and potentially skipped) full backup.
  private Date skippedFull = null;

  protected boolean isIncremental = false;

  public OBackupStrategyMixBackup(ODocument cfg, OBackupLogger logger) {
    super(cfg, logger);
  }

  @Override
  public OAutomaticBackup.MODE getMode() {
    return isIncremental
        ? OAutomaticBackup.MODE.INCREMENTAL_BACKUP
        : OAutomaticBackup.MODE.FULL_BACKUP;
  }

  protected String calculatePath(ODatabaseDocument db) {
    if (!isIncremental) {
      return defaultPath();
    }
    OBackupFinishedLog last = null;
    try {
      last = (OBackupFinishedLog) logger.findLast(OBackupLogType.BACKUP_FINISHED, getUUID());
    } catch (Exception e) {
      OLogManager.instance().error(this, "Error " + e.getMessage(), e);
    }
    try {
      if (last != null
          && ((OEnterpriseLocalPaginatedStorage) ((ODatabaseInternal) db).getStorage())
              .isLastBackupCompatibleWithUUID(new File(last.getPath()))) {
        return last.getPath();
      }
    } catch (IOException e) {
      OLogManager.instance().warn(this, "Error checking backup compatibility", e);
      return last.getPath();
    }

    return defaultPath();
  }

  protected String defaultPath() {
    long begin = System.currentTimeMillis();
    try {
      OBackupLog last = logger.findLast(OBackupLogType.BACKUP_SCHEDULED, getUUID());
      if (last != null) {
        begin = last.getUnitId();
      }
    } catch (IOException e) {
      OLogManager.instance().error(this, "Error " + e.getMessage(), e);
    }
    String basePath = cfg.field(OBackupConfig.DIRECTORY);
    String dbName = cfg.field(OBackupConfig.DBNAME);
    return basePath + File.separator + dbName + "-" + begin;
  }

  @Override
  public Date scheduleNextExecution(final OBackupListener listener) {
    final OBackupScheduledLog lastBackupSchedule = lastUnfiredSchedule();
    if (lastBackupSchedule == null) {
      final ODocument full =
          (ODocument) cfg.eval(OBackupConfig.MODES + "." + OAutomaticBackup.MODE.FULL_BACKUP);
      final String whenFull = full.field(OBackupConfig.WHEN);
      final ODocument incremental =
          (ODocument)
              cfg.eval(OBackupConfig.MODES + "." + OAutomaticBackup.MODE.INCREMENTAL_BACKUP);
      final String whenIncremental = incremental.field(OBackupConfig.WHEN);
      try {
        final OCronExpression eFull = new OCronExpression(whenFull);
        final OCronExpression eIncremental = new OCronExpression(whenIncremental);
        final Date now = new Date();

        final Date nextFull = eFull.getNextValidTimeAfter(now);
        final Date nextIncremental = eIncremental.getNextValidTimeAfter(now);

        long unitId = logger.nextOpId();
        try {
          final OBackupFinishedLog lastCompleted =
              (OBackupFinishedLog) logger.findLast(OBackupLogType.BACKUP_FINISHED, getUUID());
          if (lastCompleted != null
              && nextIncremental.before(nextFull)
              && !Boolean.TRUE.equals(lastCompleted.getPrevChange())) {
            unitId = lastCompleted.getUnitId();

            // (2) when deciding on incremental check for skipped full backup that actually came
            // before the next incremental.
            if (skippedFull != null && skippedFull.before(nextIncremental)) {
              OLogManager.instance()
                  .info(this, "Found skipped full backup (i.e. before incremental)");
              isIncremental = false;
              skippedFull = null;
            } else {
              // (3) else take incremental, but remember skipped full backup.
              OLogManager.instance()
                  .debug(this, "Found incremental backup before full (i.e. remember skipped full)");
              isIncremental = true;
              skippedFull = nextFull;
            }
          } else {
            // (4) if full exercised, forget skipped full backup.
            OLogManager.instance().debug(this, "Found full backup before incremental");
            isIncremental = false;
            skippedFull = null;
          }
        } catch (final IOException e) {
          OLogManager.instance().error(this, "Error " + e.getMessage(), e);
        }
        final Date nextExecution = nextIncremental.before(nextFull) ? nextIncremental : nextFull;
        final OBackupScheduledLog log =
            new OBackupScheduledLog(
                unitId, logger.nextOpId(), getUUID(), getDbName(), getMode().toString());
        log.nextExecution = nextExecution.getTime();
        getLogger().log(log);
        listener.onEvent(cfg, log);
        return nextExecution;
      } catch (final ParseException e) {
        OLogManager.instance().error(this, "Parse exception: " + e.getMessage(), e);
        return null;
      }
    } else {
      isIncremental =
          OAutomaticBackup.MODE
              .INCREMENTAL_BACKUP
              .toString()
              .equalsIgnoreCase(lastBackupSchedule.getMode());
      // (5) if last schedule not null and next backup is incremental, preserve skipped full
      // backup, else forget.
      final Date nextExecution = new Date(lastBackupSchedule.nextExecution);
      if (skippedFull != null && isIncremental == false) {
        skippedFull = null;
      }
      return nextExecution;
    }
  }
}