mybatis运行sql脚本工具,带参数

5 篇文章 0 订阅
	@Resource(name = "dataSource")
	DruidDataSource dataSource;

//参数替换占位符
	@Value("${scriptrun.sql.splitMark}")
	String splitMark;
	
	//转义字符
	static String[] specalMarks = {"?"};	

public IBaseResult runSqlScript(MultipartFile file, String userId, List<String> param) {
		if(null == userId || "".equals(userId.trim())){
			return DefaultResult.buildFailedResult(RetResultCode.E13002);
		}
		

		ByteArrayOutputStream baos = null;
		InputStream fis = null;
		String script = "";
		try {
			fis = file.getInputStream();
			baos = new ByteArrayOutputStream();
			byte[] buffer = new byte[1024];
			for (int len = 0; (len = fis.read(buffer)) > 0;) {
				baos.write(buffer, 0, len);
			}
			script = new String(baos.toByteArray(),defaultEncode);
		} catch (Exception e) {
			logger.error(e.getMessage(),e);
		} finally {
			try {
				if(null != baos){
					baos.close();
				}
				if(null != fis){
					fis.close();
				}
			} catch (Exception e2) {
				logger.error(e2.getMessage(),e2);
			}
		}
		
		//String script = fileUtil.getFileContentToString(fileObj);
		
		//替换参数
		if(null != param && !param.isEmpty()){
			for(int i=0;i<param.size();i++){
				int thisMarkIndex = script.indexOf(splitMark);
				
				if(param.contains(splitMark)){
					return DefaultResult.buildFailedResult(RetResultCode.E13005);
				}
				if(thisMarkIndex <= -1){
					return DefaultResult.buildFailedResult(RetResultCode.E13003);
				}

				String changeMark = splitMark;
				if(Arrays.asList(specalMarks).contains(splitMark)){
					changeMark = "["+splitMark+"]";
				}
				script = script.replaceFirst(changeMark,param.get(i));
			}
		}
		if(script.indexOf(splitMark) >-1){
			return DefaultResult.buildFailedResult(RetResultCode.E13004);
		}
		

		DruidPooledConnection conn;
		try {
			//自行获取数据库连接
			conn = dataSource.getConnection();
			ScriptRunner runner = new ScriptRunner(conn);// 创建ScriptRunner
			runner.setAutoCommit(true);// 自动提交
			 ByteArrayInputStream stream = new ByteArrayInputStream(script.getBytes());
			runner.runScript(new InputStreamReader(stream, defaultEncode));
		} catch (SQLException e) {
			logger.error(e.getMessage(),e);
			return DefaultResult.buildFailedResult(RetResultCode.E14010);
		} catch (UnsupportedEncodingException e) {
			logger.error(e.getMessage(),e);
			return DefaultResult.buildFailedResult(RetResultCode.E14010);
		}

		return DefaultResult.buildSuccessResult();
	}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值