980289fcf9762d8e2167d107060c4f122aad3172
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / utils / Dispatchers.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8
9 package org.opendaylight.controller.cluster.datastore.utils;
10
11 import com.google.common.base.Preconditions;
12 import scala.concurrent.ExecutionContext;
13
14 public class Dispatchers {
15     public static final String DEFAULT_DISPATCHER_PATH = "akka.actor.default-dispatcher";
16     public static final String CLIENT_DISPATCHER_PATH = "client-dispatcher";
17     public static final String TXN_DISPATCHER_PATH = "txn-dispatcher";
18     public static final String SHARD_DISPATCHER_PATH = "shard-dispatcher";
19     public static final String NOTIFICATION_DISPATCHER_PATH = "notification-dispatcher";
20
21     private final akka.dispatch.Dispatchers dispatchers;
22
23     public enum DispatcherType {
24         Client(CLIENT_DISPATCHER_PATH),
25         Transaction(TXN_DISPATCHER_PATH),
26         Shard(SHARD_DISPATCHER_PATH),
27         Notification(NOTIFICATION_DISPATCHER_PATH);
28
29         private final String path;
30         DispatcherType(String path) {
31             this.path = path;
32         }
33
34         private String path(akka.dispatch.Dispatchers dispatchers) {
35             if (dispatchers.hasDispatcher(path)) {
36                 return path;
37             }
38             return DEFAULT_DISPATCHER_PATH;
39         }
40
41         private ExecutionContext dispatcher(akka.dispatch.Dispatchers dispatchers) {
42             if (dispatchers.hasDispatcher(path)) {
43                 return dispatchers.lookup(path);
44             }
45             return dispatchers.defaultGlobalDispatcher();
46         }
47     }
48
49     public Dispatchers(akka.dispatch.Dispatchers dispatchers) {
50         Preconditions.checkNotNull(dispatchers, "dispatchers should not be null");
51         this.dispatchers = dispatchers;
52     }
53
54     public ExecutionContext getDispatcher(DispatcherType dispatcherType) {
55         return dispatcherType.dispatcher(this.dispatchers);
56     }
57
58     public String getDispatcherPath(DispatcherType dispatcherType) {
59         return dispatcherType.path(this.dispatchers);
60     }
61 }