FarmServiceImpl.java 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. package com.huimv.admin.service.impl;
  2. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
  3. import com.baomidou.mybatisplus.core.metadata.IPage;
  4. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  5. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  6. import com.huimv.admin.common.utils.Result;
  7. import com.huimv.admin.common.utils.ResultCode;
  8. import com.huimv.admin.entity.Farm;
  9. import com.huimv.admin.entity.SysAccountMultilevel;
  10. import com.huimv.admin.mapper.FarmMapper;
  11. import com.huimv.admin.mapper.SysAccountMultilevelMapper;
  12. import com.huimv.admin.service.IFarmService;
  13. import org.apache.commons.lang.StringUtils;
  14. import org.springframework.beans.factory.annotation.Autowired;
  15. import org.springframework.stereotype.Service;
  16. import java.util.Arrays;
  17. import java.util.List;
  18. import java.util.Map;
  19. /**
  20. * <p>
  21. * 服务实现类
  22. * </p>
  23. *
  24. * @author astupidcoder
  25. * @since 2021-10-14
  26. */
  27. @Service
  28. public class FarmServiceImpl extends ServiceImpl<FarmMapper, Farm> implements IFarmService {
  29. @Autowired
  30. private SysAccountMultilevelMapper accountMultilevelMapper;
  31. @Override
  32. public Result getFarm(Integer userId) {
  33. SysAccountMultilevel sysAccountMultilevel = accountMultilevelMapper.selectById(userId);
  34. String farmIds = sysAccountMultilevel.getFarmIds();
  35. List list ;
  36. if (userId == 1){
  37. list = this.list();
  38. }else {
  39. List<String> list1 = Arrays.asList(farmIds.split(","));
  40. list = (List)this.listByIds(list1);
  41. }
  42. return new Result(ResultCode.SUCCESS,list);
  43. }
  44. @Override
  45. public Result add(Farm farm) {
  46. farm.setLocation(farm.getFrontLocation());
  47. this.save(farm);
  48. return new Result(10000,"添加成功",true);
  49. }
  50. @Override
  51. public Result getFarm2(Integer userId) {
  52. SysAccountMultilevel sysAccountMultilevel = accountMultilevelMapper.selectById(userId);
  53. String farmIds = sysAccountMultilevel.getFarmIds();
  54. List<Farm> list ;
  55. if (userId == 1){
  56. list = this.list();
  57. }else {
  58. List<String> list1 = Arrays.asList(farmIds.split(","));
  59. list = this.listByIds(list1);
  60. }
  61. String str = "A,B,C,D,E,F,G,H,I,J,K,L,M,N,O,P,Q";
  62. String[] split = str.split(",");
  63. Integer i = 0;
  64. for (Farm farm : list) {
  65. farm.setFarmName(split[i]+"牧场");
  66. i++;
  67. }
  68. return new Result(ResultCode.SUCCESS,list);
  69. }
  70. @Override
  71. public Result listPage(Integer userId, Map paramMap) {
  72. int pageNum = (Integer)paramMap.get("pageNum");
  73. int pageSize = (Integer)paramMap.get("pageSize");
  74. String farmName = (String)paramMap.get("farmName");
  75. QueryWrapper<Farm> wrapper= new QueryWrapper<>();
  76. wrapper.like("farm_name",farmName);
  77. SysAccountMultilevel sysAccountMultilevel = accountMultilevelMapper.selectById(userId);
  78. String farmIds = sysAccountMultilevel.getFarmIds();
  79. if (StringUtils.isBlank(farmIds)){
  80. return new Result(10001,"未绑定牧场",false);
  81. }
  82. if (userId != 1){
  83. List<String> list1 = Arrays.asList(farmIds.split(","));
  84. wrapper.in("id",list1);
  85. }
  86. IPage<Farm> page = this.page(new Page<>(pageNum, pageSize), wrapper);
  87. return new Result(ResultCode.SUCCESS,page);
  88. }
  89. }