Java生成流水号 -2 支持数据库查询，Spring注入

Java代码  

1. package com.wesley.framework.dao;  
2.   
3. import java.text.DecimalFormat;  
4. import java.text.SimpleDateFormat;  
5. import java.util.Date;  
6. import java.util.HashMap;  
7. import java.util.Map;  
8.   
9. import org.apache.commons.dbcp.BasicDataSource;  
10. import org.springframework.beans.factory.xml.XmlBeanFactory;  
11. import org.springframework.core.io.ClassPathResource;  
12.   
13. import com.wesley.framework.dao.dbutils.DBHelper;  
14.   
15. public class PrimaryGenerater {  
16.   
17.     private static PrimaryGenerater primaryGenerater = null;  
18.     private DBHelper<PrimaryBean> dbHelper;  
19.     private Map<String, String> sqls = new HashMap<String, String>();  
20.     private Map<String, PrimaryBean> primarBeans = new HashMap<String, PrimaryBean>();  
21.   
22.     private PrimaryGenerater() {  
23.         super();  
24.     }  
25.   
26.     public PrimaryGenerater(BasicDataSource database, Map<String, String> sqls) {  
27.         super();  
28.         this.dbHelper = new DBHelper<PrimaryBean>(database, PrimaryBean.class);  
29.         this.sqls = sqls;  
30.         for (String key : sqls.keySet()) {  
31.             this.primarBeans.put(key, this.getPrimaryBeanByDatabase(key));  
32.         }  
33.     }  
34.   
35.     public static PrimaryGenerater newInstance(BasicDataSource database,  
36.             Map<String, String> sqls) {  
37.         synchronized (PrimaryGenerater.class) {  
38.             primaryGenerater = new PrimaryGenerater(database, sqls);  
39.         }  
40.         return primaryGenerater;  
41.     }  
42.   
43.     /** 
44.      *  
45.      * 取得PrimaryGenerater的单例实现 
46.      *  
47.      * @return 
48.      */  
49.     public static PrimaryGenerater getInstance() {  
50.         if (primaryGenerater == null) {  
51.             synchronized (PrimaryGenerater.class) {  
52.                 if (primaryGenerater == null) {  
53.                     primaryGenerater = new PrimaryGenerater();  
54.                 }  
55.             }  
56.         }  
57.         return primaryGenerater;  
58.     }  
59.   
60.     /** 
61.      * 通过 数据库查询键获得封装类 
62.      *  
63.      * @param key 
64.      * @return 
65.      */  
66.     public synchronized PrimaryBean getPrimaryBeanByDatabase(String key) {  
67.         if (!this.sqls.containsKey(key))  
68.             return null;  
69.         PrimaryBean primaryBean = this.primarBeans.get(key);  
70.         if (primaryBean != null && !primaryBean.getIsDataBase())  
71.             return primaryBean;  
72.         primaryBean = dbHelper.query(this.sqls.get(key));  
73.         return primaryBean;  
74.     }  
75.   
76.     /** 
77.      * 通过数据库查询键位生成流水号 
78.      *  
79.      * @param key 
80.      * @return 
81.      */  
82.     public synchronized String geneterNextNumberByKey(String key) {  
83.         PrimaryBean primaryBean = this.getPrimaryBeanByDatabase(key);  
84.         return this.geneterNextNumber(primaryBean);  
85.     }  
86.   
87.     /** 
88.      * 通过封装类生成流水号 
89.      *  
90.      * @param primaryBean 
91.      * @return 
92.      */  
93.     public synchronized String geneterNextNumber(PrimaryBean primaryBean) {  
94.         String nextNumber = this  
95.                 .geneterNextNumber(primaryBean.getStartNumber());  
96.         primaryBean.setStartNumber(nextNumber);  
97.         String dataString = this.geneterDataString(primaryBean.getDataString());  
98.         String serialNumber = primaryBean.getPrefix() + dataString + nextNumber;  
99.         return serialNumber;  
100.     }  
101.   
102.     /** 
103.      * 通过开始数字字符串生成下一个流水号 
104.      *  
105.      * @param startNumber 
106.      * @return 
107.      */  
108.     public synchronized String geneterNextNumber(String startNumber) {  
109.         Long temp = Long.valueOf(startNumber) + 1;  
110.         StringBuilder sb = new StringBuilder();  
111.         for (int i = 0; i < startNumber.length(); i++)  
112.             sb.append("0");  
113.         DecimalFormat df = new DecimalFormat(sb.toString());  
114.         return df.format(temp);  
115.     }  
116.   
117.     /** 
118.      * 通过 格式生成日期格式 
119.      *  
120.      * @param dataformat 
121.      * @return 
122.      */  
123.     private synchronized String geneterDataString(String dataformat) {  
124.         SimpleDateFormat formatter = new SimpleDateFormat(dataformat);  
125.         return formatter.format(new Date());  
126.     }  
127.   
128.     public static void main(String[] args) throws Exception {  
129.         XmlBeanFactory context = new XmlBeanFactory(new ClassPathResource(  
130.                 "applicationContext.xml"));  
131.         BasicDataSource dataSource = (BasicDataSource) context  
132.                 .getBean("dataSource");  
133.         Map<String, String> map = new HashMap<String, String>();  
134.         map.put("new",  
135.                 "select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' as startNumber from business union select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' as startNumber");  
136.         PrimaryGenerater.newInstance(dataSource, map);  
137.         for (int i = 0; i < 100; i++) {  
138.             System.out.println(PrimaryGenerater.getInstance()  
139.                     .geneterNextNumberByKey("new"));  
140.         }  
141.     }  
142.   
143. }  

Java代码  

1. package com.wesley.framework.dao;  
2.   
3. public class PrimaryBean {  
4.     // select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' startNumber  
5.     private String prefix;  
6.     private String dataString;  
7.     private String startNumber;  
8.     private Boolean isDataBase = false;  
9.   
10.     public String getPrefix() {  
11.         return prefix;  
12.     }  
13.   
14.     public void setPrefix(String prefix) {  
15.         this.prefix = prefix;  
16.     }  
17.   
18.     public String getDataString() {  
19.         return dataString;  
20.     }  
21.   
22.     public void setDataString(String dataString) {  
23.         this.dataString = dataString;  
24.     }  
25.   
26.     public String getStartNumber() {  
27.         return startNumber;  
28.     }  
29.   
30.     public void setStartNumber(String startNumber) {  
31.         this.startNumber = startNumber;  
32.     }  
33.   
34.     public Boolean getIsDataBase() {  
35.         return isDataBase;  
36.     }  
37.   
38.     public void setIsDataBase(Boolean isDataBase) {  
39.         this.isDataBase = isDataBase;  
40.     }  
41.   
42. }  

Java代码  

1. package com.wesley.framework.dao.dbutils;  
2.   
3. import java.sql.Connection;  
4. import java.sql.SQLException;  
5. import java.util.List;  
6.   
7. import org.apache.commons.dbcp.BasicDataSource;  
8. import org.apache.commons.dbutils.BasicRowProcessor;  
9. import org.apache.commons.dbutils.QueryRunner;  
10. import org.apache.commons.dbutils.ResultSetHandler;  
11. import org.apache.commons.dbutils.handlers.BeanHandler;  
12. import org.apache.commons.dbutils.handlers.BeanListHandler;  
13. import org.slf4j.Logger;  
14. import org.slf4j.LoggerFactory;  
15. import org.unitils.thirdparty.org.apache.commons.dbutils.DbUtils;  
16.   
17. public class DBHelper<T> {  
18.     private static final Logger logger = LoggerFactory  
19.             .getLogger(DBHelper.class);  
20.   
21.     private BasicDataSource pool;  
22.   
23.     private Class<T> beanClass = null;  
24.   
25.     public DBHelper(BasicDataSource pool) {  
26.         super();  
27.         this.pool = pool;  
28.     }  
29.   
30.     public DBHelper(BasicDataSource pool, Class<T> beanClass) {  
31.         super();  
32.         this.pool = pool;  
33.         this.beanClass = beanClass;  
34.     }  
35.   
36.     /** 
37.      * 根据无参sql语句进行查询，并返回一个对象，用于单条记录的查询 
38.      *  
39.      * @param sql 
40.      * @return Object 
41.      */  
42.     public T query(String sql) {  
43.         Connection conn = this.getPoolConnection();  
44.         if (conn == null || beanClass == null)  
45.             return null;  
46.         QueryRunner run = new QueryRunner();  
47.         ResultSetHandler<T> h = new BeanHandler<T>(beanClass);  
48.         T result = null;  
49.         try {  
50.             result =   run.query(conn, sql, h);  
51.         } catch (SQLException e) {  
52.             logger.info("SQLException in [" + sql + "]");  
53.             logger.error(e.getMessage());  
54.         } finally {  
55.             DbUtils.closeQuietly(conn);  
56.         }  
57.         return result;  
58.     }  
59.   
60.     /** 
61.      * 根据有参sql语句进行查询，并返回一个对象，用于单条记录的查询 
62.      *  
63.      * @param sql 
64.      * @param args 
65.      * @return Object 
66.      */  
67.   
68.     public T query(String sql, Object... args) {  
69.         Connection conn = this.getPoolConnection();  
70.         if (conn == null || beanClass == null)  
71.             return null;  
72.         QueryRunner run = new QueryRunner();  
73.         ResultSetHandler<T> h = new BeanHandler<T>(beanClass);  
74.         T result = null;  
75.         try {  
76.             result = (T) run.query(conn, sql, h, args);  
77.         } catch (SQLException e) {  
78.             logger.info("SQLException in [" + sql + "]");  
79.             logger.error(e.getMessage());  
80.         } finally {  
81.             DbUtils.closeQuietly(conn);  
82.         }  
83.         return result;  
84.     }  
85.   
86.     /** 
87.      * 根据无参sql语句进行查询，并返回一个数据集，用于多条记录的查询 
88.      *  
89.      * @param sql 
90.      * @return ArrayList 
91.      */  
92.     public List<T> queryRecords(String sql) {  
93.         Connection conn = this.getPoolConnection();  
94.         if (conn == null || beanClass == null)  
95.             return null;  
96.         // ResultSetHandler<?> h = new BeanListHandler(beanClass);  
97.         ResultSetHandler<List<T>> h = new BeanListHandler<T>(beanClass,  
98.                 new BasicRowProcessor(new StrategyBeanProcessor(  
99.                         new HumpMatcher())));  
100.         QueryRunner run = new QueryRunner();  
101.         List<T> result = null;  
102.         try {  
103.             result = run.query(conn, sql, h);  
104.         } catch (SQLException e) {  
105.             logger.info("SQLException in [" + sql + "]");  
106.             logger.error(e.getMessage());  
107.         } finally {  
108.             DbUtils.closeQuietly(conn);  
109.         }  
110.         return result;  
111.     }  
112.   
113.     /** 
114.      * 根据有参sql语句进行查询，并返回一个数据集，用于多条记录的查询 
115.      *  
116.      * @param sql 
117.      * @param args 
118.      * @return 
119.      */  
120.     public List<T> queryRecords(String sql, Object... args) {  
121.         Connection conn = this.getPoolConnection();  
122.         if (conn == null || beanClass == null)  
123.             return null;  
124.         QueryRunner run = new QueryRunner();  
125.         ResultSetHandler<List<T>> h = new BeanListHandler<T>(beanClass,  
126.                 new BasicRowProcessor(new StrategyBeanProcessor(  
127.                         new HumpMatcher())));  
128.         List<T> result = null;  
129.   
130.         try {  
131.             result = run.query(conn, sql, h, args);  
132.             System.out.println(result.size());  
133.         } catch (SQLException e) {  
134.             logger.info("SQLException in [" + sql + "]");  
135.             logger.error(e.getMessage());  
136.         } finally {  
137.             DbUtils.closeQuietly(conn);  
138.         }  
139.         return result;  
140.     }  
141.   
142.     /** 
143.      * 根据无参sql语句进行数据更新，并返回更新后影响的记录数 
144.      *  
145.      * @param sql 
146.      * @return int 
147.      */  
148.     public int update(String sql) {  
149.         Connection conn = this.getPoolConnection();  
150.         if (conn == null)  
151.             return 0;  
152.         QueryRunner run = new QueryRunner();  
153.         int result = 0;  
154.   
155.         try {  
156.             result = run.update(conn, sql);  
157.         } catch (SQLException e) {  
158.             logger.info("SQLException in [" + sql + "]");  
159.             logger.error(e.getMessage());  
160.         } finally {  
161.             DbUtils.closeQuietly(conn);  
162.         }  
163.         return result;  
164.     }  
165.   
166.     /** 
167.      * 根据有参sql语句进行数据更新，并返回更新后影响的记录数 
168.      *  
169.      * @param sql 
170.      * @param args 
171.      * @return int 
172.      */  
173.     public int update(String sql, Object... args) {  
174.         Connection conn = this.getPoolConnection();  
175.         if (conn == null)  
176.             return -1;  
177.   
178.         QueryRunner run = new QueryRunner();  
179.         int result = 0;  
180.   
181.         try {  
182.             result = run.update(conn, sql, args);  
183.         } catch (SQLException e) {  
184.             logger.info("SQLException in [" + sql + "]");  
185.             logger.error(e.getMessage());  
186.         } finally {  
187.             DbUtils.closeQuietly(conn);  
188.         }  
189.         return result;  
190.     }  
191.   
192.     /** 
193.      * 设置存放查询结果的Bean类，每个Bean类对象对应一条查询的结果记录 
194.      *  
195.      * @param beanClass 
196.      *            ，如User.class 
197.      */  
198.     public void setBeanClass(Class<T> beanClass) {  
199.         this.beanClass = beanClass;  
200.     }  
201.   
202.     public Connection getPoolConnection() {  
203.         int x = 3;  
204.         while (x > 0) {  
205.             try {  
206.                 return pool.getConnection();  
207.             } catch (SQLException e) {  
208.                 x--;  
209.                 logger.error(e.getMessage());  
210.             }  
211.         }  
212.         return null;  
213.     }  
214.   
215.     public void dispose() {  
216.         try {  
217.             this.pool.close();  
218.         } catch (SQLException e) {  
219.             logger.error(e.getMessage());  
220.         }  
221.     }  
222. }  

Java代码  

1. <bean id="PrimaryGenerater" class="com.wesley.framework.dao.PrimaryGenerater"  
2.         factory-method="newInstance">  
3.         <constructor-arg index="0" ref="dataSource" />  
4.         <constructor-arg index="1" >  
5.             <map>  
6.                 <entry key="bsno">  
7.                     <value>select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' as  
8.                         startNumber union select null as prefix,null as dataString,null as  
9.                         startNumber from business  
10.                     </value>  
11.                 </entry>  
12.             </map>  
13.         </constructor-arg>  
14.     </bean>