XLSWriter 스타일링, 포맷팅 수정

src/main/java/cokr/xit/base/file/xls/XLS.java

@@ -4,6 +4,8 @@ import java.io.InputStream;
 import java.io.OutputStream;
 import java.text.SimpleDateFormat;
 import java.util.Date;
+import java.util.Map;
+import java.util.function.Function;
 
 import org.springframework.core.io.ClassPathResource;
 
@@ -64,4 +66,62 @@ public abstract class XLS extends AbstractComponent {
 	 * @param out OutputStream
 	 */
 	public abstract void write(OutputStream out);
+
+	/**yyyyMMdd, 또는 yyMMdd 포맷의 문자열을 yyyy-MM-dd 포맷으로 변환하여 반환한다.
+	 * @param obj yyyyMMdd, 또는 yyMMdd 포맷의 문자열
+	 * @return yyyy-MM-dd 포맷 문자열
+	 */
+	public final String str2date(Object obj) {
+		if (isEmpty(obj)) return "";
+
+		String str = obj.toString();
+		int length = str.length(),
+			pos = length - 2;
+
+		String
+			day = str.substring(pos),
+			month = str.substring(pos = pos - 2, pos + 2),
+			year = str.substring(0, pos);
+		return String.format("%s-%s-%s", year, month, day);
+	}
+
+	/**HHmmss 포맷의 문자열을 HH:mm:ss 포맷으로 변환하여 반환한다.
+	 * @param obj HHmmss 포맷의 문자열
+	 * @return HH:mm:ss 포맷 문자열
+	 */
+	public final String str2time(Object obj) {
+		if (isEmpty(obj)) return "";
+
+		String str = obj.toString();
+		int length = str.length(),
+			pos = length - 2;
+
+		String
+			ss = str.substring(pos),
+			mm = str.substring(pos = pos - 2, pos + 2),
+			hh = str.substring(0, pos);
+		return String.format("%s:%s:%s", hh, mm, ss);
+	}
+
+	/**yyyyMMddHHmmss, 또는 yyMMddHHmmss 포맷의 문자열을 yyyy-MM-dd HH:mm:ss 포맷으로 변환하여 반환한다.
+	 * @param obj yyyyMMddHHmmss, 또는 yyMMddHHmmss 포맷의 문자열
+	 * @return yyyy-MM-dd HH:mm:ss 포맷 문자열
+	 */
+	public final String str2datetime(Object obj) {
+		if (isEmpty(obj)) return "";
+
+		String str = obj.toString();
+		int length = str.length(),
+			pos = length - 6;
+
+		String
+			date = str.substring(0, pos),
+			time = str.substring(pos);
+
+		return str2date(date) + " " + str2time(time);
+	}
+
+	public final Function<Map<?, ?>, Object> wrap(Function<Map<?, ?>, Object> func) {
+		return func;
+	}
 }

src/main/java/cokr/xit/base/file/xls/XLSWriter.java

@@ -7,6 +7,7 @@ import java.util.Collections;
 import java.util.Date;
 import java.util.Map;
 import java.util.function.Consumer;
+import java.util.function.Function;
 
 import org.apache.poi.ss.usermodel.BorderStyle;
 import org.apache.poi.ss.usermodel.CellStyle;
@@ -38,11 +39,11 @@ import org.apache.poi.xssf.usermodel.XSSFWorkbook;
  * 위 코드는 dataset의 모든 필드값들을 설정한다.
  * dataset의 특정 필드의 값들을 원하는 순서대로 설정하려면 해당 필드들의 이름을 순서대로 나열한다.
  * <pre><code> xlsx.values(dataset, "RCPT_ID", "RCPT_DT", "PAYER_NM", "RCPT_AMT");</code></pre>
- * dataset의 특정 필드의 값들에 스타일을 적용하려면 {@link KeyStyle}을 표시한다.
+ * dataset의 특정 필드의 값들에 스타일을 적용하려면 {@link Formatter}을 표시한다.
  * <pre><code> CellStyle
  *     datetime = xlsx.yyyy_mm_dd_hh_mm_ss(),
  *     numeric =  xlsx.n_nn0();
- * xlsx.values(dataset, "RCPT_ID", xlsx.keyStyle("RCPT_DT", datetime), "PAYER_NM", xlsx.keyStyle("RCPT_AMT", numeric));</code></pre>
+ * xlsx.values(dataset, "RCPT_ID", xlsx.style("RCPT_DT", datetime), "PAYER_NM", xlsx.style("RCPT_AMT", numeric));</code></pre>
  * @author mjkhan
  */
 public class XLSWriter extends XLS {
@@ -76,7 +77,7 @@ public class XLSWriter extends XLS {
 		}
 	}
 
-	private SXSSFWorkbook getWorkbook() {
+	public SXSSFWorkbook workbook() {
 		if (workbook != null)
 			return workbook;
 
@@ -102,9 +103,9 @@ public class XLSWriter extends XLS {
 	 */
 	public XLSWriter worksheet(int index) {
 		try {
-			worksheet = getWorkbook().getSheetAt(index);
+			worksheet = workbook().getSheetAt(index);
 		} catch (Exception e) {
-			worksheet = getWorkbook().createSheet();
+			worksheet = workbook().createSheet();
 		}
 		worksheet.setRandomAccessWindowSize(rowAccessWindowSize);
 		return this;
@@ -117,7 +118,7 @@ public class XLSWriter extends XLS {
 	 */
 	public XLSWriter worksheet(String name) {
 		try {
-			SXSSFWorkbook wb = getWorkbook();
+			SXSSFWorkbook wb = workbook();
 			worksheet = ifEmpty(wb.getSheet(name), () -> wb.createSheet(name));
 			worksheet.setRandomAccessWindowSize(rowAccessWindowSize);
 			return this;
@@ -216,7 +217,7 @@ public class XLSWriter extends XLS {
 					setCellStyle(val);
 				} else if (val instanceof CellStyle) {
 					setCellStyle(val);
-				} else if (val instanceof KeyStyle) {
+				} else if (val instanceof Formatter) {
 					setCellStyle(val);
 				} else {
 					col(x).value(val);
@@ -236,10 +237,10 @@ public class XLSWriter extends XLS {
 			style = (CellStyle)obj;
 		} else if (obj instanceof Styler) {
 			styler = (Styler)obj;
-		} else if (obj instanceof KeyStyle) {
+		} else if (obj instanceof Formatter) {
-			KeyStyle keyStyle = (KeyStyle)obj;
+			Formatter formatter = (Formatter)obj;
-			styler = keyStyle.styler;
+			styler = formatter.styler;
-			style = keyStyle.style;
+			style = formatter.style;
 		}
 
 		if (styler != null) {
@@ -266,11 +267,23 @@ public class XLSWriter extends XLS {
 
 		ArrayList<Object> vals = new ArrayList<>();
 		for (Object obj: keysAndStyles) {
-			if (obj instanceof KeyStyle) {
+			if (obj instanceof Formatter) {
-				KeyStyle keyStyle = (KeyStyle)obj;
+				Formatter formatter = (Formatter)obj;
-				Object val = map.get(keyStyle.key);
+				Object val = null;
+				if (formatter.key != null) {
+					val = map.get(formatter.key);
+				} else if (formatter.convert != null){
+					val = formatter.convert.apply(map);
+				}
 				vals.add(val);
-				vals.add(keyStyle);
+
+				Styler styler = formatter.styler;
+				CellStyle style = formatter.style;
+				if (styler != null)
+					style = cellStyle(styler);
+				if (style != null)
+					vals.add(style);
+
 			} else {
 				vals.add(map.get(obj));
 			}
@@ -280,7 +293,7 @@ public class XLSWriter extends XLS {
 
 	/**지정한 map의 값들을 현재 작업셀부터 행으로 설정한다.<br />
 	 * 설정할 값과 순서를 정하려면 해당 값들의 키를 지정한다.<br />
-	 * 값이 설정된 셀에 스타일을 적용하려면 {@link KeyStyle}로 키와 스타일을 지정한다.<br />
+	 * 값이 설정된 셀에 스타일을 적용하려면 {@link Formatter}로 키와 스타일을 지정한다.<br />
 	 * 다음은 map의 모든 값들을 설정하는 예다.
 	 * <pre><code> DataObject rec = new DataObject().set("NO", 1).set("PAYER", "홍길동").set("RCPT_AMT", 10000).set("RCPT_DT", new Date());
 	 * XLSWriter xlsx = new XLSWriter()
@@ -297,7 +310,7 @@ public class XLSWriter extends XLS {
 	 * ...
 	 * xlsx.rowValues(rec, "PAYER", xlsx.keyStyle("RCPT_AMT", numeric), xlsx.keyStyle("RCPT_DT", datetime));</code></pre>
 	 * @param map 맵
-	 * @param keysAndStyles 셀에 설정할 값들의 키, 또는 {@link KeyStyle 키와 스타일}
+	 * @param keysAndStyles 셀에 설정할 값들의 키, 또는 {@link Formatter 키와 스타일}
 	 * @return 현재 XLSWriter
 	 */
 	public XLSWriter rowValues(Map<?, ?> map, Object... keysAndStyles) {
@@ -306,7 +319,7 @@ public class XLSWriter extends XLS {
 
 	/**지정한 데이터셋의 값들을 현재 작업셀부터 행과 열로 설정한다.
 	 * 설정할 값과 순서를 정하려면 해당 값들의 키를 지정한다.<br />
-	 * 값이 설정된 셀에 스타일을 적용하려면 {@link KeyStyle}로 키와 스타일을 지정한다.<br />
+	 * 값이 설정된 셀에 스타일을 적용하려면 {@link Formatter}로 키와 스타일을 지정한다.<br />
 	 * 다음은 데이터셋의 모든 값들을 설정하는 예다.
 	 * <pre><code> List&lt;DataObject> dataset = ...;
 	 * XLSWriter xlsx = new XLSWriter()
@@ -321,9 +334,9 @@ public class XLSWriter extends XLS {
 	 * <pre><code> CellStyle numeric = xlsx.n_nn0(); //{@link #cellStyle(Styler)} 참고
 	 * CellStyle datetime = xlsx.yyyy_mm_dd_hh_mm_ss();
 	 * ...
-	 * xlsx.rowValues(dataset, "PAYER", xlsx.keyStyle("RCPT_AMT", numeric), xlsx.keyStyle("RCPT_DT", datetime));</code></pre>
+	 * xlsx.rowValues(dataset, "PAYER", xlsx.style("RCPT_AMT", numeric), xlsx.style("RCPT_DT", datetime));</code></pre>
 	 * @param dataset 데이터셋
-	 * @param keysAndStyles 셀에 설정할 값들의 키, 또는 {@link KeyStyle 키와 스타일}
+	 * @param keysAndStyles 셀에 설정할 값들의 키, 또는 {@link Formatter 키와 스타일}
 	 * @return 현재 XLSWriter
 	 */
 	public XLSWriter values(Iterable<? extends Map<?, ?>> dataset, Object... keysAndStyles) {
@@ -339,25 +352,14 @@ public class XLSWriter extends XLS {
 		return this;
 	}
 
-	public XLSWriter styles(Styler... stylers) {
-		if (!isEmpty(stylers)) {
-			int start = cell.getColumnIndex(),
-				x = start;
-			for (Styler styler: stylers) {
-				col(x);
-				setCellStyle(styler);
-				++x;
-			}
-			col(start);
-		}
-		return this;
-	}
-
 	/**스타일 정보에서 셀스타일을 생성하여 반환한다.
 	 * @param styler 스타일 정보
 	 * @return 셀스타일
 	 */
 	public CellStyle cellStyle(Styler styler) {
+		if (styler == null)
+			return null;
+
 		CellStyle cellStyle = workbook.createCellStyle();
 		styler.set(cellStyle);
 		if (styler.dataFormat != null) {
@@ -371,17 +373,30 @@ public class XLSWriter extends XLS {
 	 * @param style 셀스타일
 	 * @return KeyStyle
 	 */
-	public KeyStyle keyStyle(Object key, CellStyle style) {
+	public Formatter formatter(Object key) {
-		return new KeyStyle(key, style);
+		return new Formatter().key(key);
 	}
 
-	/**style에서 셀스타일을 생성한 후 지정한 키와 연결한 KeyStyle을 반환한다.
+	/**데이터 포맷 function을 설정한다.
-	 * @param key 키
+	 * @param func 데이터 포맷 function
-	 * @param style 스타일 정보
+	 * @return Formatter
-	 * @return
+	 */
+	public Formatter format(Function<Map<?, ?>, Object> func) {
+		return formatter(null).format(func);
+	}
+
+	/**지정한 데이터 키의 스타일(CellStyle, 또는 Styler)를 설정한다.
+	 * @param key 데이터 키
+	 * @param style 스타일(CellStyle, 또는 Styler)
+	 * @return Formatter
 	 */
-	public KeyStyle keyStyle(Object key, Styler style) {
+	public Formatter style(Object key, Object style) {
-		return keyStyle(key, cellStyle(style));
+		Formatter fmt = formatter(key);
+		if (style instanceof CellStyle)
+			fmt.style((CellStyle)style);
+		else if (style instanceof Styler)
+			fmt.styler((Styler)style);
+		return fmt;
 	}
 
 	/**현재 XLSWriter의 내부상태를 초기화하여 반환한다.
@@ -424,6 +439,13 @@ public class XLSWriter extends XLS {
 		return cellStyle(new Styler().dataFormat("yyyy-MM-dd").merge(Styler.CENTER));
 	}
 
+	/**시간 포맷(HH:mm:ss)의 셀스타일을 반환한다.
+	 * @return 시간 포맷(HH:mm:ss)의 셀스타일
+	 */
+	public CellStyle hh_mm_ss() {
+		return cellStyle(new Styler().dataFormat("HH:mm:ss").merge(Styler.CENTER));
+	}
+
 	/**셀스타일 정보. {@link #set(CellStyle) CellStyle을 설정}하는데 사용.
 	 * @author mjkhan
 	 */
@@ -699,30 +721,51 @@ public class XLSWriter extends XLS {
 		}
 	}
 
-	/**셀스타일을 특정키와 연결하는 정보
+	/**데이터 포맷, 셀스타일을 특정키와 연결하는 정보
 	 * @author mjkhan
 	 */
-	public static class KeyStyle {
+	public static class Formatter {
 		private Object key;
+
+		private Function<Map<?, ?>, Object> convert;
+
 		private CellStyle style;
 		private Styler styler;
 
-		/**새 KeyStyle을 생성한다.
+		/**데이터 키를 설정한다.
-		 * @param key 키
+		 * @param key 데이터 키
+		 * @return 현재 Formatter
+		 */
+		public Formatter key(Object key) {
+			this.key = key;
+			return this;
+		}
+
+		/**셀스타일을 설정한다.
 		 * @param style 셀스타일
+		 * @return 현재 Formatter
 		 */
-		KeyStyle(Object key, CellStyle style) {
+		public Formatter style(CellStyle style) {
-			this.key = notEmpty(key, "key");
 			this.style = style;
+			return this;
 		}
 
-		/**새 KeyStyle을 생성한다.
+		/**스타일러를 설정한다.
-		 * @param key 키
 		 * @param styler 스타일러
+		 * @return 현재 Formatter
 		 */
-		KeyStyle(Object key, Styler styler) {
+		public Formatter styler(Styler styler) {
-			this.key = notEmpty(key, "key");
 			this.styler = styler;
+			return this;
+		}
+
+		/**데이터 포맷 function을 설정한다.
+		 * @param func 데이터 포맷 설정
+		 * @return 현재 Formatter
+		 */
+		public Formatter format(Function<Map<?, ?>, Object> func) {
+			convert = func;
+			return this;
 		}
 	}
 }