JAVA流水号生成, 支持数据库查询,Spring注入

package com.wesley.framework.dao;

import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.dbcp.BasicDataSource;
import org.springframework.beans.factory.xml.XmlBeanFactory;
import org.springframework.core.io.ClassPathResource;

import com.wesley.framework.dao.dbutils.DBHelper;

public class PrimaryGenerater {

private static PrimaryGenerater primaryGenerater = null;
private DBHelper dbHelper;
private Map sqls = new HashMap();
private Map primarBeans = new HashMap();

private PrimaryGenerater() {
super();
}

public PrimaryGenerater(BasicDataSource database, Map sqls) {
super();
this.dbHelper = new DBHelper(database, PrimaryBean.class);
this.sqls = sqls;
for (String key : sqls.keySet()) {
this.primarBeans.put(key, this.getPrimaryBeanByDatabase(key));
}
}

public static PrimaryGenerater newInstance(BasicDataSource database,
Map sqls) {
synchronized (PrimaryGenerater.class) {
primaryGenerater = new PrimaryGenerater(database, sqls);
}
return primaryGenerater;
}

/**
*
* 取得PrimaryGenerater的单例实现
*
* @return
*/
public static PrimaryGenerater getInstance() {
if (primaryGenerater == null) {
synchronized (PrimaryGenerater.class) {
if (primaryGenerater == null) {
primaryGenerater = new PrimaryGenerater();
}
}
}
return primaryGenerater;
}

/**
* 通过 数据库查询键获得封装类
*
* @param key
* @return
*/
public synchronized PrimaryBean getPrimaryBeanByDatabase(String key) {
if (!this.sqls.containsKey(key))
return null;
PrimaryBean primaryBean = this.primarBeans.get(key);
if (primaryBean != null && !primaryBean.getIsDataBase())
return primaryBean;
primaryBean = dbHelper.query(this.sqls.get(key));
return primaryBean;
}

/**
* 通过数据库查询键位生成流水号
*
* @param key
* @return
*/
public synchronized String geneterNextNumberByKey(String key) {
PrimaryBean primaryBean = this.getPrimaryBeanByDatabase(key);
return this.geneterNextNumber(primaryBean);
}

/**
* 通过封装类生成流水号
*
* @param primaryBean
* @return
*/
public synchronized String geneterNextNumber(PrimaryBean primaryBean) {
String nextNumber = this
.geneterNextNumber(primaryBean.getStartNumber());
primaryBean.setStartNumber(nextNumber);
String dataString = this.geneterDataString(primaryBean.getDataString());
String serialNumber = primaryBean.getPrefix() + dataString + nextNumber;
return serialNumber;
}

/**
* 通过开始数字字符串生成下一个流水号
*
* @param startNumber
* @return
*/
public synchronized String geneterNextNumber(String startNumber) {
Long temp = Long.valueOf(startNumber) + 1;
StringBuilder sb = new StringBuilder();
for (int i = 0; i < startNumber.length(); i++)
sb.append("0");
DecimalFormat df = new DecimalFormat(sb.toString());
return df.format(temp);
}

/**
* 通过 格式生成日期格式
*
* @param dataformat
* @return
*/
private synchronized String geneterDataString(String dataformat) {
SimpleDateFormat formatter = new SimpleDateFormat(dataformat);
return formatter.format(new Date());
}

public static void main(String[] args) throws Exception {
XmlBeanFactory context = new XmlBeanFactory(new ClassPathResource(
"applicationContext.xml"));
BasicDataSource dataSource = (BasicDataSource) context
.getBean("dataSource");
Map map = new HashMap();
map.put("new",
"select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' as startNumber from business union select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' as startNumber");
PrimaryGenerater.newInstance(dataSource, map);
for (int i = 0; i < 100; i++) {
System.out.println(PrimaryGenerater.getInstance()
.geneterNextNumberByKey("new"));
}
}

}

package com.wesley.framework.dao;

public class PrimaryBean {
// select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' startNumber
private String prefix;
private String dataString;
private String startNumber;
private Boolean isDataBase = false;

public String getPrefix() {
return prefix;
}

public void setPrefix(String prefix) {
this.prefix = prefix;
}

public String getDataString() {
return dataString;
}

public void setDataString(String dataString) {
this.dataString = dataString;
}

public String getStartNumber() {
return startNumber;
}

public void setStartNumber(String startNumber) {
this.startNumber = startNumber;
}

public Boolean getIsDataBase() {
return isDataBase;
}

public void setIsDataBase(Boolean isDataBase) {
this.isDataBase = isDataBase;
}

}

package com.wesley.framework.dao.dbutils;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.List;

import org.apache.commons.dbcp.BasicDataSource;
import org.apache.commons.dbutils.BasicRowProcessor;
import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.ResultSetHandler;
import org.apache.commons.dbutils.handlers.BeanHandler;
import org.apache.commons.dbutils.handlers.BeanListHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.unitils.thirdparty.org.apache.commons.dbutils.DbUtils;

public class DBHelper {
private static final Logger logger = LoggerFactory
.getLogger(DBHelper.class);

private BasicDataSource pool;

private Class beanClass = null;

public DBHelper(BasicDataSource pool) {
super();
this.pool = pool;
}

public DBHelper(BasicDataSource pool, Class beanClass) {
super();
this.pool = pool;
this.beanClass = beanClass;
}

/**
* 根据无参sql语句进行查询,并返回一个对象,用于单条记录的查询
*
* @param sql
* @return Object
*/
public T query(String sql) {
Connection conn = this.getPoolConnection();
if (conn == null || beanClass == null)
return null;
QueryRunner run = new QueryRunner();
ResultSetHandler h = new BeanHandler(beanClass);
T result = null;
try {
result = run.query(conn, sql, h);
} catch (SQLException e) {
logger.info("SQLException in [" + sql + "]");
logger.error(e.getMessage());
} finally {
DbUtils.closeQuietly(conn);
}
return result;
}

/**
* 根据有参sql语句进行查询,并返回一个对象,用于单条记录的查询
*
* @param sql
* @param args
* @return Object
*/

public T query(String sql, Object... args) {
Connection conn = this.getPoolConnection();
if (conn == null || beanClass == null)
return null;
QueryRunner run = new QueryRunner();
ResultSetHandler h = new BeanHandler(beanClass);
T result = null;
try {
result = (T) run.query(conn, sql, h, args);
} catch (SQLException e) {
logger.info("SQLException in [" + sql + "]");
logger.error(e.getMessage());
} finally {
DbUtils.closeQuietly(conn);
}
return result;
}

/**
* 根据无参sql语句进行查询,并返回一个数据集,用于多条记录的查询
*
* @param sql
* @return ArrayList
*/
public List queryRecords(String sql) {
Connection conn = this.getPoolConnection();
if (conn == null || beanClass == null)
return null;
// ResultSetHandler h = new BeanListHandler(beanClass);
ResultSetHandler> h = new BeanListHandler(beanClass,
new BasicRowProcessor(new StrategyBeanProcessor(
new HumpMatcher())));
QueryRunner run = new QueryRunner();
List result = null;
try {
result = run.query(conn, sql, h);
} catch (SQLException e) {
logger.info("SQLException in [" + sql + "]");
logger.error(e.getMessage());
} finally {
DbUtils.closeQuietly(conn);
}
return result;
}

/**
* 根据有参sql语句进行查询,并返回一个数据集,用于多条记录的查询
*
* @param sql
* @param args
* @return
*/
public List queryRecords(String sql, Object... args) {
Connection conn = this.getPoolConnection();
if (conn == null || beanClass == null)
return null;
QueryRunner run = new QueryRunner();
ResultSetHandler> h = new BeanListHandler(beanClass,
new BasicRowProcessor(new StrategyBeanProcessor(
new HumpMatcher())));
List result = null;

try {
result = run.query(conn, sql, h, args);
System.out.println(result.size());
} catch (SQLException e) {
logger.info("SQLException in [" + sql + "]");
logger.error(e.getMessage());
} finally {
DbUtils.closeQuietly(conn);
}
return result;
}

/**
* 根据无参sql语句进行数据更新,并返回更新后影响的记录数
*
* @param sql
* @return int
*/
public int update(String sql) {
Connection conn = this.getPoolConnection();
if (conn == null)
return 0;
QueryRunner run = new QueryRunner();
int result = 0;

try {
result = run.update(conn, sql);
} catch (SQLException e) {
logger.info("SQLException in [" + sql + "]");
logger.error(e.getMessage());
} finally {
DbUtils.closeQuietly(conn);
}
return result;
}

/**
* 根据有参sql语句进行数据更新,并返回更新后影响的记录数
*
* @param sql
* @param args
* @return int
*/
public int update(String sql, Object... args) {
Connection conn = this.getPoolConnection();
if (conn == null)
return -1;

QueryRunner run = new QueryRunner();
int result = 0;

try {
result = run.update(conn, sql, args);
} catch (SQLException e) {
logger.info("SQLException in [" + sql + "]");
logger.error(e.getMessage());
} finally {
DbUtils.closeQuietly(conn);
}
return result;
}

/**
* 设置存放查询结果的Bean类,每个Bean类对象对应一条查询的结果记录
*
* @param beanClass
* ,如User.class
*/
public void setBeanClass(Class beanClass) {
this.beanClass = beanClass;
}

public Connection getPoolConnection() {
int x = 3;
while (x > 0) {
try {
return pool.getConnection();
} catch (SQLException e) {
x--;
logger.error(e.getMessage());
}
}
return null;
}

public void dispose() {
try {
this.pool.close();
} catch (SQLException e) {
logger.error(e.getMessage());
}
}
}

		factory-method="newInstance">




select 'SP-' as prefix,'yyyyMMdd' as dataString,'000000' as
startNumber union select null as prefix,null as dataString,null as
startNumber from business




你可能感兴趣的:(JAVA流水号生成, 支持数据库查询,Spring注入)