package com.bosshand.virgo.api.controller; import com.bosshand.virgo.api.model.ProjectItemTargetRoom; import com.bosshand.virgo.api.model.Tag; import com.bosshand.virgo.api.service.ProjectItemTargetRoomService; import com.bosshand.virgo.core.response.Response; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; @RestController @RequestMapping("projectItemTargetRoom") @Api(tags = {"房源管理"}) public class ProjectItemTargetRoomController { @Autowired ProjectItemTargetRoomService projectItemTargetRoomService; @ApiOperation("获取") @RequestMapping(value = "/{currPage}/{pageSize}", method = RequestMethod.POST) public Response list(@RequestBody ProjectItemTargetRoom projectItemTargetRoom, @PathVariable int currPage, @PathVariable int pageSize) { int totalCount = projectItemTargetRoomService.getTotalCount(projectItemTargetRoom); List dataList = projectItemTargetRoomService.getLimit(projectItemTargetRoom, currPage, pageSize); List tags = new ArrayList<>(); for (ProjectItemTargetRoom room : dataList) { tags.clear(); if (room.getTagIds() != null) { for (String tagId : room.getTagIds().split(",")) { tags.add(projectItemTargetRoomService.getTag(Long.parseLong(tagId))); } } room.setTagList(tags); } Map result = new HashMap<>(); result.put("dataList", dataList); result.put("totalCount", totalCount); return Response.ok(result); } @ApiOperation("新增") @RequestMapping(value = "", method = RequestMethod.POST) public Response insert(@RequestBody ProjectItemTargetRoom projectItemTargetRoom) { projectItemTargetRoomService.insert(projectItemTargetRoom); return Response.ok(); } @ApiOperation("删除") @RequestMapping(value = "/{id}", method = RequestMethod.DELETE) public Response delete(@PathVariable long id) { projectItemTargetRoomService.delete(id); return Response.ok(); } @ApiOperation("修改") @RequestMapping(value = "", method = RequestMethod.PUT) public Response update(@RequestBody ProjectItemTargetRoom projectItemTargetRoom) { projectItemTargetRoomService.update(projectItemTargetRoom); return Response.ok(); } @ApiOperation("详情") @RequestMapping(value = "/{id}", method = RequestMethod.GET) public Response get(@PathVariable long id) { ProjectItemTargetRoom room = projectItemTargetRoomService.get(id); String tagIds = room.getTagIds(); List tags = new ArrayList<>(); if (tagIds != null) { for (String tagId : tagIds.split(",")) { tags.add(projectItemTargetRoomService.getTag(Long.parseLong(tagId))); } } room.setTagList(tags); return Response.ok(room); } }