Merged HEAD (5.1) to 5.1.N (5.1.1)

117473 bhorje: CM-690 trait extension


git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/BRANCHES/DEV/5.1.N/root@117558 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261

source/java/org/alfresco/traitextender/ExtendedTrait.java

@@ -0,0 +1,62 @@
+/* 
+ * Copyright (C) 2005-2015 Alfresco Software Limited.
+ *
+ * This file is part of Alfresco
+ *
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Alfresco 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 Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see http://www.gnu.org/licenses/.
+ */
+package org.alfresco.traitextender;
+
+import java.util.concurrent.ConcurrentHashMap;
+
+public class ExtendedTrait<T extends Trait>
+{
+    private ConcurrentHashMap<Class<?>, Object> extensions = new ConcurrentHashMap<Class<?>, Object>();
+
+    private T trait;
+
+    public ExtendedTrait(T trait)
+    {
+        super();
+        this.trait = trait;
+    }
+
+    public T getTrait()
+    {
+        return trait;
+    }
+
+    public <E> E getExtension(Class<E> extensionAPI)
+    {
+        @SuppressWarnings("unchecked")
+        E extension=(E) extensions.get(extensionAPI);
+        
+        return extension;
+    }
+    
+    public synchronized <E> E extend(Class<E> extensionAPI, ExtensionFactory<E> factory)
+    {
+        @SuppressWarnings("unchecked")
+        E extension=(E) extensions.get(extensionAPI);
+        
+        if (extension==null)
+        {
+            extension = factory.createExtension(trait);
+            extensions.put(extensionAPI, extension);
+            
+        }
+        
+        return extension;
+    }
+}