br.com.objectos.io.flat.IntegerColumn Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of io-flat Show documentation
flat file utilities.
The newest version!
/*
 * Copyright 2015 Objectos, Fábrica de Software LTDA.
 *
 * 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 br.com.objectos.io.flat;

import java.util.Objects;
import java.util.Set;

import br.com.objectos.io.flat.annotation.IntegerOption;

import com.google.common.base.Strings;
import com.google.common.collect.ImmutableSet;

/**
 * @author [email protected] (Marcio Endo)
 */
class IntegerColumn extends Column {

  @SuppressWarnings("unused")
  private final Set optionSet;

  IntegerColumn(int index, int length, Set optionSet) {
    super(index, length);
    this.optionSet = optionSet;
  }

  public static IntegerColumn get(int length, IntegerOption... options) {
    int index = 0;
    return get(index, length, options);
  }

  public static IntegerColumn get(int index, int length, IntegerOption... options) {
    Set optionSet = ImmutableSet.copyOf(options);
    return new IntegerColumn(index, length, optionSet);
  }

  public static IntegerColumn get(int length, IntValueCondition condition, String text) {
    int index = 0;
    return get(index, length, condition, text);
  }

  public static IntegerColumn get(int index, int length, IntValueCondition condition, String text) {
    Set optionSet = ImmutableSet.of();
    return new Conditional(index, length, optionSet, condition, text);
  }

  @Override
  Token parse(Line line, String text) {
    try {
      int intValue = Integer.parseInt(text.trim());
      return new IntegerValue(this, text, intValue);
    } catch (NumberFormatException e) {
      return ColumnParseError.exception(this, line, text, e);
    }
  }

  private static class Conditional extends IntegerColumn {

    private final IntValueCondition condition;
    private final String text;

    public Conditional(int index, int length, Set optionSet, IntValueCondition condition, String text) {
      super(index, length, optionSet);
      this.condition = condition;
      this.text = text;
    }

    @Override
    Token parse(Line line, String text) {
      return Objects.equals(normal(text), normal(this.text))
          ? condition.token(this, text)
          : super.parse(line, text);
    }

    private String normal(String input) {
      return Strings.nullToEmpty(input).trim();
    }

  }

}