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

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