1

13693261870

2022-11-01 7e6fd5f1a403618d9d2116c483783951e17a3509

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
package com.lf.server.service.show;
 
import com.lf.server.entity.ctrl.MarkJsonEntity;
import com.lf.server.entity.show.MarkEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.*;
import com.lf.server.mapper.show.MarkMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
 
/**
 * 标绘
 * @author WWW
 */
@Service
public class MarkService implements MarkMapper {
    @Autowired
    MarkMapper markMapper;
 
    @Autowired
    PathHelper pathHelper;
 
    @Override
    public Integer selectCount(Integer uid) {
        return markMapper.selectCount(uid);
    }
 
    @Override
    public List<MarkEntity> selectByPage(Integer uid, Integer limit, Integer offset) {
        return markMapper.selectByPage(uid, limit, offset);
    }
 
    @Override
    public List<MarkEntity> selectAll() {
        return markMapper.selectAll();
    }
 
    @Override
    public MarkEntity selectById(int id) {
        return markMapper.selectById(id);
    }
 
    @Override
    public Integer insert(MarkEntity entity) {
        return markMapper.insert(entity);
    }
 
    @Override
    public Integer inserts(List<MarkEntity> list) {
        return markMapper.inserts(list);
    }
 
    @Override
    public Integer delete(int id) {
        return markMapper.delete(id);
    }
 
    @Override
    public Integer deletes(List<Integer> ids) {
        return markMapper.deletes(ids);
    }
 
    @Override
    public Integer update(MarkEntity entity) {
        return markMapper.update(entity);
    }
 
    @Override
    public Integer updates(List<MarkEntity> list) {
        return markMapper.updates(list);
    }
 
    public String downloadShp(UserEntity ue, List<MarkJsonEntity> list, HttpServletRequest req, HttpServletResponse res) throws Exception {
        String parent = pathHelper.getTempPath(ue.getId());
        String path = getShpDir(ue, parent);
 
        List<MarkJsonEntity> points = getMarkByType(list, "POINT");
        if (points.size() > 0) {
            String pointFile = GdalHelper.createShp(points, path, "POINT");
        }
        List<MarkJsonEntity> lines = getMarkByType(list, "LINESTRING");
        if (lines.size() > 0) {
            String lineFile = GdalHelper.createShp(lines, path, "LINESTRING");
        }
        List<MarkJsonEntity> polygons = getMarkByType(list, "POLYGON");
        if (polygons.size() > 0) {
            String polygonFile = GdalHelper.createShp(polygons, path, "POLYGON");
        }
 
        File[] files = new File(path).listFiles();
        if (files == null || files.length == 0) {
            return "Shp文件生成失败";
        }
 
        String zip = getZip();
        ZipHelper.zip(zip, parent);
        //download(res, zip);
        FileHelper.deleteDir(parent);
 
        return "";
    }
 
    private String getShpDir(UserEntity ue, String parent) {
        String path = parent + File.separator + WebHelper.getRandomInt(100000, 1000000);
 
        File file = new File(path);
        if (!file.exists() && !file.isDirectory()) {
            file.mkdirs();
        }
 
        return path;
    }
 
    private List<MarkJsonEntity> getMarkByType(List<MarkJsonEntity> list, String type) {
        List<MarkJsonEntity> rs = new ArrayList<MarkJsonEntity>();
        for (MarkJsonEntity mark : list) {
            if (StringHelper.isEmpty(mark.getWkt())) {
                continue;
            }
            if (mark.getWkt().indexOf(type) > -1) {
                rs.add(mark);
            }
        }
 
        return rs;
    }
 
    private String getZip() {
        String path = pathHelper.getDownloadFullPath() + File.separator + WebHelper.getRandomInt(100000, 1000000) + ".zip";
 
        File file = new File(path);
        if (file.exists() && !file.isDirectory()) {
            file.delete();
        }
 
        return path;
    }
 
 
}