Object manager model

plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/registry/ObjectManagerDescriptor.java

+/*
+ * DBeaver - Universal Database Manager
+ * Copyright (C) 2010-2017 Serge Rider (serge@jkiss.org)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jkiss.dbeaver.registry;
+
+import org.eclipse.core.runtime.IConfigurationElement;
+import org.jkiss.dbeaver.model.edit.DBEObjectManager;
+import org.jkiss.dbeaver.model.impl.AbstractDescriptor;
+
+/**
+ * ObjectManagerDescriptor
+ */
+public class ObjectManagerDescriptor extends AbstractDescriptor
+{
+    private String id;
+    private ObjectType managerType;
+    private ObjectType objectType;
+    private DBEObjectManager managerInstance;
+
+    ObjectManagerDescriptor(IConfigurationElement config)
+    {
+        super(config);
+
+        this.id = config.getAttribute(RegistryConstants.ATTR_CLASS);
+        this.managerType = new ObjectType(id);
+        this.objectType = new ObjectType(config.getAttribute(RegistryConstants.ATTR_OBJECT_TYPE));
+    }
+
+    void dispose()
+    {
+        objectType = null;
+        managerType = null;
+        managerInstance = null;
+    }
+
+    public String getId()
+    {
+        return id;
+    }
+
+    public ObjectType getObjectType() {
+        return objectType;
+    }
+
+    public boolean appliesToType(Class clazz)
+    {
+        return objectType.matchesType(clazz);
+    }
+
+    public synchronized DBEObjectManager getManager()
+    {
+        if (managerInstance != null) {
+            return managerInstance;
+        }
+        Class<? extends DBEObjectManager> clazz = managerType.getObjectClass(DBEObjectManager.class);
+        if (clazz == null) {
+            throw new IllegalStateException("Can't instantiate object manager '" + managerType.getImplName() + "'");
+        }
+        try {
+            managerInstance = clazz.newInstance();
+        } catch (Throwable ex) {
+            throw new IllegalStateException("Error instantiating object manager '" + clazz.getName() + "'", ex);
+        }
+        return managerInstance;
+    }
+
+    @Override
+    public String toString() {
+        return id;
+    }
+
+}
\ No newline at end of file

plugins/org.jkiss.dbeaver.core/src/org/jkiss/dbeaver/registry/ObjectManagerRegistry.java

+/*
+ * DBeaver - Universal Database Manager
+ * Copyright (C) 2010-2017 Serge Rider (serge@jkiss.org)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jkiss.dbeaver.registry;
+
+import org.eclipse.core.runtime.IConfigurationElement;
+import org.eclipse.core.runtime.IExtensionRegistry;
+import org.eclipse.core.runtime.Platform;
+import org.jkiss.dbeaver.model.edit.DBEObjectManager;
+import org.jkiss.dbeaver.model.edit.DBERegistry;
+import org.jkiss.dbeaver.registry.editor.EntityEditorDescriptor;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * ObjectManagerRegistry
+ */
+public class ObjectManagerRegistry implements DBERegistry {
+
+    private static final String TAG_MANAGER = "manager"; //NON-NLS-1
+
+    private static ObjectManagerRegistry instance = null;
+
+    public synchronized static ObjectManagerRegistry getInstance() {
+        if (instance == null) {
+            instance = new ObjectManagerRegistry(Platform.getExtensionRegistry());
+        }
+        return instance;
+    }
+
+    private List<ObjectManagerDescriptor> entityManagers = new ArrayList<ObjectManagerDescriptor>();
+    private Map<String, ObjectManagerDescriptor> entityManagerMap = new HashMap<>();
+    private Map<String, Boolean> nullEntityManagerMap = new HashMap<>();
+
+    public ObjectManagerRegistry(IExtensionRegistry registry) {
+        // Load datasource providers from external plugins
+        IConfigurationElement[] extElements = registry.getConfigurationElementsFor(EntityEditorDescriptor.EXTENSION_ID);
+        for (IConfigurationElement ext : extElements) {
+            if (TAG_MANAGER.equals(ext.getName())) {
+                ObjectManagerDescriptor descriptor = new ObjectManagerDescriptor(ext);
+                entityManagers.add(descriptor);
+            }
+        }
+        for (ObjectManagerDescriptor em : entityManagers) {
+            entityManagerMap.put(em.getObjectType().getImplName(), em);
+        }
+    }
+
+    public void dispose() {
+        for (ObjectManagerDescriptor descriptor : entityManagers) {
+            descriptor.dispose();
+        }
+        entityManagers.clear();
+        entityManagerMap.clear();
+        nullEntityManagerMap.clear();
+    }
+
+    private ObjectManagerDescriptor getEntityManager(Class objectType) {
+        String targetTypeName = objectType.getName();
+
+        // 1. Try exact match
+        ObjectManagerDescriptor manager = entityManagerMap.get(targetTypeName);
+        if (manager != null) {
+            return manager;
+        }
+        if (nullEntityManagerMap.containsKey(targetTypeName)) {
+            return null;
+        }
+        // 2. Find first applicable
+        for (ObjectManagerDescriptor descriptor : entityManagers) {
+            if (descriptor.appliesToType(objectType)) {
+                entityManagerMap.put(targetTypeName, descriptor);
+                return descriptor;
+            }
+        }
+        // TODO: need to re-validate. Maybe cache will break some lazy loaded bundles?
+        nullEntityManagerMap.put(targetTypeName, Boolean.TRUE);
+        return null;
+    }
+
+    public DBEObjectManager<?> getObjectManager(Class<?> aClass) {
+        ObjectManagerDescriptor entityManager = getEntityManager(aClass);
+        return entityManager == null ? null : entityManager.getManager();
+    }
+
+    public <T> T getObjectManager(Class<?> objectClass, Class<T> managerType) {
+        final DBEObjectManager<?> objectManager = getObjectManager(objectClass);
+        if (objectManager != null && managerType.isAssignableFrom(objectManager.getClass())) {
+            return managerType.cast(objectManager);
+        } else {
+            return null;
+        }
+    }
+
+}

plugins/org.jkiss.dbeaver.ext.postgresql/plugin.xml

@@ -363,6 +363,53 @@
         </repository>
     </extension>
 
+    <!--extension point="org.jkiss.dbeaver.objectManager">
+        <creator
+                class="org.jkiss.dbeaver.ext.postgresql.ui.editors.PostgreDatabaseCreatePage"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreDatabase"
+                type="new"/>
+
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreDatabaseManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreDatabase"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreSchemaManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreSchema"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreTableManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreTable"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreTableColumnManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreTableColumn"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreSequenceManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreSequence"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreConstraintManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreTableConstraint"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreForeignKeyManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreTableForeignKey"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreIndexManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreIndex"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreViewManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreView"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreMViewManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreMaterializedView"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreTriggerManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreTrigger"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreProcedureManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreProcedure"/>
+        <manager
+                class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreRoleManager"
+                objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreRole"/>
+    </extension-->
+
     <extension point="org.jkiss.dbeaver.dataSourceView">
         <view
                 id="org.jkiss.dbeaver.ext.postgresql.dataSourceWizardView"
@@ -405,6 +452,11 @@
     </extension>
 
     <extension point="org.jkiss.dbeaver.databaseEditor">
+        <creator
+            class="org.jkiss.dbeaver.ext.postgresql.ui.editors.PostgreDatabaseCreatePage"
+            objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreDatabase"
+            type="new"/>
+
         <manager
             class="org.jkiss.dbeaver.ext.postgresql.edit.PostgreDatabaseManager"
             objectType="org.jkiss.dbeaver.ext.postgresql.model.PostgreDatabase"/>