download

package th.in.oneauthen.servlet;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang3.StringUtils;

/**
 * Servlet implementation class DownloadServlet
 */
@WebServlet("/DownloadServlet")
public class DownloadServlet extends HttpServlet {
    private static final long serialVersionUID = 1L;

    /**
     * @see HttpServlet#HttpServlet()
     */
    public DownloadServlet() {
        super();
        // TODO Auto-generated constructor stub
    }

    /**
     * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response)
     */
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

        String FileLocation = (String) request.getSession(false).getAttribute(SigningServlet.PARAM_FILE_LOCATION);
        String check  = (String) request.getSession().getAttribute(LoginServlet.CHECK);
        if (check == null){
            check = "0";
        }


        if(check.equals("1")){
            byte[] signPDF = (byte[]) request.getSession().getAttribute(SigningServlet.PDF_DATA);
            response.setContentType("application/pdf");
            response.addHeader("Content-Disposition", "attachment; filename=signDocument.pdf" );
            response.setContentLength(signPDF.length);
            response.getOutputStream(). write(signPDF);


           request.getSession().setAttribute(LoginServlet.CHECK, "0");



        }
        else {
             response.sendRedirect("Error.jsp");
        }

        if (!StringUtils.isEmpty(FileLocation)){
            String Del = FileLocation;
            Files.delete(Paths.get(Del));
        }

    }

//  protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
//
//          byte[] signPDF = Files.readAllBytes(Paths.get("E:\\BallzaR5R5\\test.pdf"));
//          response.setContentType("application/pdf");
//          response.addHeader("Content-Disposition", "attachment; filename=signDocument.pdf" );
//          response.setContentLength(signPDF.length);
//          response.getOutputStream(). write(signPDF);
//
//
//         request.getSession().setAttribute(LoginServlet.CHECK, "0");
//
//  }


    /**
     * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)
     */
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // TODO Auto-generated method stub
        doGet(request, response);
    }

}