codetemplete.java

生成与Oracle相关的Ibatis相关配置文件及Java源码

/*
 * Copyright (c) 2008 胜利油田胜利软件有限责任公司. All rights reserved.
 */
package com.victorysoft.code.base;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;

/**
 * 
 * @author 于景洋
 * @newtime Oct 20, 2008,7:45:37 AM
 * @version 1.0
 * @see
 * @since JDK 1.5.0
 */
public class CodeTemplete {
	private Map<String, String> templeteMap = null;

	@SuppressWarnings("unchecked")
	public CodeTemplete(Class clazz, String tmplete) {
		this.templeteMap = new HashMap<String, String>();
		InputStream in = clazz.getResourceAsStream(tmplete);
		InputStreamReader ir = new InputStreamReader(in);
		BufferedReader br = new BufferedReader(ir);
		String line = "", handle = "", handle_val = "";
		try {
			while ((line = br.readLine()) != null) {
				if (line.trim().startsWith("///")) {
					continue;
				}
				if (line.startsWith("$")) {
					if (handle.length() > 0) {
						templeteMap.put(handle, handle_val);
					}
					if ("BEEND".equals(handle)) {
						break;
					}
					handle = line.trim();
					handle_val = "";
				} else {
					int pos = line.indexOf("///");
					if (pos != -1) {
						handle_val += line.substring(0, pos) + "\n";
					} else {
						handle_val += line + "\n";
					}
				}
			}
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			try {
				if (in != null) {
					in.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}

	}

	/**
	 * 根据关键字获取模板
	 * 
	 * @param keyword
	 *            关键字
	 * @return
	 */
	public String getTemplete(String keyword) {
		return this.templeteMap.get(keyword);
	}
}