|  | @@ -1,6 +1,7 @@
 | 
	
		
			
				|  |  |  package com.kmall.gen.controller;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  import com.alibaba.fastjson.JSON;
 | 
	
		
			
				|  |  | +import com.alibaba.fastjson.JSONArray;
 | 
	
		
			
				|  |  |  import com.kmall.gen.service.SysGeneratorService;
 | 
	
		
			
				|  |  |  import com.kmall.common.utils.PageUtils;
 | 
	
		
			
				|  |  |  import com.kmall.common.utils.Query;
 | 
	
	
		
			
				|  | @@ -10,6 +11,7 @@ import org.apache.commons.io.IOUtils;
 | 
	
		
			
				|  |  |  import org.apache.shiro.authz.annotation.RequiresPermissions;
 | 
	
		
			
				|  |  |  import org.springframework.beans.factory.annotation.Autowired;
 | 
	
		
			
				|  |  |  import org.springframework.stereotype.Controller;
 | 
	
		
			
				|  |  | +import org.springframework.web.bind.annotation.RequestBody;
 | 
	
		
			
				|  |  |  import org.springframework.web.bind.annotation.RequestMapping;
 | 
	
		
			
				|  |  |  import org.springframework.web.bind.annotation.RequestParam;
 | 
	
		
			
				|  |  |  import org.springframework.web.bind.annotation.ResponseBody;
 | 
	
	
		
			
				|  | @@ -60,9 +62,8 @@ public class SysGeneratorController {
 | 
	
		
			
				|  |  |          //获取表名,不进行xss过滤
 | 
	
		
			
				|  |  |          HttpServletRequest orgRequest = XssHttpServletRequestWrapper.getOrgRequest(request);
 | 
	
		
			
				|  |  |          String tables = orgRequest.getParameter("tables");
 | 
	
		
			
				|  |  | -        tableNames = JSON.parseArray(tables).toArray(tableNames);
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -        byte[] data = sysGeneratorService.generatorCode(tableNames);
 | 
	
		
			
				|  |  | +        byte[] data = sysGeneratorService.generatorCode(tables.split(","));
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |          response.reset();
 | 
	
		
			
				|  |  |          response.setHeader("Content-Disposition", "attachment; filename=\"AutoCode.zip\"");
 |