Чистый код. Создание, анализ и рефакторинг

    errorArgumentId = argChar;

    throw e;

  }

  return true;

}

---

private class StringArgumentMarshaler extends ArgumentMarshaler {

  private String stringValue = "";

  public void set(Iterator currentArgument) throws ArgsException {

    try {

      stringValue = currentArgument.next();

    } catch (NoSuchElementException e) {

      errorCode = ErrorCode.MISSING_STRING;

      throw new ArgsException();

    }

  }

  public void set(String s) {

  }

  public Object get() {

    return stringValue;

  }

}

private class IntegerArgumentMarshaler extends ArgumentMarshaler {

  private int intValue = 0;

  public void set(Iterator currentArgument) throws ArgsException {

    String parameter = null;

    try {

      parameter = currentArgument.next();

      set(parameter);

    } catch (NoSuchElementException e) {

      errorCode = ErrorCode.MISSING_INTEGER;

      throw new ArgsException();

    } catch (ArgsException e) {

      errorParameter = parameter;

      errorCode = ErrorCode.INVALID_INTEGER;

      throw e;

    }

  }

  public void set(String s) throws ArgsException {

    try {

      intValue = Integer.parseInt(s);

    } catch (NumberFormatException e) {

      throw new ArgsException();

    }

  }

  public Object get() {

    return intValue;

  }

}

А теперь завершающий штрих: убираем цепочку if-else!

private boolean setArgument(char argChar) throws ArgsException {

  ArgumentMarshaler m = marshalers.get(argChar);

  if (m == null)

    return false;

  try {

    m.set(currentArgument);

    return true;

  } catch (ArgsException e) {

    valid = false;

    errorArgumentId = argChar;

    throw e;

  }

}

Избавляемся от лишних функций в IntegerArgumentMarshaler и слегка чистим код:

private class IntegerArgumentMarshaler extends ArgumentMarshaler {

    private int intValue = 0

    public void set(Iterator currentArgument) throws ArgsException {

      String parameter = null;

      try {

        parameter = currentArgument.next();

        intValue = Integer.parseInt(parameter);

      } catch (NoSuchElementException e) {

        errorCode = ErrorCode.MISSING_INTEGER;

        throw new ArgsException();

      } catch (NumberFormatException e) {

        errorParameter = parameter;

        errorCode = ErrorCode.INVALID_INTEGER;

        throw new ArgsException();

      }

    }

    public Object get() {

      return intValue;

    }

  }

  }

ArgumentMarshaler преобразуется в интерфейс:

private interface ArgumentMarshaler {

  void set(Iterator currentArgument) throws ArgsException;

  Object get();

}

А теперь посмотрите, как легко добавлять новые типы аргументов в эту структуру. Количество изменений минимально, а все изменения логически изолированы. Начнем с добавления нового тестового сценария, проверяющего правильность работы аргументов double:

public void testSimpleDoublePresent() throws Exception {

  Args args = new Args("x##", new String[] {"-x","42.3"});

  assertTrue(args.isValid());

  assertEquals(1, args.cardinality());

  assertTrue(args.has('x'));

  assertEquals(42.3, args.getDouble('x'), .001);

}

Чистим код разбора форматной строки и добавляем обнаружение ## для аргументов типа double.