opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/OperationCompleter.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 package org.opendaylight.controller.cluster.datastore;
   9
  10 import akka.dispatch.OnComplete;
  11 import com.google.common.base.Preconditions;
  12 import java.util.concurrent.Semaphore;
  13 import org.opendaylight.controller.cluster.datastore.messages.BatchedModificationsReply;
  14
  15 public final class OperationCompleter extends OnComplete<Object> {
  16     private final Semaphore operationLimiter;
  17
  18     OperationCompleter(Semaphore operationLimiter){
  19         this.operationLimiter = Preconditions.checkNotNull(operationLimiter);
  20     }
  21
  22     @Override
  23     public void onComplete(Throwable throwable, Object message) {
  24         if(message instanceof BatchedModificationsReply) {
  25             this.operationLimiter.release(((BatchedModificationsReply)message).getNumBatched());
  26         } else {
  27             this.operationLimiter.release();
  28         }
  29     }
  30 }