MT extension

git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@6785 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261

source/java/org/alfresco/repo/content/TenantRoutingFileContentStore.java

@@ -0,0 +1,139 @@
+/*
+ * 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.content;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.alfresco.repo.content.filestore.FileContentStore;
+import org.alfresco.repo.security.authentication.AuthenticationUtil;
+import org.alfresco.repo.tenant.Tenant;
+import org.alfresco.repo.tenant.TenantDeployer;
+import org.alfresco.repo.tenant.TenantService;
+
+/**
+ * Content Store that supports tenant routing, if multi-tenancy is enabled.
+ * 
+ * Note: Need to initialise before the dictionary service, in the case that models are dynamically loaded for the tenant.
+ */
+public class TenantRoutingFileContentStore extends AbstractRoutingContentStore implements TenantDeployer
+{
+    Map<String, FileContentStore> tenantFileStores = new HashMap<String, FileContentStore>();
+    
+    private String defaultRootDirectory;
+    private TenantService tenantService;
+
+
+    public void setDefaultRootDir(String defaultRootDirectory)
+    {
+        this.defaultRootDirectory = defaultRootDirectory;
+    }
+    
+    public void setTenantService(TenantService tenantService)
+    {
+        this.tenantService = tenantService;
+    }
+    
+    protected ContentStore selectWriteStore(ContentContext ctx)
+    {
+        return getTenantFileStore(tenantService.getCurrentUserDomain());
+    }
+    
+    public List<ContentStore> getAllStores()
+    {
+        if (tenantService.isEnabled())
+        {
+            String currentUser = AuthenticationUtil.getCurrentUserName();
+            if ((currentUser == null) || (currentUser.equals(AuthenticationUtil.getSystemUserName())))
+            {
+                // return enabled stores across all tenants, if running as system/null user, for example, ContentStoreCleaner scheduled job      
+                List<ContentStore> allEnabledStores = new ArrayList<ContentStore>();
+                for (String tenantDomain : tenantFileStores.keySet())
+                {
+                    allEnabledStores.add(tenantFileStores.get(tenantDomain)); // note: cache should only contain enabled stores
+                }
+                return allEnabledStores;
+            }
+        }
+        return Arrays.asList(getTenantFileStore(tenantService.getCurrentUserDomain()));
+    }
+    
+    private ContentStore getTenantFileStore(String tenantDomain)
+    {
+        return tenantFileStores.get(tenantDomain);
+    }
+    
+    private void putTenantFileStore(String tenantDomain, FileContentStore fileStore)
+    {
+        tenantFileStores.put(tenantDomain, fileStore);
+    }
+    
+    private void removeTenantFileStore(String tenantDomain)
+    {
+        tenantFileStores.remove(tenantDomain);
+    }
+   
+    public void init()
+    {
+        String tenantDomain = "";
+        String rootDir = defaultRootDirectory;
+        
+        Tenant tenant = tenantService.getTenant(tenantService.getCurrentUserDomain());
+        if (tenant != null)
+        {
+            if (tenant.getRootContentStoreDir() != null)
+            {
+               rootDir = tenant.getRootContentStoreDir();
+            }
+            tenantDomain = tenant.getTenantDomain();
+        }
+
+        putTenantFileStore(tenantDomain, new FileContentStore(new File(rootDir)));
+    }
+    
+    public void destroy()
+    {
+        removeTenantFileStore(tenantService.getCurrentUserDomain());
+    }
+    
+    public void onEnableTenant()
+    {
+        init();
+    }
+    
+    public void onDisableTenant()
+    {
+        destroy();
+    }
+    
+    public String getDefaultRootDir()
+    {
+        return this.defaultRootDirectory;
+    }
+}