src/main/java/org/onap/music/lockingservice/MusicLockingService.java

   1 /*
   2  * ============LICENSE_START========================================== org.onap.music
   3  * =================================================================== Copyright (c) 2017 AT&T
   4  * Intellectual Property ===================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
   6  * in compliance with the License. You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software distributed under the License
  11  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  12  * or implied. See the License for the specific language governing permissions and limitations under
  13  * the License.
  14  *
  15  * ============LICENSE_END=============================================
  16  * ====================================================================
  17  */
  18 package org.onap.music.lockingservice;
  19
  20
  21 import java.io.IOException;
  22 import java.util.StringTokenizer;
  23 import java.util.concurrent.CountDownLatch;
  24
  25 import org.apache.zookeeper.KeeperException;
  26 import org.apache.zookeeper.WatchedEvent;
  27 import org.apache.zookeeper.Watcher;
  28 import org.apache.zookeeper.Watcher.Event.KeeperState;
  29 import org.apache.zookeeper.ZooKeeper;
  30 import org.onap.music.eelf.logging.EELFLoggerDelegate;
  31 import org.onap.music.exceptions.MusicLockingException;
  32 import org.onap.music.exceptions.MusicServiceException;
  33 import org.onap.music.main.MusicUtil;
  34
  35
  36 public class MusicLockingService implements Watcher {
  37
  38
  39     private static final int SESSION_TIMEOUT = 180000;
  40     ZkStatelessLockService zkLockHandle = null;
  41     private CountDownLatch connectedSignal = new CountDownLatch(1);
  42     private static EELFLoggerDelegate logger =
  43                     EELFLoggerDelegate.getLogger(MusicLockingService.class);
  44
  45     public MusicLockingService() throws MusicServiceException {
  46         try {
  47             ZooKeeper zk = new ZooKeeper(MusicUtil.getMyZkHost(), SESSION_TIMEOUT, this);
  48             connectedSignal.await();
  49             zkLockHandle = new ZkStatelessLockService(zk);
  50         } catch (IOException e) {
  51             logger.error(EELFLoggerDelegate.errorLogger, e.getMessage());
  52             throw new MusicServiceException("IO Error has occured" + e.getMessage());
  53         } catch (InterruptedException e) {
  54             logger.error(EELFLoggerDelegate.errorLogger, e.getMessage());
  55             throw new MusicServiceException("Exception Occured " + e.getMessage());
  56         }
  57     }
  58
  59     public ZkStatelessLockService getzkLockHandle() {
  60         return zkLockHandle;
  61     }
  62
  63     public MusicLockingService(String lockServer) {
  64         try {
  65             ZooKeeper zk = new ZooKeeper(lockServer, SESSION_TIMEOUT, this);
  66             connectedSignal.await();
  67             zkLockHandle = new ZkStatelessLockService(zk);
  68         } catch (IOException | InterruptedException e) {
  69             logger.error(EELFLoggerDelegate.errorLogger, e.getMessage());
  70         }
  71     }
  72
  73     public void createLockaIfItDoesNotExist(String lockName) {
  74         if (zkLockHandle.checkIfLockExists(lockName) == false) {
  75             String lockHolder = null;
  76             MusicLockState ml = new MusicLockState(MusicLockState.LockStatus.UNLOCKED, lockHolder);
  77             byte[] data = ml.serialize();
  78             zkLockHandle.createLock(lockName, data);
  79         }
  80     }
  81
  82     public void setLockState(String lockName, MusicLockState mls) {
  83         byte[] data = mls.serialize();
  84         zkLockHandle.setNodeData(lockName, data);
  85     }
  86
  87     public MusicLockState getLockState(String lockName) throws MusicLockingException {
  88
  89         byte[] data = null;
  90         try{
  91                 data = zkLockHandle.getNodeData(lockName);
  92         }catch (Exception ex){
  93                 logger.error(EELFLoggerDelegate.errorLogger,ex.getMessage());
  94         }
  95         if(data !=null)
  96         return MusicLockState.deSerialize(data);
  97         else
  98         throw new  MusicLockingException("Invalid lock or acquire failed");
  99     }
 100
 101     public String createLockId(String lockName) {
 102         String lockIdWithSlash = zkLockHandle.createLockId(lockName);
 103         return lockIdWithSlash.replace('/', '$');
 104     }
 105
 106     public boolean isMyTurn(String lockIdWithDollar) {
 107         String lockId = lockIdWithDollar.replace('$', '/');
 108         StringTokenizer st = new StringTokenizer(lockId);
 109         String lockName = "/" + st.nextToken("/");
 110         try {
 111             return zkLockHandle.lock(lockName, lockId);
 112         } catch (KeeperException | InterruptedException e) {
 113             logger.error(EELFLoggerDelegate.errorLogger, e.getMessage());
 114         }
 115         return false;
 116     }
 117
 118     public void unlockAndDeleteId(String lockIdWithDollar) {
 119         String lockId = lockIdWithDollar.replace('$', '/');
 120         zkLockHandle.unlock(lockId);
 121     }
 122
 123     public void deleteLock(String lockName) {
 124         zkLockHandle.deleteLock(lockName);
 125     }
 126
 127     public String whoseTurnIsIt(String lockName) {
 128         String lockHolder = zkLockHandle.currentLockHolder(lockName);
 129         return lockHolder.replace('/', '$');
 130
 131     }
 132
 133     public void process(WatchedEvent event) { // Watcher interface
 134         if (event.getState() == KeeperState.SyncConnected) {
 135             connectedSignal.countDown();
 136         }
 137     }
 138
 139
 140     public void close() {
 141         zkLockHandle.close();
 142     }
 143
 144         public boolean lockIdExists(String lockIdWithDollar) {
 145                 String lockId = lockIdWithDollar.replace('$', '/');
 146                 return zkLockHandle.checkIfLockExists(lockId);
 147         }
 148
 149 }