mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-06-16 17:55:15 +00:00
cd1aa771b4
Oh AVMLockingServiceTest will fail if not started from a clean database. I'll fix that up in a bit. git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@5845 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
490 lines
17 KiB
Java
490 lines
17 KiB
Java
/*
|
|
* Copyright (C) 2005-2007 Alfresco Software Limited.
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
* As a special exception to the terms and conditions of version 2.0 of
|
|
* the GPL, you may redistribute this Program in connection with Free/Libre
|
|
* and Open Source Software ("FLOSS") applications as described in Alfresco's
|
|
* FLOSS exception. You should have recieved a copy of the text describing
|
|
* the FLOSS exception, and it is also available here:
|
|
* http://www.alfresco.com/legal/licensing
|
|
*/
|
|
|
|
package org.alfresco.repo.avm.locking;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
|
|
import org.alfresco.repo.attributes.Attribute;
|
|
import org.alfresco.repo.attributes.ListAttributeValue;
|
|
import org.alfresco.repo.attributes.MapAttributeValue;
|
|
import org.alfresco.repo.attributes.StringAttributeValue;
|
|
import org.alfresco.repo.transaction.RetryingTransactionHelper;
|
|
import org.alfresco.service.cmr.attributes.AttrQueryEquals;
|
|
import org.alfresco.service.cmr.attributes.AttributeService;
|
|
import org.alfresco.service.cmr.avm.AVMBadArgumentException;
|
|
import org.alfresco.service.cmr.avm.AVMExistsException;
|
|
import org.alfresco.service.cmr.avm.AVMNotFoundException;
|
|
import org.alfresco.service.cmr.avm.locking.AVMLock;
|
|
import org.alfresco.service.cmr.avm.locking.AVMLockingService;
|
|
import org.alfresco.util.MD5;
|
|
import org.alfresco.util.Pair;
|
|
|
|
/**
|
|
* Implementation of the lock service.
|
|
* @author britt
|
|
*/
|
|
public class AVMLockingServiceImpl implements AVMLockingService
|
|
{
|
|
public static final String LOCK_TABLE = ".avm_lock_table";
|
|
public static final String WEB_PROJECTS = "web_projects";
|
|
public static final String USERS = "users";
|
|
public static final String STORES = "stores";
|
|
|
|
/**
|
|
* AttributeService reference.
|
|
*/
|
|
private AttributeService fAttributeService;
|
|
|
|
/**
|
|
* Transaction Helper reference.
|
|
*/
|
|
private RetryingTransactionHelper fRetryingTransactionHelper;
|
|
|
|
public AVMLockingServiceImpl()
|
|
{
|
|
}
|
|
|
|
/**
|
|
* Setter for AttributeService reference.
|
|
* @param service
|
|
*/
|
|
public void setAttributeService(AttributeService service)
|
|
{
|
|
fAttributeService = service;
|
|
}
|
|
|
|
/**
|
|
* Setter for RetryingTransactionHelper reference.
|
|
* @param helper
|
|
*/
|
|
public void setRetryingTransactionHelper(RetryingTransactionHelper helper)
|
|
{
|
|
fRetryingTransactionHelper = helper;
|
|
}
|
|
|
|
public void init()
|
|
{
|
|
RetryingTransactionHelper.RetryingTransactionCallback<Object> callback =
|
|
new RetryingTransactionHelper.RetryingTransactionCallback<Object>()
|
|
{
|
|
public Object execute()
|
|
{
|
|
Attribute table = fAttributeService.getAttribute(LOCK_TABLE);
|
|
if (table != null)
|
|
{
|
|
return null;
|
|
}
|
|
fAttributeService.setAttribute("", LOCK_TABLE, new MapAttributeValue());
|
|
fAttributeService.setAttribute(LOCK_TABLE, WEB_PROJECTS, new MapAttributeValue());
|
|
fAttributeService.setAttribute(LOCK_TABLE, USERS, new MapAttributeValue());
|
|
fAttributeService.setAttribute(LOCK_TABLE, STORES, new MapAttributeValue());
|
|
return null;
|
|
}
|
|
};
|
|
fRetryingTransactionHelper.doInTransaction(callback, false);
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#getLock(java.lang.String, java.lang.String)
|
|
*/
|
|
public AVMLock getLock(String webProject, String path)
|
|
{
|
|
path = normalizePath(path);
|
|
List<String> keys = new ArrayList<String>();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(WEB_PROJECTS);
|
|
keys.add(webProject);
|
|
List<Pair<String, Attribute>> attrs =
|
|
fAttributeService.query(keys, new AttrQueryEquals(MD5.Digest(path.getBytes())));
|
|
if (attrs.size() == 0)
|
|
{
|
|
return null;
|
|
}
|
|
return new AVMLock(attrs.get(0).getSecond());
|
|
}
|
|
|
|
/**
|
|
* Utility to get relative paths into canonical form.
|
|
* @param path The incoming path.
|
|
* @return The normalized path.
|
|
*/
|
|
private String normalizePath(String path)
|
|
{
|
|
while (path.startsWith("/"))
|
|
{
|
|
path = path.substring(1);
|
|
}
|
|
while (path.endsWith("/"))
|
|
{
|
|
path = path.substring(0, path.length() - 1);
|
|
}
|
|
return path.replaceAll("/+", "/");
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#getUsersLocks(java.lang.String)
|
|
*/
|
|
public List<AVMLock> getUsersLocks(String user)
|
|
{
|
|
List<String> keys = new ArrayList<String>();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(USERS);
|
|
keys.add(user);
|
|
Attribute userLocks = fAttributeService.getAttribute(keys);
|
|
List<AVMLock> locks = new ArrayList<AVMLock>();
|
|
if (userLocks == null)
|
|
{
|
|
return locks;
|
|
}
|
|
for (Attribute entry : userLocks)
|
|
{
|
|
String webProject = entry.get("web_project").getStringValue();
|
|
String path = entry.get("path").getStringValue();
|
|
locks.add(getLock(webProject, path));
|
|
}
|
|
return locks;
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#lockPath(org.alfresco.service.cmr.avm.locking.AVMLock)
|
|
*/
|
|
public void lockPath(AVMLock lock)
|
|
{
|
|
List<String> keys = new ArrayList<String>();
|
|
Attribute lockData = lock.getAttribute();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(WEB_PROJECTS);
|
|
keys.add(lock.getWebProject());
|
|
if (fAttributeService.getAttribute(keys) == null)
|
|
{
|
|
throw new AVMExistsException("Lock Exists: " + keys);
|
|
}
|
|
fAttributeService.setAttribute(keys, MD5.Digest(lock.getPath().getBytes()), lockData);
|
|
Attribute reverseEntry = new MapAttributeValue();
|
|
reverseEntry.put("web_project", new StringAttributeValue(lock.getWebProject()));
|
|
reverseEntry.put("path", new StringAttributeValue(lock.getPath()));
|
|
keys.clear();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(USERS);
|
|
for (String user : lock.getOwners())
|
|
{
|
|
keys.add(user);
|
|
Attribute userEntry = fAttributeService.getAttribute(keys);
|
|
keys.remove(2);
|
|
if (userEntry == null)
|
|
{
|
|
fAttributeService.setAttribute(keys, user, new ListAttributeValue());
|
|
}
|
|
keys.add(user);
|
|
fAttributeService.addAttribute(keys, reverseEntry);
|
|
keys.remove(2);
|
|
}
|
|
String store = lock.getStore();
|
|
keys.clear();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(STORES);
|
|
keys.add(store);
|
|
Attribute storeEntry = fAttributeService.getAttribute(keys);
|
|
keys.remove(2);
|
|
if (storeEntry == null)
|
|
{
|
|
fAttributeService.setAttribute(keys, store, new ListAttributeValue());
|
|
}
|
|
keys.add(store);
|
|
fAttributeService.addAttribute(keys, reverseEntry);
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#removeLock(java.lang.String, java.lang.String)
|
|
*/
|
|
public void removeLock(String webProject, String path)
|
|
{
|
|
path = normalizePath(path);
|
|
String pathKey = MD5.Digest(path.getBytes());
|
|
List<String> keys = new ArrayList<String>();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(WEB_PROJECTS);
|
|
keys.add(webProject);
|
|
keys.add(pathKey);
|
|
Attribute lockData = fAttributeService.getAttribute(keys);
|
|
if (lockData == null)
|
|
{
|
|
throw new AVMNotFoundException("Lock does not exist: " + webProject + " " + path);
|
|
}
|
|
keys.remove(3);
|
|
fAttributeService.removeAttribute(keys, pathKey);
|
|
AVMLock lock = new AVMLock(lockData);
|
|
List<String> userKeys = new ArrayList<String>();
|
|
userKeys.add(LOCK_TABLE);
|
|
userKeys.add(USERS);
|
|
for (String user : lock.getOwners())
|
|
{
|
|
userKeys.add(user);
|
|
Attribute userLocks = fAttributeService.getAttribute(userKeys);
|
|
for (int i = userLocks.size() - 1; i >= 0; i--)
|
|
{
|
|
Attribute lockInfo = userLocks.get(i);
|
|
if (lockInfo.get("web_project").getStringValue().equals(lock.getWebProject())
|
|
&& lockInfo.get("path").getStringValue().equals(lock.getPath()))
|
|
{
|
|
fAttributeService.removeAttribute(userKeys, i);
|
|
break;
|
|
}
|
|
}
|
|
userKeys.remove(2);
|
|
}
|
|
List<String> storeKeys = new ArrayList<String>();
|
|
storeKeys.add(LOCK_TABLE);
|
|
storeKeys.add(STORES);
|
|
String store = lock.getStore();
|
|
storeKeys.add(store);
|
|
Attribute storeLocks = fAttributeService.getAttribute(storeKeys);
|
|
for (int i = storeLocks.size() - 1; i >= 0; i--)
|
|
{
|
|
Attribute lockInfo = storeLocks.get(i);
|
|
if (lockInfo.get("web_project").getStringValue().equals(lock.getWebProject()) &&
|
|
lockInfo.get("path").getStringValue().equals(lock.getPath()))
|
|
{
|
|
fAttributeService.removeAttribute(storeKeys, i);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#addWebProject(java.lang.String)
|
|
*/
|
|
public void addWebProject(String webProject)
|
|
{
|
|
List<String> keys = new ArrayList<String>();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(WEB_PROJECTS);
|
|
keys.add(webProject);
|
|
if (fAttributeService.getAttribute(keys) != null)
|
|
{
|
|
throw new AVMExistsException("Web Project Exists: " + webProject);
|
|
}
|
|
keys.remove(2);
|
|
fAttributeService.setAttribute(keys, webProject, new MapAttributeValue());
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#getWebProjectLocks(java.lang.String)
|
|
*/
|
|
public List<AVMLock> getWebProjectLocks(String webProject)
|
|
{
|
|
List<String> keys = new ArrayList<String>();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(WEB_PROJECTS);
|
|
keys.add(webProject);
|
|
Attribute locksMap = fAttributeService.getAttribute(keys);
|
|
List<AVMLock> result = new ArrayList<AVMLock>();
|
|
if (locksMap != null)
|
|
{
|
|
for (Attribute lockData : locksMap.values())
|
|
{
|
|
result.add(new AVMLock(lockData));
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#removeWebProject(java.lang.String)
|
|
*/
|
|
public void removeWebProject(String webProject)
|
|
{
|
|
List<String> userKeys = new ArrayList<String>();
|
|
userKeys.add(LOCK_TABLE);
|
|
userKeys.add(USERS);
|
|
List<String> users = fAttributeService.getKeys(userKeys);
|
|
// TODO This works incredibly slowly. AttributeService has to support
|
|
// extended querying on values.
|
|
for (String user : users)
|
|
{
|
|
userKeys.add(user);
|
|
Attribute userLocks = fAttributeService.getAttribute(userKeys);
|
|
Iterator<Attribute> iter = userLocks.iterator();
|
|
while (iter.hasNext())
|
|
{
|
|
Attribute lockInfo = iter.next();
|
|
if (lockInfo.get("web_project").getStringValue().equals(webProject))
|
|
{
|
|
iter.remove();
|
|
}
|
|
}
|
|
userKeys.remove(2);
|
|
fAttributeService.setAttribute(userKeys, user, userLocks);
|
|
}
|
|
List<String> storeKeys = new ArrayList<String>();
|
|
storeKeys.add(LOCK_TABLE);
|
|
storeKeys.add(STORES);
|
|
List<String> stores = fAttributeService.getKeys(storeKeys);
|
|
// TODO Ditto.
|
|
for (String store : stores)
|
|
{
|
|
storeKeys.add(store);
|
|
Attribute storeLocks = fAttributeService.getAttribute(storeKeys);
|
|
Iterator<Attribute> iter = storeLocks.iterator();
|
|
while (iter.hasNext())
|
|
{
|
|
Attribute lockInfo = iter.next();
|
|
if (lockInfo.get("web_project").getStringValue().equals(webProject))
|
|
{
|
|
iter.remove();
|
|
}
|
|
}
|
|
storeKeys.remove(2);
|
|
fAttributeService.setAttribute(storeKeys, store, storeLocks);
|
|
}
|
|
List<String> keys = new ArrayList<String>();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(WEB_PROJECTS);
|
|
fAttributeService.removeAttribute(keys, webProject);
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#getStoreLocks(java.lang.String)
|
|
*/
|
|
public List<AVMLock> getStoreLocks(String store)
|
|
{
|
|
List<AVMLock> locks = new ArrayList<AVMLock>();
|
|
List<String> keys = new ArrayList<String>();
|
|
keys.add(LOCK_TABLE);
|
|
keys.add(STORES);
|
|
keys.add(store);
|
|
List<String> lockKeys = new ArrayList<String>();
|
|
lockKeys.add(LOCK_TABLE);
|
|
lockKeys.add(WEB_PROJECTS);
|
|
Attribute storeLocks = fAttributeService.getAttribute(keys);
|
|
for (Attribute lockInfo : storeLocks)
|
|
{
|
|
String webProject = lockInfo.get("web_project").getStringValue();
|
|
String path = lockInfo.get("path").getStringValue();
|
|
lockKeys.add(webProject);
|
|
lockKeys.add(MD5.Digest(path.getBytes()));
|
|
Attribute lockData = fAttributeService.getAttribute(lockKeys);
|
|
locks.add(new AVMLock(lockData));
|
|
lockKeys.remove(3);
|
|
lockKeys.remove(2);
|
|
}
|
|
return locks;
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#modifyLock(java.lang.String, java.lang.String, java.lang.String, java.lang.String, java.util.List, java.util.List)
|
|
*/
|
|
public void modifyLock(String webProject, String path, String newPath, String newStore, List<String> usersToRemove, List<String> usersToAdd)
|
|
{
|
|
AVMLock lock = getLock(webProject, path);
|
|
if (lock == null)
|
|
{
|
|
throw new AVMNotFoundException("Lock not found for " + webProject + ":" + path);
|
|
}
|
|
removeLock(webProject, path);
|
|
if (newPath != null)
|
|
{
|
|
lock.setPath(newPath);
|
|
}
|
|
if (newStore != null)
|
|
{
|
|
lock.setStore(newStore);
|
|
}
|
|
if (usersToRemove != null)
|
|
{
|
|
for (String user : usersToRemove)
|
|
{
|
|
lock.getOwners().remove(user);
|
|
}
|
|
}
|
|
if (usersToAdd != null)
|
|
{
|
|
for (String user : usersToAdd)
|
|
{
|
|
if (lock.getOwners().contains(user))
|
|
{
|
|
continue;
|
|
}
|
|
lock.getOwners().add(user);
|
|
}
|
|
}
|
|
lockPath(lock);
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#removeStoreLocks(java.lang.String)
|
|
*/
|
|
public void removeStoreLocks(String store)
|
|
{
|
|
List<String> storeKeys = new ArrayList<String>();
|
|
storeKeys.add(LOCK_TABLE);
|
|
storeKeys.add(STORES);
|
|
storeKeys.add(store);
|
|
Attribute storeLocks = fAttributeService.getAttribute(storeKeys);
|
|
if (storeLocks == null)
|
|
{
|
|
return;
|
|
}
|
|
for (Attribute lockInfo : storeLocks)
|
|
{
|
|
removeLock(lockInfo.get("web_project").getStringValue(),
|
|
lockInfo.get("path").getStringValue());
|
|
}
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see org.alfresco.service.cmr.avm.locking.AVMLockingService#motherMayI(java.lang.String, java.lang.String)
|
|
*/
|
|
public boolean motherMayI(String webProject, String avmPath, String user)
|
|
{
|
|
String[] storePath = avmPath.split(":");
|
|
if (storePath.length != 2)
|
|
{
|
|
throw new AVMBadArgumentException("Malformed AVM Path : " + avmPath);
|
|
}
|
|
AVMLock lock = getLock(webProject, storePath[1]);
|
|
if (lock == null)
|
|
{
|
|
return true;
|
|
}
|
|
if (!lock.getStore().equals(storePath[0]))
|
|
{
|
|
return false;
|
|
}
|
|
List<String> owners = lock.getOwners();
|
|
if (owners.contains(user))
|
|
{
|
|
return true;
|
|
}
|
|
// TODO Figure out how the users, groups, roles mess works and give an answer
|
|
// based on that. <rant>Our entire access control and user, group, role approach
|
|
// is apalling and needs to be replaced.</rant>
|
|
return false;
|
|
}
|
|
}
|