springboot1.zip

package com.example.wxp.controller; //hotel import com.example.wxp.entity.Hotel; import com.example.wxp.entity.Room; import com.example.wxp.service.HotelService; import com.example.wxp.service.RoomService; import com.example.wxp.util.PageUtil; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.ui.Model; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletRequest; import java.io.File; import java.io.IOException; import java.text.SimpleDateFormat; import java.util.*; @Controller @RequestMapping("room") public class RoomController { @Autowired private RoomService roomService; @Autowired private HotelService hotelService; //查询全部房间信息 @RequestMapping("selectRoomList") public String selectRoomList(Model model, String currentPage, String hid){ int currentpage = 0; if (hid == null || hid.trim().equals("")) { if (currentPage == null || currentPage.trim().equals("")) { currentpage = 1; } else { currentpage = Integer.parseInt(currentPage); } Map<String, Object> map = new HashMap<String, Object>(); int count = roomService.getPageCount(); int pageSize = 2; PageUtil<Room> param = new PageUtil<Room>(); param.setTotalCount(count); param.setPageSize(pageSize); param.setCurrentPage(currentpage); int startRow = param.getStartRow(); map.put("startRow", startRow); map.put("pageSize", pageSize); List<Room> roomList = roomService.selectRoomByPage(startRow, pageSize); param.setList(roomList); model.addAttribute("roomList", roomList); model.addAttribute("util", param); model.addAttribute("currentPage", currentpage); List<Hotel> hotelList = hotelService.selectHotel(); model.addAttribute("hotelList", hotelList); return "empList"; }else{ int hotelId = Integer.parseInt(hid); if (currentPage == null || currentPage.trim().equals("")) { currentpage = 1; } else { currentpage = Integer.parseInt(currentPage); } Map<String, Object> map = new HashMap<String, Object>(); int count = roomService.selectcountById(hotelId); int pageSize = 2; PageUtil<Room> param = new PageUtil<Room>(); param.setTotalCount(count); param.setPageSize(pageSize); param.setCurrentPage(currentpage); int startRow = param.getStartRow(); int totalPage = param.getTotalPage(); map.put("startRow", startRow); map.put("pageSize", pageSize); map.put("hotelId", hotelId); List<Room> roomList = roomService.selectRoomByHId(map); param.setList(roomList); model.addAttribute("roomList", roomList); model.addAttribute("util", param); model.addAttribute("currentPage", currentpage); List<Hotel> hotelList = hotelService.selectHotel(); model.addAttribute("hotelList", hotelList); model.addAttribute("hid", hid); return "empList"; } } //查询全部房间信息 @RequestMapping("addRoom") public String addRoom(Room room,HttpServletRequest request, MultipartFile uploadFile) throws IOException { if(uploadFile.getSize()!=0) { String newFileName = uploadFile(request, uploadFile); room.setRoomPicture(newFileName); } System.out.println(room.toString()); roomService.addRoom(room); return "redirect:/room/selectRoomList"; } //上传文件 public String uploadFile(HttpServletRequest request, MultipartFile uploadFile) throws IOException { SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmssSS"); String res = sdf.format(new Date()); //uploads文件夹位置 String rootPath =request.getServletContext().getRealPath("/hotelPictures"); //原始名称 String originalFilename = uploadFile.getOriginalFilename(); //新的文件名称 String newFileName = Math.random()+res+originalFilename.substring(originalFilename.lastIndexOf(".")); //创建年月文件夹 Calendar date = Calendar.getInstance(); File dateDirs = new File(date.get(Calendar.YEAR) + File.separator + (date.get(Calendar.MONTH)+1)); //新文件 File newFile = new File(rootPath+File.separator+dateDirs+File.separator+res); //判断目标文件所在的目录是否存在 if(!newFile.getParentFile().exists()) { //如果目标文件所在的目录不存在，则创建父目录 newFile.getParentFile().mkdirs(); } //将内存中的数据写入磁盘 uploadFile.transferTo(newFile); //完整的url //String fileUrl = "/resource/uploads/"+date.get(Calendar.YEAR)+ "/"+(date.get(Calendar.MONTH)+1)+ "/"+ newFileName; String fileUrl = date.get(Calendar.YEAR)+"/"+(date.get(Calendar.MONTH)+1)+ "/"+res; return fileUrl; } //删除房间信息 @RequestMapping("delRoomById") public String delRoomById(int roomId){ roomService.delRoomById(roomId); return "redirect:/room/selectRoomList"; } //跳转房源修改页面 @RequestMapping("toEditRoomById") public String toEditRoomById(int roomId,Model model){ Room room = roomService.selectRoomByRId(roomId); model.addAttribute("room", room); List<Hotel> hotelList = hotelService.selectHotel(); model.addAttribute("hotelList", hotelList); return "/backstage/updateRoominfo"; } //房源修改 @RequestMapping("editRoom") public String editRoom(Room room1,HttpServletRequest request, MultipartFile uploadFile) throws IOException { if(uploadFile.getSize()!=0) { String newFileName = uploadFile(request, uploadFile); room1.setRoomPicture(newFileName); } System.out.println(room1.toString()); roomService.editRoom(room1); return "redirect:/room/selectRoomList"; } }