コード例 #1
0
  /**
   * {@inheritDoc}
   *
   * @see org.komodo.shell.BuiltInShellCommand#tabCompletion(java.lang.String, java.util.List)
   */
  @Override
  public TabCompletionModifier tabCompletion(
      final String lastArgument, final List<CharSequence> candidates) throws Exception {
    final Arguments args = getArguments();

    try {
      List<String> existingDatasourceNames =
          ServerUtils.getDatasourceNames(getWorkspaceTeiidInstance());
      Collections.sort(existingDatasourceNames);

      if (args.isEmpty()) {
        if (lastArgument == null) {
          candidates.addAll(existingDatasourceNames);
        } else {
          for (final String item : existingDatasourceNames) {
            if (item.startsWith(lastArgument)) {
              candidates.add(item);
            }
          }
        }
      }
    } catch (Exception ex) {
      print();
      print(MESSAGE_INDENT, I18n.bind(ServerCommandsI18n.connectionErrorWillDisconnect));
      WkspStatusServerManager.getInstance(getWorkspaceStatus()).disconnectDefaultServer();
    }

    return TabCompletionModifier.AUTO;
  }
コード例 #2
0
  /**
   * {@inheritDoc}
   *
   * @see org.komodo.shell.BuiltInShellCommand#doExecute()
   */
  @Override
  protected CommandResult doExecute() {
    CommandResult result = null;

    try {
      String datasourceName =
          requiredArgument(0, I18n.bind(ServerCommandsI18n.missingDatasourceName));

      final String overwriteArg = optionalArgument(1, null);
      final boolean overwrite = !StringUtils.isBlank(overwriteArg);
      // make sure overwrite arg is valid
      if (overwrite && !VALID_OVERWRITE_ARGS.contains(overwriteArg)) {
        return new CommandResultImpl(
            false, I18n.bind(WorkspaceCommandsI18n.overwriteArgInvalid, overwriteArg), null);
      }

      // If datasource with same name is in workspace, make sure we can overwrite
      boolean hasDS =
          getWorkspaceManager(getTransaction())
              .hasChild(getTransaction(), datasourceName, DataVirtLexicon.Connection.NODE_TYPE);
      if (hasDS && !overwrite) {
        return new CommandResultImpl(
            false,
            I18n.bind(ServerCommandsI18n.datasourceOverwriteNotEnabled, datasourceName),
            null);
      }

      // Validates that a server is connected
      CommandResult validationResult = validateHasConnectedWorkspaceServer();
      if (!validationResult.isOk()) {
        return validationResult;
      }

      // Get the Data Source from the server
      TeiidDataSource serverDS = null;
      try {
        // Check the data source name to make sure its valid
        List<String> existingSourceNames =
            ServerUtils.getDatasourceNames(getWorkspaceTeiidInstance());
        if (!existingSourceNames.contains(datasourceName)) {
          return new CommandResultImpl(
              false, I18n.bind(ServerCommandsI18n.serverDatasourceNotFound, datasourceName), null);
        }
        // Get the data source
        serverDS = getWorkspaceTeiidInstance().getDataSource(datasourceName);
      } catch (Exception ex) {
        result =
            new CommandResultImpl(
                false, I18n.bind(ServerCommandsI18n.connectionErrorWillDisconnect), ex);
        WkspStatusServerManager.getInstance(getWorkspaceStatus()).disconnectDefaultServer();
        return result;
      }
      if (serverDS == null) {
        return new CommandResultImpl(
            false, I18n.bind(ServerCommandsI18n.serverDatasourceNotFound, datasourceName), null);
      }

      // If overwriting, delete existing first
      if (hasDS) {
        final KomodoObject datasourceToDelete =
            getWorkspaceManager(getTransaction())
                .getChild(getTransaction(), datasourceName, DataVirtLexicon.Connection.NODE_TYPE);
        getWorkspaceManager(getTransaction()).delete(getTransaction(), datasourceToDelete);
      }
      // Create the Data Source and set properties
      Datasource newDatasource =
          getWorkspaceManager(getTransaction())
              .createDatasource(getTransaction(), null, datasourceName);
      setRepoDatasourceProperties(newDatasource, serverDS.getProperties());

      print(MESSAGE_INDENT, I18n.bind(ServerCommandsI18n.datasourceCopyToRepoFinished));
      result = CommandResult.SUCCESS;
    } catch (final Exception e) {
      result = new CommandResultImpl(e);
    }

    return result;
  }