hdy
7 天以前 b9df2fc5c64a1d989991655a9e42e4d1f2ec4075
billion-generator/src/main/java/com/billion/generator/controller/GenController.java
@@ -6,6 +6,9 @@
import java.util.List;
import java.util.Map;
import javax.servlet.http.HttpServletResponse;
import com.billion.common.annotation.DataSource;
import com.billion.common.enums.DataSourceType;
import org.apache.commons.io.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
@@ -84,6 +87,7 @@
     */
    @PreAuthorize("@ss.hasPermi('tool:gen:list')")
    @GetMapping("/db/list")
    @DataSource(value = DataSourceType.SLAVE)
    public TableDataInfo dataList(GenTable genTable)
    {
        startPage();
@@ -250,7 +254,7 @@
        response.reset();
        response.addHeader("Access-Control-Allow-Origin", "*");
        response.addHeader("Access-Control-Expose-Headers", "Content-Disposition");
        response.setHeader("Content-Disposition", "attachment; filename=\"ruoyi.zip\"");
        response.setHeader("Content-Disposition", "attachment; filename=\"billion.zip\"");
        response.addHeader("Content-Length", "" + data.length);
        response.setContentType("application/octet-stream; charset=UTF-8");
        IOUtils.write(data, response.getOutputStream());