Bug 7449: Add custom dispatcher for message serialization/slicing
[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     public static final String SERIALIZATION_DISPATCHER_PATH = "serialization-dispatcher";
21
22     private final akka.dispatch.Dispatchers dispatchers;
23
24     public enum DispatcherType {
25         Client(CLIENT_DISPATCHER_PATH),
26         Transaction(TXN_DISPATCHER_PATH),
27         Shard(SHARD_DISPATCHER_PATH),
28         Notification(NOTIFICATION_DISPATCHER_PATH),
29         Serialization(SERIALIZATION_DISPATCHER_PATH);
30
31         private final String path;
32
33         DispatcherType(String path) {
34             this.path = path;
35         }
36
37         private String path(akka.dispatch.Dispatchers dispatchers) {
38             if (dispatchers.hasDispatcher(path)) {
39                 return path;
40             }
41             return DEFAULT_DISPATCHER_PATH;
42         }
43
44         private ExecutionContext dispatcher(akka.dispatch.Dispatchers dispatchers) {
45             if (dispatchers.hasDispatcher(path)) {
46                 return dispatchers.lookup(path);
47             }
48             return dispatchers.defaultGlobalDispatcher();
49         }
50     }
51
52     public Dispatchers(akka.dispatch.Dispatchers dispatchers) {
53         Preconditions.checkNotNull(dispatchers, "dispatchers should not be null");
54         this.dispatchers = dispatchers;
55     }
56
57     public ExecutionContext getDispatcher(DispatcherType dispatcherType) {
58         return dispatcherType.dispatcher(this.dispatchers);
59     }
60
61     public String getDispatcherPath(DispatcherType dispatcherType) {
62         return dispatcherType.path(this.dispatchers);
63     }
64 }