利用Freemarker实现表到功能界面的一键生成



利用Freemarker实现表到功能界面的一键生成

2012-04-20 17:50:03

标签：执行文件 程序编写 表 工程 开发

原创作品，允许转载，转载时请务必以超链接形式标明文章 原始出处 、作者信息和本声明。否则将追究法律责任。/article/4461317.html

利用Freemarker产生项目所需的常用文件，可以根据一个表产生对应的ibtais的java及xml文件，及前台的JSP文件，这样可以大大的提高普通程序编写人员的开发效率，及从表到界面及CURD功能利用这个工程可以直接生成好，下面提供下最核心的文件产生类，代码如下：

另外提供编译过的工程文件，可执行文件，在附件中，由于大小控制，已经删除里面的freemarker-2.3.15.jar文件。

01.package com.hundsun.pms.generator.util;

02.

03.import com.hundsun.pms.generator.api.Constants;

04.import com.hundsun.pms.generator.model.page.Page;

05.import freemarker.template.Configuration;

06.import freemarker.template.DefaultObjectWrapper;

07.import freemarker.template.Template;

08.import freemarker.template.TemplateException;

09.import org.apache.commons.lang.StringUtils;

10.import org.springframework.util.ResourceUtils;

11.

12.import java.io.*;

13.import java.net.URL;

14.import java.util.Locale;

15.import java.util.Map;

16.

17./**

18. * Created by IntelliJ IDEA.

19. * User: pangj

20. * Date: 2012-3-6

21. * Time: 10:05:26

22. * 文件操作工具类

23. */

24.public class FreemarkerUtil {

25.

26. public static String PROJECT_ROOT = "projectRoot";

27.

28. public static String PACKAGE_NAME = "packageName";

29.

30. public static String CLASS_SIMPLE_NAME = "classSimpleName";

31.

32. public static String MODULES_PACKAGE = "modulesPackage";

33.

34. public static String METHOD_NAME = "methodName";

35.

36. //模板根目录

37. private static final String templateRootWithClasspath = "classpath*:com/hundsun/pms/generator/resource/template";

38.

39. private static final String templateRoot = "com/hundsun/pms/generator/resource/template";

40. //工程根目录

41. private static String projectRoot;

42. //包名

43. private static String packageName;

44. //服务所在的包路径

45. private static String modulesPackage;

46.

47. private static String classSimpleName;

48.

49. private static String templateName;

50.

51. private static String methodName;

52.

53. public static final String FILE_BO_NAME_SUFFIX = "Bo.java";

54.

55. public static final String FILE_BOIMPL_NAME_SUFFIX = "BoImpl.java";

56.

57. public static final String FILE_DAO_NAME_SUFFIX = "Dao.java";

58.

59. public static final String FILE_DAOIMPL_NAME_SUFFIX = "DaoImpl.java";

60.

61. public static final String FILE_DTO_NAME_SUFFIX = ".java";

62.

63. public static final String FILE_SQL_MAP_SUFFIX = ".xml";

64.

65. public static final String FILE_SERVICE_SUFFIX = ".service";

66.

67. public static final String FILE_COMPONENT_SUFFIX = ".component";

68.

69. public static final String FILE_MODULE_NAME = "module.xml";

70.

71. public static final String FILE_PAGE_SUFFIX = ".jsp";

72.

73.

74. /**

75. * 数据绑定至模板

76. *

77. * @param tpName 模板名

78. * @param param 参数

79. * @return

80. */

81. public static void process(String tpName, Map<String, Object> param) {

82. init(tpName, param);

83. Writer out = null;

84. try {

85. Template temp = getTemplate(templateName);

86. String fileDir = getFileDir();

87. File dir = new File(fileDir);

88. if (!dir.exists()) {

89. if (!dir.mkdirs()) {

90. MsgUtil.print("创建目录：" + fileDir + "失败");

91. return;

92. }

93. }

94. File f = new File(fileDir + getFileName());

95. out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(f), "utf-8"));

96. try {

97. temp.setEncoding("utf-8");

98. temp.process(param, out);

99. MsgUtil.print("文件：" + getFileDir() + getFileName() + "生成完成！");

100. } catch (TemplateException e) {

101. MsgUtil.print("文件：" + getFileDir() + getFileName() + "生成失败，请检查原因！");

102. MsgUtil.print(e.getMessage());

103. }

104. out.flush();

105. out.close();

106. } catch (IOException e) {

107. e.printStackTrace();

108. MsgUtil.print(e.getMessage());

109. }

110. }

111.

112. /**

113. * 生成页面

114. *

115. * @param tpName 模板名

116. * @param jsp jsp页面全路径

117. * @param page 页面对象

118. */

119. public static void process(String tpName, String jsp, Page page) {

120. Writer writer = null;

121. try {

122. Template tmp = FreemarkerUtil.getTemplate(tpName);

123. String dir = jsp.substring(0, jsp.lastIndexOf("\\"));

124. File fdir = new File(dir);

125. if (!fdir.exists()) {

126. if (!fdir.mkdirs()) {

127. MsgUtil.print("创建目录：" + fdir + "失败");

128. return;

129. }

130. }

131. File file = new File(jsp);

132. writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));

133. tmp.setEncoding("utf-8");

134. tmp.process(page, writer);

135. writer.flush();

136. writer.close();

137. MsgUtil.print("页面：" + jsp + " 生成成功!");

138. } catch (TemplateException e) {

139. e.printStackTrace();

140. MsgUtil.print(e.getMessage());

141. } catch (IOException e) {

142. e.printStackTrace();

143. MsgUtil.print(e.getMessage());

144. }

145. }

146.

147. public static Template getTemplate(String templateName) {

148. Configuration cfg = new Configuration();

149. Template temp = null;

150. URL path = FreemarkerUtil.class.getClassLoader().getSystemResource("");

151. File tmpRootFile = null;

152. if (path == null) {

153. tmpRootFile = getResourceURL("resource/template");

154. } else {

155. String tmpRoot = path + templateRoot;

156. tmpRoot = tmpRoot.replaceAll("file:/", "");

157. tmpRootFile = new File(tmpRoot);

158. }

159. if (tmpRootFile == null) {

160. throw new RuntimeException("无法取得模板根路径！");

161. }

162. try {

163. cfg.setDefaultEncoding("utf-8");

164. cfg.setOutputEncoding("utf-8");

165. cfg.setDirectoryForTemplateLoading(tmpRootFile);

166. /*cfg.setDirectoryForTemplateLoading(getResourceURL());*/

167. cfg.setObjectWrapper(new DefaultObjectWrapper());

168. temp = cfg.getTemplate(templateName);

169. } catch (IOException e) {

170. e.printStackTrace();

171. MsgUtil.print(e.getMessage());

172. }

173. return temp;

174. }

175.

176. public static File getResourceURL(String templatePath) {

177. try {

178. URL url = ResourceUtils.getURL(templateRootWithClasspath);

179. String path = url.getPath();

180. path = path.replace(templateRootWithClasspath, templatePath);

181. MsgUtil.print("File path is:"+path);

182. path = StringUtils.replace(path, "%20", " ");

183. return new File(path);

184. } catch (FileNotFoundException e) {

185. throw new RuntimeException("不能获取路径：" + templateRootWithClasspath);

186. }

187. }

188.

189. private static void init(String tpName, Map<String, Object> param) {

190. projectRoot = (String) param.get(PROJECT_ROOT);

191. packageName = (String) param.get(PACKAGE_NAME);

192. classSimpleName = (String) param.get(CLASS_SIMPLE_NAME);

193. modulesPackage = (String) param.get(MODULES_PACKAGE);

194. methodName = (String) param.get(METHOD_NAME);

195. templateName = tpName;

196. }

197.

198. private static String getFileName() {

199. if (Constants.TEMPLATE_BO.equals(templateName)) {

200. return classSimpleName + FILE_BO_NAME_SUFFIX;

201. } else if (Constants.TEMPLATE_BOIMPL.equals(templateName)) {

202. return classSimpleName + FILE_BOIMPL_NAME_SUFFIX;

203. } else if (Constants.TEMPLATE_DAO.equals(templateName)) {

204. return classSimpleName + FILE_DAO_NAME_SUFFIX;

205. } else if (Constants.TEMPLATE_DAOIMPL.equals(templateName)) {

206. return classSimpleName + FILE_DAOIMPL_NAME_SUFFIX;

207. } else if (Constants.TEMPLATE_DTO.equals(templateName)) {

208. return classSimpleName + FILE_DTO_NAME_SUFFIX;

209. } else if (Constants.TEMPLATE_SQLMAP.equals(templateName)) {

210. return classSimpleName + FILE_SQL_MAP_SUFFIX;

211. } else if (Constants.TEMPLATE_SERVICE.equals(templateName)) {

212. return methodName + FILE_SERVICE_SUFFIX;

213. } else if (Constants.TEMPLATE_COMPONENT.equals(templateName)) {

214. return NamingRuleConvert.firstLetterToLowerCase(classSimpleName) + FILE_COMPONENT_SUFFIX;

215. } else if (Constants.TEMPLATE_CS.equals(templateName) || Constants.TEMPLATE_DS.equals(templateName) ||

216. Constants.TEMPLATE_MODULE.equals(templateName)) {

217. return FILE_MODULE_NAME;

218. }

219. return null;

220. }

221.

222. private static String getFileDir() {

223. String path = null;

224. if (packageName != null)

225. path = packageName.replaceAll("\\.", "/");

226. String servicePath = null;

227. if (!StringUtils.isEmpty(modulesPackage)) {

228. servicePath = modulesPackage.replaceAll("\\.", "/");

229. }

230. if (Constants.TEMPLATE_BO.equals(templateName)) {

231. return projectRoot + Constants.SRC_DIR + path + "/bo/";

232. } else if (Constants.TEMPLATE_BOIMPL.equals(templateName)) {

233. return projectRoot + Constants.SRC_DIR + path + "/bo/impl/";

234. } else if (Constants.TEMPLATE_DAO.equals(templateName)) {

235. return projectRoot + Constants.SRC_DIR + path + "/dao/";

236. } else if (Constants.TEMPLATE_DAOIMPL.equals(templateName)) {

237. return projectRoot + Constants.SRC_DIR + path + "/dao/impl/";

238. } else if (Constants.TEMPLATE_DTO.equals(templateName)) {

239. return projectRoot + Constants.SRC_DIR + path + "/dto/";

240. } else if (Constants.TEMPLATE_SQLMAP.equals(templateName)) {

241. return projectRoot + Constants.SRC_DIR + path + "/dao/maps/";

242. } else if (Constants.TEMPLATE_SERVICE.equals(templateName) || (Constants.TEMPLATE_CS.equals(templateName))) {

243. return projectRoot + "/modules/" + servicePath + "/cs/";

244. } else if (Constants.TEMPLATE_COMPONENT.equals(templateName) || Constants.TEMPLATE_DS.equals(templateName)) {

245. return projectRoot + "/modules/" + servicePath + "/ds/";

246. } else if (Constants.TEMPLATE_MODULE.equals(templateName)) {

247. return projectRoot + "/modules/" + servicePath + "/";

248. }

249. return null;

250. }

251.}