uploadcontroller.java

来自「spring2.0技术手册随书事例,结合例子学习起来更轻松」· Java 代码 · 共 57 行

package onlyfun.caterpillar;

import java.io.BufferedOutputStream;
import java.io.FileOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.springframework.validation.BindException;
import org.springframework.web.bind.ServletRequestDataBinder;
import org.springframework.web.multipart.
                       support.ByteArrayMultipartFileEditor;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.
                                   mvc.SimpleFormController;

public class UploadController extends SimpleFormController {
    private String path;
    
    public UploadController() {
        setCommandClass(FileForm.class);
    }
    
    protected ModelAndView onSubmit(
                            HttpServletRequest request,
                            HttpServletResponse response,
                            Object command,
                            BindException errors) 
                                        throws Exception {
        FileForm form = (FileForm) command;
        
        String storedPath = path + 
                            System.getProperty("file.separator") + 
                            form.getName();
        
        BufferedOutputStream bufferedOutputStream = 
                new BufferedOutputStream(
                       new FileOutputStream(storedPath));
        
        bufferedOutputStream.write(form.getContents());
        bufferedOutputStream.close();
        
        return new ModelAndView(getSuccessView(), 
                                "filename", form.getName());
    }
    
    protected void initBinder(HttpServletRequest request, 
                          ServletRequestDataBinder binder) 
                                            throws Exception {
        super.initBinder(request, binder); 
        binder.registerCustomEditor(
                byte[].class, new ByteArrayMultipartFileEditor());
    }
    
    public void setPath(String path) {
        this.path = path;
    }
}