diff --git a/src/com/hellogithub/servlet/retUserAvServlet.java b/src/com/hellogithub/servlet/retUserAvServlet.java index 692340e..328b522 100644 --- a/src/com/hellogithub/servlet/retUserAvServlet.java +++ b/src/com/hellogithub/servlet/retUserAvServlet.java @@ -38,6 +38,7 @@ public class retUserAvServlet extends HttpServlet { { userEntity = userService.selectUserByName(userEntity.getUserName()); FileInputStream fis = new FileInputStream (userEntity.getUserAvatar()); + String type = userEntity.getUserAvatar().substring(userEntity.getUserAvatar().lastIndexOf(".")+1); int size = fis.available(); if(size != 0) @@ -46,8 +47,7 @@ public class retUserAvServlet extends HttpServlet { fis.read(data) ; fis.close(); resp.reset(); - resp.setHeader("Content-Type", "image/jpg"); -// resp.setContentType("image/jpg"); + resp.setHeader("Content-Type", "image/"+type); OutputStream os = resp.getOutputStream() ; os.write(data); os.flush();