nablarch.test.core.reader.HeaderLine Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nablarch-testing Show documentation
There is a newer version: 2.2.0
package nablarch.test.core.reader;

import nablarch.test.core.util.ListWrapper;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

import static nablarch.test.NablarchTestUtils.*;

/** ヘッダ行を表すクラス。 */
class HeaderLine {

    /** キー一覧 */
    private final List keys;

    /** マーカーカラムの位置一覧 */
    private final Set markerIndices = new HashSet();

    /** 有効なカラム名一覧(マーカーカラムを除いたカラム名一覧) */
    private final List effectiveColumnNames;

    /**
     * コンストラクタ。
     *
     * @param headerLine ヘッダ行
     */
    HeaderLine(List headerLine) {
        List keys = trimTailCopy(headerLine);   // キャッシュを破壊しないようにコピーして編集
        if (keys == null) {
            this.keys = Collections.emptyList();
        } else {
            this.keys = keys;
        }
        ListWrapper wrapper = ListWrapper.wrap(this.keys);
        markerIndices.addAll(wrapper.indicesOf(MARKER_COLUMN_CONDITION));
        effectiveColumnNames = wrapper.exclude(MARKER_COLUMN_CONDITION);
    }

    /**
     * 有効なカラム名一覧を取得する。
     *
     * @return 有効なカラム名一覧
     */
    String[] getEffectiveColumnNames() {
        return effectiveColumnNames.toArray(new String[effectiveColumnNames.size()]);
    }

    /**
     * マーカーカラムを除外したマップを取得する。
     *
     * @param line 行データ
     * @return Map(キーはヘッダ行から取得される)
     */
    Map getMapExcludingMarkerColumns(List line) {
        // 目視による比較がしやすいのでTreeMapを使用（キーでソートされる）
        Map result = new TreeMap();
        List values = excludeMarkerColumns(line);
        for (int i = 0; i < values.size(); i++) {
            result.put(effectiveColumnNames.get(i), values.get(i));
        }
        return result;
    }

    /**
     * マーカーカラムを除外したリストを取得する。
     *
     * @param line 行データ
     * @return マーカーカラムに該当する要素が除外されたリスト
     */
    List excludeMarkerColumns(List line) {
        List notMarked = new ArrayList();
        for (int i = 0; i < keys.size(); i++) {
            if (markerIndices.contains(i)) {
                continue;  // マーカーカラムならスキップ
            }
            String val = (i >= line.size()) ? "" : line.get(i);
            notMarked.add(val);
        }
        return notMarked;
    }

    /** マーカーカラムの条件 */
    private static final ListWrapper.Condition MARKER_COLUMN_CONDITION
            = new ListWrapper.Condition() {
        /** {@inheritDoc} */
        public boolean evaluate(String element) {
            return element != null
                    && element.startsWith("[")
                    && element.endsWith("]");
        }
    };
}