tikhomirov@617: /*
tikhomirov@617:  * Copyright (c) 2013 TMate Software Ltd
tikhomirov@617:  *  
tikhomirov@617:  * This program is free software; you can redistribute it and/or modify
tikhomirov@617:  * it under the terms of the GNU General Public License as published by
tikhomirov@617:  * the Free Software Foundation; version 2 of the License.
tikhomirov@617:  *
tikhomirov@617:  * This program is distributed in the hope that it will be useful,
tikhomirov@617:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@617:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@617:  * GNU General Public License for more details.
tikhomirov@617:  *
tikhomirov@617:  * For information on how to redistribute this software under
tikhomirov@617:  * the terms of a license other than GNU General Public License
tikhomirov@617:  * contact TMate Software at support@hg4j.com
tikhomirov@617:  */
tikhomirov@617: package org.tmatesoft.hg.internal;
tikhomirov@617: 
tikhomirov@617: import static org.tmatesoft.hg.util.LogFacility.Severity.Error;
tikhomirov@617: 
tikhomirov@617: import org.tmatesoft.hg.core.HgRepositoryLockException;
tikhomirov@617: import org.tmatesoft.hg.repo.HgRepository;
tikhomirov@617: import org.tmatesoft.hg.repo.HgRepositoryLock;
tikhomirov@617: import org.tmatesoft.hg.util.LogFacility;
tikhomirov@617: 
tikhomirov@617: /**
tikhomirov@617:  * Helper to lock both storage and working directory
tikhomirov@617:  * 
tikhomirov@617:  * @author Artem Tikhomirov
tikhomirov@617:  * @author TMate Software Ltd.
tikhomirov@617:  */
tikhomirov@617: public final class CompleteRepoLock {
tikhomirov@617: 
tikhomirov@617: 	private final HgRepository repo;
tikhomirov@617: 	private HgRepositoryLock wdLock, storeLock;
tikhomirov@617: 
tikhomirov@617: 	public CompleteRepoLock(HgRepository hgRepo) {
tikhomirov@617: 		repo = hgRepo;
tikhomirov@617: 	}
tikhomirov@617: 
tikhomirov@617: 	public void acquire() throws HgRepositoryLockException {
tikhomirov@617: 		wdLock = repo.getWorkingDirLock();
tikhomirov@617: 		storeLock = repo.getStoreLock();
tikhomirov@617: 		wdLock.acquire();
tikhomirov@617: 		try {
tikhomirov@617: 			storeLock.acquire();
tikhomirov@617: 		} catch (HgRepositoryLockException ex) {
tikhomirov@617: 			try {
tikhomirov@617: 				wdLock.release();
tikhomirov@617: 			} catch (HgRepositoryLockException e2) {
tikhomirov@617: 				final LogFacility log = repo.getSessionContext().getLog();
tikhomirov@617: 				log.dump(getClass(), Error, e2, "Nested exception ignored once failed to acquire store lock");
tikhomirov@617: 			}
tikhomirov@617: 			throw ex;
tikhomirov@617: 		}
tikhomirov@617: 
tikhomirov@617: 	}
tikhomirov@617: 	
tikhomirov@617: 	public void release() throws HgRepositoryLockException {
tikhomirov@617: 		try {
tikhomirov@617: 			storeLock.release();
tikhomirov@617: 		} catch (HgRepositoryLockException ex) {
tikhomirov@617: 			try {
tikhomirov@617: 				wdLock.release();
tikhomirov@617: 			} catch (HgRepositoryLockException e2) {
tikhomirov@617: 				final LogFacility log = repo.getSessionContext().getLog();
tikhomirov@617: 				log.dump(getClass(), Error, e2, "Nested exception ignored when releasing working directory lock");
tikhomirov@617: 			}
tikhomirov@617: 			throw ex;
tikhomirov@617: 		}
tikhomirov@617: 		wdLock.release();
tikhomirov@617: 	}
tikhomirov@617: }