Spring生成下载链接

时间:2016-03-21 08:56:18

标签: spring spring-security-oauth2 spring-web

有没有办法使用spring web,security-oauth stack生成用于下载文件的临时链接?

例如domain.com/document/ed3dk4kfjw34k43kd4k3cc仅适用于当前会话?

1 个答案:

答案 0 :(得分:3)

您可以尝试将Map<String, String>添加到会话中。之后,您可以将生成的唯一字符串和文件名存储到此映射中。每当需要通过唯一生成的字符串加载文件时,您将按字符串找到真实文件名并将其发送给客户端。用于演示想法的简单组件:

@Component
@Scope(value = "session")
public class SessionFileMap {

    private Map<String, String> fileMap = new HashMap<>();

    public String getUniqueString(String fileName){
        for(String uniqueName: fileMap.keySet()){ 
            //check, if file already in map, return it
            if(fileMap.get(uniqueName).equals(fileName)) return uniqueName;
        }            
        //otherwise, create new
        String uniqueName = generateUniqueName();
        fileMap.put(uniqueName, fileName);
        return uniqueName;
    }

    public String getFileName(String uniqueString){
        if(fileMap.containsKey(uniqueString)){
            return fileMap.get(uniqueString);
        }
        return null;
    }

    private String generateUniqueName() {
        String uniqueString = //generation of unique string
        return uniqueString;
    }
}

当然,您必须使此组件范围session。和there is很好的例子,如何生成唯一的字符串。现在使用此组件的示例:

@Controller
@Scope(value = "session")
public class FileController {

    @Autowired
    private SessionFileMap fileMap;

    @Autowired
    private ApplicationContext context;

    @RequestMapping("/file")
    public String showLink(ModelMap model, HttpSession session){
        String uniqueString = fileMap.getUniqueString("/filepath/filename.ext");
        model.addAttribute("uniqueString", uniqueString);
        return "file";
    }

    @RequestMapping("/download/{uniqueString}")
    public void download(@PathVariable("uniqueString") String uniqueString,
                           HttpServletResponse response){
        String fileName = fileMap.getFileName(uniqueString);
        try{
            Resource resource = context.getResource("file:"+fileName);
            try (InputStream is = resource.getInputStream()) {

                //prepare all headers for download ...

                IOUtils.copy(is, response.getOutputStream());
                response.flushBuffer();
            }
        }catch(Exception e){
            throw new RuntimeException(e);
        }
    }
}

控制器必须具有session的范围以及组件。如果您注意到,我使用IOUtils.copy()中的org.apache.commons来复制流,但您可以根据需要进行复制。在视图中,链接将如下所示:

<html>
<head>
    <title></title>
</head>
<body>
  <a href="/download/${uniqueString}">Download</a>
</body>
</html>

这只是基本思想的证明。所有细节都取决于您。