kongdeqiang
2022-09-19 a9862e81851bbe037edc6bb1c7f562c1e55c0d7f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
package com.boying.controller;
 
import com.boying.common.BaseController;
import com.boying.common.util.LngLatUtil;
import com.boying.common.util.StringUtil;
import com.boying.entity.*;
import com.boying.service.StreetLogoService;
import com.boying.service.StreetReRoadService;
import com.boying.service.StreetService;
import com.boying.service.ViolationTypeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
 
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
@RestController
@RequestMapping("street")
public class StreetController extends BaseController {
 
    @Autowired
    private StreetService streetService;
    @Autowired
    private StreetReRoadService streetReRoadService;
    @Autowired
    private StreetLogoService streetLogoService;
 
    @PostMapping("findPage")
    public Object findPage(int page,int pageSize,Integer type) {
        Pageable pageable = PageRequest.of(page-1,pageSize, Sort.Direction.DESC,"id");
        Specification<Street> specification = new Specification<Street>() {
            @Override
            public Predicate toPredicate(Root<Street> root, CriteriaQuery<?> cq, CriteriaBuilder cb) {
                List<Predicate> list = new ArrayList<Predicate>();
                if(type!=null){
                    list.add(cb.equal(root.get("type"), type));
                }
 
                Predicate[] arr = new Predicate[list.size()];
                cq.where(list.toArray(arr));
                return null;
            }
        };
        Page<Street> pages = streetService.findPage(pageable,specification);
        return success("",pages);
    }
 
    @PostMapping("save")
    public Object save(Street street) {
        Specification<Street> specification = new Specification<Street>() {
            @Override
            public Predicate toPredicate(Root<Street> root, CriteriaQuery<?> cq, CriteriaBuilder cb) {
                List<Predicate> list = new ArrayList<Predicate>();
                if(!StringUtil.isNullOrEmpty(street.getName())){
                    list.add(cb.equal(root.get("name").as(String.class), street.getName()));
                }
 
                Predicate[] arr = new Predicate[list.size()];
                cq.where(list.toArray(arr));
                return null;
            }
        };
        List all = streetService.findAll(specification);
        if(street.getId()==null){
            if(all.size()>0){
                return error("街道名称重复了");
            }
        }else{
            if(all.size()>1){
                return error("街道名称重复了");
            }
        }
 
        streetService.save(street);
        return success("保存成功");
    }
 
    @PostMapping("save2")
    public Object save2(StreetReRoad streetReRoad) {
        streetReRoadService.save(streetReRoad);
        return success("保存成功");
    }
 
 
    @PostMapping("delete")
    public Object delete(Long id) {
        streetService.delete(id);
        return success("删除成功");
    }
 
    @PostMapping("delete2")
    public Object delete2(Long id) {
        streetReRoadService.delete(id);
        return success("删除成功");
    }
 
    @PostMapping("findById")
    public Object findById(Long id,int type) {
        if(type==0){
            List<StreetReRoad> list = streetService.findById2(id);
            List<Street> list2 = new ArrayList<>();
            for(StreetReRoad streetReRoad:list){
                Street street = (Street) streetService.findById(streetReRoad.getStreetId());
//                streetReRoad.setName(street.getName());
//                streetReRoad.setCode(street.getCode());
//                streetReRoad.setType(street.getType());
                street.setId(streetReRoad.getId());
                list2.add(street);
            }
            return success("",list2);
        }else{
            List<Street> list = streetService.findById3(id);
            return success("",list);
        }
    }
 
    @PostMapping("findList")
    public Object findAll() {
        Map<String,Object> map = new HashMap<>();
        Specification<Street> specification = new Specification<Street>() {
            @Override
            public Predicate toPredicate(Root<Street> root, CriteriaQuery<?> cq, CriteriaBuilder cb) {
                List<Predicate> list = new ArrayList<Predicate>();
                list.add(cb.equal(root.get("type"), 0));
                Predicate[] arr = new Predicate[list.size()];
                cq.where(list.toArray(arr));
                return null;
            }
        };
        map.put("list1",streetService.findAll(specification));
        Specification<Street> specification2 = new Specification<Street>() {
            @Override
            public Predicate toPredicate(Root<Street> root, CriteriaQuery<?> cq, CriteriaBuilder cb) {
                List<Predicate> list = new ArrayList<Predicate>();
                list.add(cb.equal(root.get("type"), 1));
                Predicate[] arr = new Predicate[list.size()];
                cq.where(list.toArray(arr));
                return null;
            }
        };
        map.put("list2",streetService.findAll(specification2));
        return success("",map);
    }
 
    @PostMapping("findLogo")
    public Object findLogo(String lng ,String lat) {
        List<StreetLogo> all = streetLogoService.findAll();
        List<StreetLogo> all2 = new ArrayList<>();
        for(StreetLogo logo:all){
            double distance = LngLatUtil.getDistance(lat, lng, logo.getLat(), logo.getLng());
            if(distance<=200){
                all2.add(logo);
            }
        }
        return success("",all2);
    }
}