import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.stereotype.Controller; import org.springframework.util.ClassUtils; import org.springframework.web.bind.annotation.RequestMapping; import javax.servlet.ServletException; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.*; @Controller public class PageController { private static ClassLoader classLoader = ClassUtils.getDefaultClassLoader(); private static final Logger log = LoggerFactory.getLogger(PageController.class); @RequestMapping("/") public void root(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { request.getRequestDispatcher("/index.html").forward(request, response); } @RequestMapping("**.html")
public void page(HttpServletRequest request, HttpServletResponse response) throws IOException {
String servletPath = request.getServletPath();
log.info(request.getMethod() + " " + servletPath);
String filePath = "templates" + servletPath;
InputStream inputStream = classLoader.getResourceAsStream(filePath);
if (inputStream == null) {
response.sendError(404);
return;
}
response.setContentType("text/html; charset=utf-8");
OutputStream outputStream = response.getOutputStream();
int readChar;
while ((readChar = inputStream.read()) != -1) {
outputStream.write(readChar);
}
outputStream.flush();
inputStream.close();
} }
