package cn.hellohao.controller; import java.io.FileInputStream; import java.text.SimpleDateFormat; import java.util.*; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpSession; import cn.hellohao.pojo.*; import cn.hellohao.service.*; import cn.hellohao.service.impl.*; import cn.hellohao.utils.IPPortUtil; import cn.hellohao.utils.*; import com.alibaba.fastjson.JSONObject; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.ui.Model; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.multipart.MultipartFile; import com.alibaba.fastjson.JSONArray; @Controller public class UpdateImgController { @Autowired private NOSImageupload nOSImageupload; @Autowired private UserService userService; @Autowired private KeysService keysService; @Autowired private OSSImageupload ossImageupload; @Autowired private ConfigService configService; @Autowired private UploadConfigService uploadConfigService; @Autowired private USSImageupload ussImageupload; @Autowired private KODOImageupload kodoImageupload; @Autowired private COSImageupload cosImageupload; @Autowired private FTPImageupload ftpImageupload; @RequestMapping({"/", "/index"}) public String indexImg(Model model, HttpSession httpSession) { Print.Normal("欢迎使用Hellohao图床源码。者也许是最好用的java图床项目。"); Print.Normal("当前项目路径："+System.getProperty("user.dir")); Config config = configService.getSourceype();//查询当前系统使用的存储源类型。 UploadConfig uploadConfig = uploadConfigService.getUpdateConfig(); Integer filesizetourists = 0; Integer filesizeuser = 0; Integer imgcounttourists = 0; Integer imgcountuser = 0; if(uploadConfig.getFilesizetourists()!=null){filesizetourists = uploadConfig.getFilesizetourists();} if(uploadConfig.getFilesizeuser()!=null){filesizeuser = uploadConfig.getFilesizeuser();} if(uploadConfig.getImgcounttourists()!=null){imgcounttourists = uploadConfig.getImgcounttourists();} if(uploadConfig.getImgcountuser()!=null){imgcountuser = uploadConfig.getImgcountuser();} //Boolean b =false; Keys key = null; if(config.getSourcekey()!=0 && config.getSourcekey()!=5){ key= keysService.selectKeys(config.getSourcekey());//然后根据类型再查询key //b = StringUtils.doNull(key);//判断对象是否有空值 if(key.getStorageType()!=0 && key.getStorageType()!=null){ if(key.getStorageType()==1){ nOSImageupload.Initialize(key);//实例化网易 }else if (key.getStorageType()==2){ OSSImageupload.Initialize(key); }else if(key.getStorageType()==3){ USSImageupload.Initialize(key); }else if(key.getStorageType()==4){ KODOImageupload.Initialize(key); }else if(key.getStorageType()==6){ COSImageupload.Initialize(key); }else if(key.getStorageType()==7){ FTPImageupload.Initialize(key); } else{ Print.Normal("为获取到存储参数，或者使用存储源是本地的。"); } } } //if(b){ //} User u = (User) httpSession.getAttribute("user"); String email = (String) httpSession.getAttribute("email"); if (email != null) { //登陆成功 Integer ret = userService.login(u.getEmail(), u.getPassword()); if (ret > 0) { User user = userService.getUsers(u.getEmail()); model.addAttribute("username", user.getUsername()); model.addAttribute("level", user.getLevel()); model.addAttribute("loginid", 100); model.addAttribute("imgcount", imgcountuser); model.addAttribute("filesize", filesizeuser*1024*1024); } else { model.addAttribute("loginid", -1); model.addAttribute("imgcount", imgcounttourists); } } else { model.addAttribute("loginid", -2); model.addAttribute("imgcount", imgcounttourists); model.addAttribute("filesize", filesizetourists*1024*1024); } model.addAttribute("suffix", uploadConfig.getSuffix()); model.addAttribute("config", config); model.addAttribute("uploadConfig", uploadConfig); Integer isupdate = 1; if(uploadConfig.getIsupdate()!=1){ isupdate = (u == null) ? 0: 1;//如果u等于null那么isupdate就等于0，否则等于1 //isupdate=0; } model.addAttribute("ykxz", isupdate); return "index"; } @RequestMapping(value = "/upimg") @ResponseBody public String upimg( HttpSession session , @RequestParam(value = "file", required = false) MultipartFile[] file) throws Exception { Config config = configService.getSourceype();//查询当前系统使用的存储源类型。 UploadConfig uploadConfig = uploadConfigService.getUpdateConfig(); User u = (User) session.getAttribute("user"); JSONArray jsonArray = new JSONArray(); JSONObject jsonObject = new JSONObject(); Keys key = keysService.selectKeys(config.getSourcekey()); Boolean b =false; if(config.getSourcekey()==5){ b =true; }else{ b = StringUtils.doNull(config.getSourcekey(),key);//判断对象是否有空值 } if(b){ long stime = System.currentTimeMillis(); String userpath = "tourist"; if(uploadConfig.getUrltype()==2){ java.text.DateFormat dateFormat = new java.text.SimpleDateFormat("yyyy/MM/dd"); userpath = dateFormat.format(new Date()); }else{ if (u != null) { userpath = u.getUsername(); } } Map map = new HashMap<>(); for (MultipartFile multipartFile : file) { // 获取ImageReader对象的迭代器 //获取文件名 String fileName = multipartFile.getOriginalFilename(); String lastname = fileName.substring(fileName.lastIndexOf(".") + 1);//获取文件后缀 if (!multipartFile.isEmpty()) { //判断文件是否为空 map.put(lastname, multipartFile); System.out.println("===name==="+multipartFile.getOriginalFilename()); //文件名 //multipartFile.getSize(); //文件大小 } } Map m = null; if(key.getStorageType()==1){ m = nOSImageupload.Imageupload(map, userpath,null); }else if (key.getStorageType()==2){ m = ossImageupload.ImageuploadOSS(map, userpath,null); }else if(key.getStorageType()==3){ m = ussImageupload.ImageuploadUSS(map, userpath,null); }else if(key.getStorageType()==4){ m = kodoImageupload.ImageuploadKODO(map, userpath,null); }else if(key.getStorageType()==5){ m = LocUpdateImg.ImageuploadLOC(map, userpath,null); }else if(key.getStorageType()==6){ m = cosImageupload.ImageuploadCOS(map, userpath,null); }else if(key.getStorageType()==7){ m = ftpImageupload.ImageuploadFTP(map, userpath,null); } else{ System.err.println("未获取到对象存储参数，上传失败。"); } Images img = new Images(); SimpleDateFormat df=new SimpleDateFormat("yyyy-MM-dd"); String times = df.format(new Date()); System.out.println("上传图片的时间是："+times); for (Map.Entry entry : m.entrySet()) { if(key.getStorageType()==5){ if(config.getDomain()!=null){ jsonObject.put("imgurls",config.getDomain()+"/links/"+entry.getKey().getImgurl()); jsonObject.put("imgnames",entry.getKey().getImgname()); img.setImgurl(config.getDomain()+"/links/"+entry.getKey().getImgurl());//图片链接 }else{ jsonObject.put("imgurls",config.getDomain()+"/links/"+entry.getKey().getImgurl()); jsonObject.put("imgnames",entry.getKey().getImgname()); img.setImgurl("http://"+IPPortUtil.getLocalIP()+":"+IPPortUtil.getLocalPort()+"/links/"+entry.getKey().getImgurl());//图片链接 } }else{ jsonObject.put("imgurls",entry.getKey().getImgurl()); jsonObject.put("imgnames",entry.getKey().getImgname()); img.setImgurl(entry.getKey().getImgurl());//图片链接 } jsonArray.add(jsonObject); img.setUpdatetime(times); img.setSource(key.getStorageType()); if (u == null) { img.setUserid(0); } else { img.setUserid(u.getId());//用户id } img.setSizes((entry.getValue()) / 1024); img.setImgname(SetText.getSubString(entry.getKey().getImgurl(), key.getRequestAddress() + "/", "")); img.setAbnormal(0); userService.insertimg(img); long etime = System.currentTimeMillis(); System.out.println("上传图片所用时长：" + String.valueOf(etime - stime) + "ms"); } }else{ jsonObject.put("imgurls",-1); } //开辟新进程鉴黄。现在已经改成定时器 //查询鉴黄功能是否启动，1为启用 // Imgreview imgreview = imgreviewService.selectByPrimaryKey(1); // if (imgreview.getUsing() == 1) { // //上传完成后开辟新的线程进行图片鉴黄。 // JianHuangThread thread = new JianHuangThread(imgreviewService, key, u, m); // thread.start(); // } System.out.println(jsonArray.toString()); return jsonObject.toString(); } //根据网络图片url上传 @PostMapping(value = "/upurlimg") @ResponseBody public String upurlimg(HttpSession session, String imgurl, HttpServletRequest request) throws Exception { Config config = configService.getSourceype();//查询当前系统使用的存储源类型。 UploadConfig uploadConfig = uploadConfigService.getUpdateConfig(); User u = (User) session.getAttribute("user"); String userpath = "tourist"; if(uploadConfig.getUrltype()==2){ java.text.DateFormat dateFormat = new java.text.SimpleDateFormat("yyyy/MM/dd"); userpath = dateFormat.format(new Date()); }else{ if (u != null) { userpath = u.getUsername(); } } JSONArray jsonArray = new JSONArray(); Keys key = keysService.selectKeys(config.getSourcekey()); long imgsize = ImgUrlUtil.getFileLength(imgurl); Integer youke = uploadConfig.getFilesizetourists(); Integer yonghu = uploadConfig.getFilesizeuser(); String uuid= UUID.randomUUID().toString().replace("-", ""); Boolean bo =false; if(config.getSourcekey()==5){ bo =true; }else{ bo = StringUtils.doNull(config.getSourcekey(),key);//判断对象是否有空值 } //先判断对象存储key是不是null Print.warning("上传地址是："+request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"); if(bo){ long stime = System.currentTimeMillis(); //判断是会员还是游客 if(u!=null){ //判断文件大小 if(imgsize>0 && imgsize<=(yonghu*1024*1024)){ try{ boolean bl =ImgUrlUtil.downLoadFromUrl(imgurl, uuid, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"); if(bl==true){ FileInputStream is = new FileInputStream(request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid); byte[] b = new byte[3]; is.read(b, 0, b.length); String xxx = ImgUrlUtil.bytesToHexString(b); xxx = xxx.toUpperCase(); System.out.println("头文件是：" + xxx); String ooo = TypeDict.checkType(xxx); System.out.println("后缀名是：" + ooo); if(is!=null){ is.close(); } //判断文件头是否是图片 if(!ooo.equals("0000")){ Map map = new HashMap<>(); map.put(ooo, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid); Map m = null; if(key.getStorageType()==1){ m = nOSImageupload.Imageupload(null, userpath,map); }else if (key.getStorageType()==2){ m = ossImageupload.ImageuploadOSS(null, userpath,map); }else if(key.getStorageType()==3){ m = ussImageupload.ImageuploadUSS(null, userpath,map); }else if(key.getStorageType()==4){ m = kodoImageupload.ImageuploadKODO(null, userpath,map); }else if(key.getStorageType()==5){ m = LocUpdateImg.ImageuploadLOC(null,userpath,map); }else if(key.getStorageType()==6){ m = cosImageupload.ImageuploadCOS(null,userpath,map); }else if(key.getStorageType()==7){ m = ftpImageupload.ImageuploadFTP(null,userpath,map); } else{ System.err.println("未获取到对象存储参数，上传失败。"); } Images img = new Images(); SimpleDateFormat df=new SimpleDateFormat("yyyy-MM-dd"); String times = df.format(new Date()); System.out.println("上传图片的时间是："+times); for (Map.Entry entry : m.entrySet()) { if(key.getStorageType()==5){ if(config.getDomain()!=null){ jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl()); img.setImgurl(config.getDomain()+"/links/"+entry.getKey().getImgurl());//图片链接 }else{ jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl()); img.setImgurl("http://"+IPPortUtil.getLocalIP()+":"+IPPortUtil.getLocalPort()+"/links/"+entry.getKey().getImgurl());//图片链接 } }else{ jsonArray.add(entry.getKey().getImgurl()); img.setImgurl(entry.getKey().getImgurl()); } img.setUpdatetime(times); img.setSource(key.getStorageType()); if (u == null) { img.setUserid(0);//用户id } else { img.setUserid(u.getId());//用户id } img.setSizes((entry.getValue())); img.setImgname(SetText.getSubString(entry.getKey().getImgurl(), key.getRequestAddress() + "/", "")); img.setAbnormal(0); userService.insertimg(img); long etime = System.currentTimeMillis(); System.out.println("上传图片所用时长：" + String.valueOf(etime - stime) + "ms"); } }else{ jsonArray.add(-3); } } }catch (Exception e) { // TODO: handle exception Print.warning(e.toString()); jsonArray.add(-4); } }else{ //文件过大 jsonArray.add(-2); } }else{ if(imgsize>0 && imgsize<=(youke*1024*1024)){ try{ boolean bl = ImgUrlUtil.downLoadFromUrl(imgurl, uuid, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"); if(bl==true){ FileInputStream is = new FileInputStream(request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid); byte[] b = new byte[3]; is.read(b, 0, b.length); String xxx = ImgUrlUtil.bytesToHexString(b); xxx = xxx.toUpperCase(); //System.out.println("头文件是：" + xxx); String ooo = TypeDict.checkType(xxx); //System.out.println("后缀名是：" + ooo); if(is!=null){ is.close(); } //判断文件头是否是图片 if(!xxx.equals("0000")){ Map map = new HashMap<>(); map.put(ooo, request.getSession().getServletContext().getRealPath("/")+"/hellohaotmp/"+uuid); Map m = null; if(key.getStorageType()==1){ m = nOSImageupload.Imageupload(null, userpath,map); }else if (key.getStorageType()==2){ m = ossImageupload.ImageuploadOSS(null, userpath,map); }else if(key.getStorageType()==3){ m = ussImageupload.ImageuploadUSS(null, userpath,map); }else if(key.getStorageType()==4){ m = kodoImageupload.ImageuploadKODO(null, userpath,map); }else if(key.getStorageType()==5){ m =LocUpdateImg.ImageuploadLOC(null,userpath, map); }else if(key.getStorageType()==6){ m =cosImageupload.ImageuploadCOS(null,userpath, map); }else if(key.getStorageType()==7){ m = ftpImageupload.ImageuploadFTP(null,userpath,map); } else{ System.err.println("未获取到对象存储参数，上传失败。"); } Images img = new Images(); SimpleDateFormat df=new SimpleDateFormat("yyyy-MM-dd"); String times = df.format(new Date()); System.out.println("上传图片的时间是："+times); for (Map.Entry entry : m.entrySet()) { if(key.getStorageType()==5){ if(config.getDomain()!=null){ jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl()); img.setImgurl(config.getDomain()+"/links/"+entry.getKey().getImgurl());//图片链接 }else{ jsonArray.add(config.getDomain()+"/links/"+entry.getKey().getImgurl()); img.setImgurl("http://"+IPPortUtil.getLocalIP()+":"+IPPortUtil.getLocalPort()+"/links/"+entry.getKey().getImgurl());//图片链接 } }else{ jsonArray.add(entry.getKey().getImgurl()); } //img.setImgurl(entry.getKey());//图片链接 img.setUpdatetime(times); img.setSource(key.getStorageType()); if (u == null) { img.setUserid(0);//用户id } else { img.setUserid(u.getId());//用户id } img.setSizes((entry.getValue())); img.setImgname(SetText.getSubString(entry.getKey().getImgurl(), key.getRequestAddress() + "/", "")); img.setAbnormal(0); userService.insertimg(img); long etime = System.currentTimeMillis(); System.out.println("上传图片所用时长：" + String.valueOf(etime - stime) + "ms"); } }else{ jsonArray.add(-3); } } }catch (Exception e) { // TODO: handle exception Print.warning(e.toString()); jsonArray.add(-4); } }else{ //文件过大 jsonArray.add(-2); } } }else{ jsonArray.add(-1); } return jsonArray.toString(); /** * 错误返回值含义： * -1 存储源key未配置 * -2 目标图片太大或者不存在 * -3 文件类型不符合要求 * */ } //刪除用戶 @RequestMapping("/sentence") @ResponseBody public String sentence(HttpSession session, Integer id) { JSONArray jsonArray = new JSONArray(); String text = Sentence.getURLContent(); jsonArray.add(text); return jsonArray.toString(); } //ajax查询用户是否已经登录 @RequestMapping(value = "/islogin") @ResponseBody public String islogin(HttpSession session) { JSONObject jsonObject = new JSONObject(); User user = (User) session.getAttribute("user"); if(user!=null){ if (user.getEmail() != null && user.getPassword() != null) { jsonObject.put("username",user.getUsername()); jsonObject.put("level",user.getLevel()); jsonObject.put("lgoinret",1); }else{ jsonObject.put("lgoinret",0); } } return jsonObject.toString(); } @RequestMapping(value = "/getsentence") @ResponseBody public String getsentence(HttpSession session) { JSONObject jsonObject = new JSONObject(); String yiyan = Sentence.getURLContent(); return yiyan; } @RequestMapping("/err") public String err() { return "err"; } }