View Javadoc
1   /*
2    * Copyright 2013 the original author or authors.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  
17  package com.orangesignal.csv.handlers;
18  
19  import java.io.IOException;
20  import java.util.ArrayList;
21  import java.util.Collection;
22  import java.util.List;
23  import java.util.Map;
24  
25  import com.orangesignal.csv.CsvReader;
26  import com.orangesignal.csv.CsvWriter;
27  import com.orangesignal.csv.filters.CsvNamedValueFilter;
28  import com.orangesignal.csv.io.CsvColumnNameMapReader;
29  import com.orangesignal.csv.io.CsvColumnNameMapWriter;
30  
31  /**
32   * 項目名と項目値のマップのリストで区切り文字形式データアクセスを行うハンドラを提供します。
33   *
34   * @author Koji Sugisawa
35   * @since 1.1
36   */
37  public class ColumnNameMapListHandler extends AbstractCsvListHandler<Map<String, String>, ColumnNameMapListHandler> {
38  
39  	/**
40  	 * 項目名のリストを保持します。
41  	 */
42  	private List<String> columnNames;
43  
44  	/**
45  	 * 区切り文字形式データの列見出し (ヘッダ) 行を出力するかどうかを保持します。
46  	 * 
47  	 * @since 2.1
48  	 */
49  	private boolean header = true;
50  
51  	/**
52  	 * 区切り文字形式データフィルタを保持します。
53  	 */
54  	private CsvNamedValueFilter valueFilter;
55  
56  	/**
57  	 * デフォルトコンストラクタです。
58  	 */
59  	public ColumnNameMapListHandler() {}
60  
61  	/**
62  	 * 指定された項目名を項目名の一覧へ追加します。
63  	 * 
64  	 * @param columnName 項目名
65  	 * @return このオブジェクトへの参照
66  	 */
67  	public ColumnNameMapListHandler addColumn(final String columnName) {
68  		if (columnNames == null) {
69  			columnNames = new ArrayList<String>();
70  		}
71  		columnNames.add(columnName);
72  		return this;
73  	}
74  
75  	/**
76  	 * 指定された項目名の一覧を設定します。
77  	 * 
78  	 * @param columnNames 項目名の一覧
79  	 * @return このオブジェクトへの参照
80  	 */
81  	public ColumnNameMapListHandler columnNames(final Collection<String> columnNames) {
82  		this.columnNames = new ArrayList<String>(columnNames);
83  		return this;
84  	}
85  
86  	/**
87  	 * 区切り文字形式データフィルタを設定します。
88  	 * 
89  	 * @param filter 区切り文字形式データフィルタ
90  	 * @return このオブジェクトへの参照
91  	 * @since 1.2.3
92  	 */
93  	public ColumnNameMapListHandler filter(final CsvNamedValueFilter filter) {
94  		this.valueFilter = filter;
95  		return this;
96  	}
97  
98  	/**
99  	 * 区切り文字形式データの列見出し (ヘッダ) 行を出力するかどうかを設定します。
100 	 * 
101 	 * @param header 区切り文字形式データの列見出し (ヘッダ) 行を出力するかどうか
102 	 * @return このオブジェクトへの参照
103 	 * @since 2.1
104 	 */
105 	public ColumnNameMapListHandler header(final boolean header) {
106 		this.header = header;
107 		return this;
108 	}
109 
110 	@Override
111 	public List<Map<String, String>> load(final CsvReader reader, final boolean ignoreScalar) throws IOException {
112 		@SuppressWarnings("resource")
113 		final CsvColumnNameMapReader r = new CsvColumnNameMapReader(reader, columnNames);
114 		r.setFilter(valueFilter);
115 
116 		// データ部を処理します。
117 		final List<Map<String, String>> results = new ArrayList<Map<String, String>>();
118 		int offset = 0;
119 
120 		List<String> values;
121 		while ((values = r.readValues()) != null && (ignoreScalar || limit <= 0 || results.size() < limit)) {
122 			if (!ignoreScalar && offset < this.offset) {
123 				offset++;
124 				continue;
125 			}
126 			results.add(r.toMap(values));
127 		}
128 		return results;
129 	}
130 
131 	@Override
132 	public void save(final List<Map<String, String>> list, final CsvWriter writer) throws IOException {
133 		@SuppressWarnings("resource")
134 		final CsvColumnNameMapWriter w = new CsvColumnNameMapWriter(writer, columnNames, header);
135 		w.setFilter(valueFilter);
136 
137 		// データ部を処理します。
138 		for (final Map<String, String> map : list) {
139 			w.write(map);
140 		}
141 	}
142 
143 }