Search in sources :

Example 11 with DBNContainer

use of org.jkiss.dbeaver.model.navigator.DBNContainer in project dbeaver by dbeaver.

the class ModelPropertyTester method test.

@Override
public boolean test(Object receiver, String property, Object[] args, Object expectedValue) {
    if (!(receiver instanceof DBSObject)) {
        return false;
    }
    Display display = Display.getCurrent();
    if (display == null) {
        return false;
    }
    switch(property) {
        case PROP_CHILD_OF_TYPE:
            {
                {
                    DBSObject object = (DBSObject) receiver;
                    if (object instanceof DBSObjectContainer) {
                        if (expectedValue instanceof String) {
                            try {
                                Class<?> expectedChildClass = Class.forName((String) expectedValue);
                                Class<? extends DBSObject> childType = ((DBSObjectContainer) object).getPrimaryChildType(null);
                                return expectedChildClass.isAssignableFrom(childType);
                            } catch (Exception e) {
                                return false;
                            }
                        }
                    }
                }
                return false;
            }
        case PROP_IS_TABLE_CONTAINER:
            {
                DBSObject object = DBUtils.getPublicObject((DBSObject) receiver);
                if (object instanceof DBNContainer) {
                    Object valueObject = ((DBNContainer) object).getValueObject();
                    if (valueObject instanceof DBSObject) {
                        object = (DBSObject) valueObject;
                    }
                }
                if (object instanceof DBSSchema) {
                    return true;
                }
                if (object instanceof DBSObjectContainer) {
                    try {
                        Class<? extends DBSObject> primaryChildType = ((DBSObjectContainer) object).getPrimaryChildType(null);
                        if (DBSDataContainer.class.isAssignableFrom(primaryChildType)) {
                            return true;
                        }
                    } catch (DBException e) {
                        log.debug(e);
                    }
                }
                return false;
            }
    }
    return false;
}
Also used : DBException(org.jkiss.dbeaver.DBException) DBNContainer(org.jkiss.dbeaver.model.navigator.DBNContainer) DBException(org.jkiss.dbeaver.DBException) DBSSchema(org.jkiss.dbeaver.model.struct.rdb.DBSSchema) DBSObject(org.jkiss.dbeaver.model.struct.DBSObject) DBSObjectContainer(org.jkiss.dbeaver.model.struct.DBSObjectContainer) DBSObject(org.jkiss.dbeaver.model.struct.DBSObject) DBSDataContainer(org.jkiss.dbeaver.model.struct.DBSDataContainer) Display(org.eclipse.swt.widgets.Display)

Example 12 with DBNContainer

use of org.jkiss.dbeaver.model.navigator.DBNContainer in project dbeaver by dbeaver.

the class NavigatorHandlerObjectMove method execute.

@Override
public Object execute(ExecutionEvent event) throws ExecutionException {
    final ISelection selection = HandlerUtil.getCurrentSelection(event);
    DBNNode node = NavigatorUtils.getSelectedNode(selection);
    if (node == null || !(node.getParentNode() instanceof DBNContainer)) {
        return null;
    }
    DBSObject object = ((DBNDatabaseNode) node).getObject();
    if (!(object instanceof DBPOrderedObject)) {
        return null;
    }
    @SuppressWarnings("unchecked") DBEObjectReorderer<DBSObject> objectReorderer = DBWorkbench.getPlatform().getEditorsRegistry().getObjectManager(object.getClass(), DBEObjectReorderer.class);
    if (objectReorderer == null) {
        return null;
    }
    DBPOrderedObject orderedObject = (DBPOrderedObject) object;
    try {
        // Sibling objects - they are involved in reordering process
        List<DBSObject> siblingObjects = new ArrayList<>();
        for (DBNNode siblingNode : node.getParentNode().getChildren(new VoidProgressMonitor())) {
            if (siblingNode instanceof DBNDatabaseNode) {
                DBSObject siblingObject = ((DBNDatabaseNode) siblingNode).getObject();
                if (siblingObject.getClass() != object.getClass()) {
                    log.warn("Sibling object class " + siblingObject.getClass() + " differs from moving object class " + object.getClass().getName());
                } else {
                    siblingObjects.add(siblingObject);
                }
            } else {
                log.warn("Wrong sibling node type: " + siblingNode);
            }
        }
        CommandTarget commandTarget = getCommandTarget(HandlerUtil.getActiveWorkbenchWindow(event), node.getParentNode(), object.getClass(), false);
        String actionId = event.getCommand().getId();
        switch(actionId) {
            case NavigatorCommands.CMD_OBJECT_MOVE_UP:
                objectReorderer.setObjectOrdinalPosition(commandTarget.getContext(), object, siblingObjects, orderedObject.getOrdinalPosition() - 1);
                break;
            case NavigatorCommands.CMD_OBJECT_MOVE_DOWN:
                objectReorderer.setObjectOrdinalPosition(commandTarget.getContext(), object, siblingObjects, orderedObject.getOrdinalPosition() + 1);
                break;
        }
        if (object.isPersisted() && commandTarget.getEditor() == null) {
            Map<String, Object> options = DBPScriptObject.EMPTY_OPTIONS;
            if (!showScript(HandlerUtil.getActiveWorkbenchWindow(event), commandTarget.getContext(), options, "Reorder script")) {
                commandTarget.getContext().resetChanges(true);
                return false;
            } else {
                ObjectSaver orderer = new ObjectSaver(commandTarget.getContext(), options);
                TasksJob.runTask("Change object '" + object.getName() + "' position", orderer);
            }
        }
    } catch (DBException e) {
        DBWorkbench.getPlatformUI().showError("Object move", "Error during object reposition", e);
    }
    return null;
}
Also used : DBException(org.jkiss.dbeaver.DBException) DBPOrderedObject(org.jkiss.dbeaver.model.DBPOrderedObject) DBNNode(org.jkiss.dbeaver.model.navigator.DBNNode) DBNContainer(org.jkiss.dbeaver.model.navigator.DBNContainer) ArrayList(java.util.ArrayList) DBSObject(org.jkiss.dbeaver.model.struct.DBSObject) ISelection(org.eclipse.jface.viewers.ISelection) DBPOrderedObject(org.jkiss.dbeaver.model.DBPOrderedObject) DBSObject(org.jkiss.dbeaver.model.struct.DBSObject) DBPScriptObject(org.jkiss.dbeaver.model.DBPScriptObject) VoidProgressMonitor(org.jkiss.dbeaver.model.runtime.VoidProgressMonitor) DBNDatabaseNode(org.jkiss.dbeaver.model.navigator.DBNDatabaseNode)

Aggregations

DBNContainer (org.jkiss.dbeaver.model.navigator.DBNContainer)12 DBSObject (org.jkiss.dbeaver.model.struct.DBSObject)12 DBException (org.jkiss.dbeaver.DBException)8 DBPScriptObject (org.jkiss.dbeaver.model.DBPScriptObject)7 DBNNode (org.jkiss.dbeaver.model.navigator.DBNNode)6 DBNDatabaseNode (org.jkiss.dbeaver.model.navigator.DBNDatabaseNode)5 ArrayList (java.util.ArrayList)4 ISelection (org.eclipse.jface.viewers.ISelection)4 DBPOrderedObject (org.jkiss.dbeaver.model.DBPOrderedObject)4 DBECommand (org.jkiss.dbeaver.model.edit.DBECommand)4 DBEObjectRenamer (org.jkiss.dbeaver.model.edit.DBEObjectRenamer)4 VoidProgressMonitor (org.jkiss.dbeaver.model.runtime.VoidProgressMonitor)3 Display (org.eclipse.swt.widgets.Display)2 DBPRefreshableObject (org.jkiss.dbeaver.model.DBPRefreshableObject)2 DBEObjectMaker (org.jkiss.dbeaver.model.edit.DBEObjectMaker)2 DBSDataContainer (org.jkiss.dbeaver.model.struct.DBSDataContainer)2 DBSObjectContainer (org.jkiss.dbeaver.model.struct.DBSObjectContainer)2 DBSSchema (org.jkiss.dbeaver.model.struct.rdb.DBSSchema)2 EntityEditorsRegistry (org.jkiss.dbeaver.registry.editor.EntityEditorsRegistry)2 DBPDataSource (org.jkiss.dbeaver.model.DBPDataSource)1