SupervisionStrategyFactory.java

/*
 * Copyright 2014-2016 CyberVision, Inc.
 *
 * 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.
 */

package org.kaaproject.kaa.server.operations.service.akka.actors.supervision;

import akka.actor.OneForOneStrategy;
import akka.actor.SupervisorStrategy;
import akka.actor.SupervisorStrategy.Directive;
import akka.japi.Function;
import org.kaaproject.kaa.server.operations.service.akka.AkkaContext;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import scala.concurrent.duration.Duration;



public final class SupervisionStrategyFactory {

  private static final Logger LOG = LoggerFactory.getLogger(SupervisionStrategyFactory.class);

  private SupervisionStrategyFactory() {
  }

  public static SupervisorStrategy createIoRouterStrategy(AkkaContext context) {
    return buildResumeOrEscalateStrategy();
  }

  public static SupervisorStrategy createOpsActorStrategy(AkkaContext context) {
    return buildResumeOnRuntimeErrorStrategy();
  }

  public static SupervisorStrategy createTenantActorStrategy(AkkaContext context) {
    return buildResumeOnRuntimeErrorStrategy();
  }

  public static SupervisorStrategy createApplicationActorStrategy(AkkaContext context) {
    return buildRestartOrEscalateStrategy();
  }

  private static SupervisorStrategy buildResumeOrEscalateStrategy() {
    return new OneForOneStrategy(-1, Duration.Inf(),
            new Function<Throwable, SupervisorStrategy.Directive>() {
        @Override
        public Directive apply(Throwable throwable) throws Exception {
          logException(throwable);
          if (throwable instanceof Error) {
            return OneForOneStrategy.escalate();
          } else {
            return OneForOneStrategy.resume();
          }
        }
      });
  }

  private static SupervisorStrategy buildRestartOrEscalateStrategy() {
    return new OneForOneStrategy(-1, Duration.Inf(),
            new Function<Throwable, SupervisorStrategy.Directive>() {
        @Override
        public Directive apply(Throwable throwable) throws Exception {
          logException(throwable);
          if (throwable instanceof Error) {
            return OneForOneStrategy.escalate();
          } else {
            return OneForOneStrategy.restart();
          }
        }
      });
  }

  private static SupervisorStrategy buildResumeOnRuntimeErrorStrategy() {
    return new OneForOneStrategy(-1, Duration.Inf(),
            new Function<Throwable, SupervisorStrategy.Directive>() {
        @Override
        public Directive apply(Throwable throwable) throws Exception {
          logException(throwable);
          if (throwable instanceof Error) {
            return OneForOneStrategy.escalate();
          } else if (throwable instanceof RuntimeException) {
            return OneForOneStrategy.resume();
          } else {
            return OneForOneStrategy.restart();
          }
        }
      });
  }

  private static void logException(Throwable throwable) {
    LOG.error("Supervisor strategy got exception: {}", throwable.getMessage(), throwable);
  }

}