app/src/main/java/net/ktnx/mobileledger/utils/LockHolder.java

   1 /*
   2  * Copyright © 2019 Damyan Ivanov.
   3  * This file is part of MoLe.
   4  * MoLe is free software: you can distribute it and/or modify it
   5  * under the term of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your opinion), any later version.
   8  *
   9  * MoLe is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12  * GNU General Public License terms for details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with MoLe. If not, see <https://www.gnu.org/licenses/>.
  16  */
  17
  18 package net.ktnx.mobileledger.utils;
  19
  20 import java.util.concurrent.locks.Lock;
  21
  22 public class LockHolder implements AutoCloseable {
  23     private Lock rLock, wLock;
  24     LockHolder(Lock rLock) {
  25         this.rLock = rLock;
  26         this.wLock = null;
  27     }
  28     LockHolder(Lock rLock, Lock wLock) {
  29         this.rLock = rLock;
  30         this.wLock = wLock;
  31     }
  32     @Override
  33     public void close() {
  34         if (wLock != null) { wLock.unlock(); wLock = null; }
  35         if (rLock != null) { rLock.unlock(); rLock = null; }
  36     }
  37     public void downgrade() {
  38         if (rLock == null) throw new IllegalStateException("no locks are held");
  39
  40         if (wLock == null) return;
  41
  42         wLock.unlock();
  43         wLock = null;
  44     }
  45 }