首页 > 代码库 > Java 感知Mysql存储过程变量数量

Java 感知Mysql存储过程变量数量

 在项目中,可能会遇到sybase 移植到 mysql的情况,因为sybase 支持存储过程的可变参数,而mysql不能支持,所以,在调用mysql的时候,需要感知存储过程到底有几个参数,来合理的配置参数数量:

如下是代码

package com.xxx.util;import java.sql.Connection;import java.sql.DriverManager;import java.sql.ResultSet;import java.sql.Statement;import java.util.Hashtable;import java.util.Map;import java.util.regex.Matcher;import java.util.regex.Pattern;public class ProcedureHelper {		/*是否是开发模式*/	static boolean DEV = false;	static Map<String,Integer> CACHE = new Hashtable<String,Integer>(100);  	static Pattern PATTERN_ARGS = Pattern.compile("`\\s*\\((.*)\\)\\s*begin",Pattern.DOTALL|Pattern.CASE_INSENSITIVE);	static Pattern PATTERN_ARG = Pattern.compile("(in|out|inout)[^-,]*(,|$)",Pattern.DOTALL|Pattern.CASE_INSENSITIVE);	private static int doWork(Connection con,String proc) throws Exception {		Statement stmt = null;		stmt = con.createStatement(); //3、Statement 接口需要通过Connection 接口进行实例化操作		ResultSet rs = stmt.executeQuery("show create procedure "+proc);				if(rs.next()){			//第三列为 存储过程的代码			String code = rs.getString(3);			Matcher m = PATTERN_ARGS.matcher(code);			if(m.find()){				String args = m.group(1);				if(args.matches("\\s*"))					return 0;				else{					Matcher m2 = PATTERN_ARG.matcher(args);					int num = 0;					while(m2.find()){						//System.out.println(m2.group());						num++;					}					return num;				}			}else{				throw new RuntimeException("存储过程: "+proc+"无法通过正则表达式获取参数个数:\n"+code);							}		}		rs.close();		stmt.close();		throw new RuntimeException("没有该存储过程");			}	/**	 * 注意, 在存储过程的参数中,不能写注释,特别是 in xxx, 格式的注释,不然会被错误的识别	 * @param con JDBC连接, 且不在函数内释放资源	 * @param proc 调用的存储过程	 * @param args 传递给存储过程的参数	 * @throws 解析失败	 * */	public static int getProcedureArgsNumber(Connection con,String proc) throws Exception {		if(con == null || proc == null|| proc.equals("")){			throw new RuntimeException("proc, con 参数不能为空");		}		//这里可以做缓存		return doWork(con,proc);	}	public static void main(String arg[]) throws Exception{				Connection con = null; //表示数据库的连接对象  		Class.forName("com.mysql.jdbc.Driver"); //1、使用CLASS 类加载驱动程序  		con = DriverManager.getConnection("jdbc:mysql://127.0.0.1:3306/test?useUnicode=true&characterEncoding=UTF-8","root","root"); //2、连接数据库		System.out.println(getProcedureArgsNumber(con,"my_procedure"));		con.close();	}}

  该代码对存储过程的感知,是利用正则表达式的,所以在匹配的时候,可能遇到一些极端的情况,如 注释中包含了符合规则的 语句,不过一般来说,这种情况极少发生。

Java 感知Mysql存储过程变量数量