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();
	}

}

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

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。