package com.moon.server.service.data; import com.alibaba.fastjson.JSON; import com.moon.server.entity.all.StaticData; import com.moon.server.entity.ctrl.PubEntity; import com.moon.server.entity.data.MetaEntity; import com.moon.server.entity.data.MetaPubEntity; import com.moon.server.entity.data.PublishEntity; import com.moon.server.entity.shujian.*; import com.moon.server.helper.HttpHelper; import com.moon.server.helper.RestHelper; import com.moon.server.helper.StringHelper; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.springframework.beans.factory.annotation.Value; import org.springframework.http.HttpMethod; import org.springframework.http.ResponseEntity; import org.springframework.stereotype.Service; import org.springframework.web.client.RestTemplate; import javax.annotation.Resource; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.lang.reflect.Field; import java.util.*; /** * 数简.服务 * @author WWW */ @Service public class ShuJianService { @Value("${shujian.ak}") String ak; @Value("${shujian.url}") String sjUrl; @Resource MetaService metaService; @Resource PublishService publishService; private final static Log log = LogFactory.getLog(ShuJianService.class); public final static List TYPES = new ArrayList<>(Arrays.asList("DOM", "DEM")); /** * 分页查询数简的颜色表 */ public void selectSjColorTables(Integer pageSize, Integer pageIndex, HttpServletRequest req, HttpServletResponse res) { try { String url = String.format("%s/sj_raster/v6/api/color_table?%s&page_count=%d&page_num=%d&key=", sjUrl, ak, pageSize, pageIndex); HttpHelper httpHelper = new HttpHelper(); httpHelper.service(req, res, null, url); } catch (Exception ex) { log.error(ex.getMessage(), ex); } } /** * 插入数简服务 */ public Integer insertSjService(PubEntity pub) { CreateLayerEntity layerEntity = createLayerEntity(pub); Integer layerId = postLayerEntity(layerEntity); if (null == layerId) { return 0; } CreateServiceEntity serviceEntity = createServiceEntity(pub); Integer serviceId = postServiceEntity(serviceEntity); if (null == serviceId) { return 0; } PublishEntity publishEntity = createPublishEntity(pub, layerId, serviceId); Integer id = publishService.insert(publishEntity); if (null == id) { return id; } return insertMetaPub(pub, id); } /** * 创建图层实体类 */ public CreateLayerEntity createLayerEntity(PubEntity pub) { String epsg = getEpsg(pub); String ids = StringHelper.join(pub.getIds(), ","); List list = metaService.selectByIds(ids); CreateLayerEntity entity = new CreateLayerEntity(); entity.setCreate_service(0); CreateLayerEntity.Layer layer = entity.getLayer(); layer.setName(pub.getName()); layer.setProjection(epsg); layer.setType(TYPES.indexOf(pub.getType())); layer.setEnhance_type(pub.getEnhanceType()); layer.setNodata(getNoData(pub, list)); CreateLayerEntity.ImagePathList imagePathList = entity.getImage_path_list().get(0); imagePathList.setPath(""); List imageLists = entity.getImage_list(); CreateLayerEntity.ImageList imageList = imageLists.get(0); imageList.setPath(""); return entity; } private List getNoData(PubEntity pub, List metas) { int bands = 1; for (MetaEntity meta : metas) { if (StringHelper.isEmpty(meta.getBands())) { continue; } try { int b = Integer.parseInt(meta.getBands()); if (b > bands) { bands = b; } } catch (Exception e) { // } } List list = new ArrayList<>(); for (int i = 0; i < bands; i++) { list.add(String.valueOf(pub.getNodata())); } return list; } /** * 发送图层实体类 */ public Integer postLayerEntity(CreateLayerEntity layerEntity) { Map map = getMapData(layerEntity); String url = String.format("%s/sj_raster/sj_raster/v6/api/layer?%s", sjUrl, ak); LayerResultEntity rs = postForRest(url, map, LayerResultEntity.class); boolean flag = null == rs || !StaticData.SUCCESS.equals(rs.getStatus()) || null == rs.getResult().getLayer_id(); return flag ? null : rs.getResult().getLayer_id(); } /** * 创建服务实体类 */ public CreateServiceEntity createServiceEntity(PubEntity pubEntity) { CreateServiceEntity entity = new CreateServiceEntity(); List layerList = entity.getLayer_list(); CreateServiceEntity.Services services = entity.getService(); services.setName(""); services.setAccess_name(StringHelper.getGuid()); services.setPub_status(1); services.setRect_minx(-180.0); services.setRect_miny(-90.0); services.setRect_maxx(180.0); services.setRect_maxy(90.0); services.setProjection("ESRI:" + StaticData.I104903); services.setSat_id(-1); services.setSensor_id(-1); services.setImage_type(-1); return entity; } /** * 发送图层实体类 */ public Integer postServiceEntity(CreateServiceEntity serviceEntity) { Map map = getMapData(serviceEntity); String url = String.format("%s/sj_raster/v6/api/service?op=create_ex&%s", sjUrl, ak); ServiceResultEntity rs = postForRest(url, map, ServiceResultEntity.class); boolean flag = null == rs || !StaticData.SUCCESS.equals(rs.getStatus()) || null == rs.getResult(); return flag ? null : rs.getResult(); } /** * 创建发布实体类 */ public PublishEntity createPublishEntity(PubEntity pub, Integer layerId, Integer serviceId) { PublishEntity entity = new PublishEntity(); entity.setType(pub.getType()); entity.setName(pub.getName()); entity.setMin(pub.getMin()); entity.setMax(pub.getMax()); entity.setBak(getEpsg(pub)); entity.setCreateUser(pub.getUserId()); entity.setDirid(pub.getDircode()); entity.setDepid(pub.getDepcode()); entity.setStatus(3); entity.setLayerid(layerId); entity.setServiceid(serviceId); entity.setJson(getColorTableJson(pub)); return entity; } /** * 获取EPSG编码 */ public String getEpsg(PubEntity pub) { if (StaticData.I104903 == pub.getEpsgCode()) { return "ESRI:" + pub.getEpsgCode(); } return "EPSG:" + pub.getEpsgCode(); } /** * 获取颜色表的JSON字符串 */ public String getColorTableJson(PubEntity pub) { if (null == pub.getColorTable() && null == pub.getGradientColorTable()) { return null; } if (null != pub.getGradientColorTable()) { return JSON.toJSONString(pub.getGradientColorTable()); } return JSON.toJSONString(pub.getColorTable()); } /** * 插入元数据发布类 */ private int insertMetaPub(PubEntity pub, Integer pubid) { int count = 0; for (Integer metaid : pub.getIds()) { MetaPubEntity mp = new MetaPubEntity(); mp.setMetaid(metaid); mp.setPubid(pubid); mp.setCreateUser(pub.getUserId()); count += publishService.insertMetaPub(mp); } return count; } /** * 插入数简服务 */ public Integer updateSjService(PubEntity pub, HttpServletRequest req, HttpServletResponse res) { PublishEntity entity = publishService.selectById(pub.getPubid()); if (null == entity) { return 0; } if (null != entity.getLayerid()) { deleteLayer(entity.getLayerid()); } if (null != entity.getServiceid()) { deleteService(entity.getServiceid()); } return 1; } /** * 删除数简服务 */ public Integer deletesSjServices(List ids) { String strs = StringHelper.join(ids, ","); List list = publishService.selectByIds(strs); if (null == list || list.isEmpty()) { return 0; } for (PublishEntity entity : list) { if (null != entity.getLayerid()) { deleteLayer(entity.getLayerid()); } if (null != entity.getServiceid()) { deleteService(entity.getServiceid()); } } return publishService.deletes(strs); } /** * 删除图层 */ public boolean deleteLayer(Integer id) { try { DeleteEntity entity = new DeleteEntity(); entity.getId_list().add(id); Map map = getMapData(entity); String url = String.format("%s/sj_raster/sj_raster/v6/api/layer?%s", sjUrl, ak); DeleteResultEntity rs = deleteForRest(url, map); return null != rs && StaticData.SUCCESS.equals(rs.getStatus()); } catch (Exception ex) { log.error(ex.getMessage(), ex); return false; } } /** * 删除服务 */ public boolean deleteService(Integer id) { try { DeleteEntity entity = new DeleteEntity(); entity.getId_list().add(id); Map map = getMapData(entity); String url = String.format("%s/sj_raster/sj_raster/v6/api/service?%s", sjUrl, ak); DeleteResultEntity rs = deleteForRest(url, map); return null != rs && StaticData.SUCCESS.equals(rs.getStatus()); } catch (Exception ex) { log.error(ex.getMessage(), ex); return false; } } /** * post请求(Rest) */ public T postForRest(String url, Map map, Class clazz) { RestTemplate rest = RestHelper.getRestTemplate(); return rest.postForObject(url, map, clazz); } /** * delete请求(Rest) */ public DeleteResultEntity deleteForRest(String url, Map map) { RestTemplate rest = RestHelper.getRestTemplate(); ResponseEntity rs = rest.exchange(url, HttpMethod.DELETE, null, DeleteResultEntity.class, map); return rs.getBody(); } /** * 获取Map数据 */ public Map getMapData(T t) { Map map = new HashMap<>(1); Field[] fields = t.getClass().getDeclaredFields(); for (Field field : fields) { try { if ("serialVersionUID".equals(field.getName())) { continue; } field.setAccessible(true); Object obj = field.get(t); map.put(field.getName(), obj); } catch (Exception ex) { // } } return map; } }