day05-路线规划之Neo4j入门

shuhongfan
1 parent 8f018e3b
Showing 22 changed files with 705 additions and 1 deletions
.idea/.gitignore → .gitignore
sl-express-sdn/pom.xml
sl-express-sdn/src/main/java/com/sl/sdn/SDNApplication.java
sl-express-sdn/src/main/java/com/sl/sdn/dto/OrganDTO.java
sl-express-sdn/src/main/java/com/sl/sdn/dto/TransportLineNodeDTO.java
sl-express-sdn/src/main/java/com/sl/sdn/entity/line/TransportLine.java
sl-express-sdn/src/main/java/com/sl/sdn/entity/node/AgencyEntity.java
sl-express-sdn/src/main/java/com/sl/sdn/entity/node/BaseEntity.java
sl-express-sdn/src/main/java/com/sl/sdn/entity/node/OLTEntity.java
sl-express-sdn/src/main/java/com/sl/sdn/entity/node/TLTEntity.java
sl-express-sdn/src/main/java/com/sl/sdn/enums/OrganTypeEnum.java
sl-express-sdn/src/main/java/com/sl/sdn/repository/AgencyRepository.java
sl-express-sdn/src/main/java/com/sl/sdn/repository/OLTRepository.java
sl-express-sdn/src/main/java/com/sl/sdn/repository/OrganRepository.java
sl-express-sdn/src/main/java/com/sl/sdn/repository/TLTRepository.java
sl-express-sdn/src/main/java/com/sl/sdn/repository/TransportLineRepository.java
sl-express-sdn/src/main/java/com/sl/sdn/repository/impl/OrganRepositoryImpl.java
sl-express-sdn/src/main/java/com/sl/sdn/repository/impl/TransportLineRepositoryImpl.java
sl-express-sdn/src/main/resources/application.yml
sl-express-sdn/src/test/java/com/sl/sdn/repository/AgencyRepositoryTest.java
 # 默认忽略的文件
 /shelf/
-/workspace.xml
+/.idea/workspace.xml
 # 基于编辑器的 HTTP 客户端请求
 /httpRequests/
 # Datasource local storage ignored files
@@ -8,3 +8,11 @@
 /dataSources.local.xml
 # Zeppelin 忽略的文件
 /ZeppelinRemoteNotebooks/
+### Example user template template
+### Example user template
+
+# IntelliJ project files
+.idea
+*.iml
+out
+gen
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>com.sl-express</groupId>
+        <artifactId>sl-express-parent</artifactId>
+        <version>1.4</version>
+    </parent>
+
+    <groupId>com.sl-express.sdn</groupId>
+    <artifactId>sl-express-sdn</artifactId>
+    <version>1.0-SNAPSHOT</version>
+
+    <properties>
+        <maven.compiler.source>11</maven.compiler.source>
+        <maven.compiler.target>11</maven.compiler.target>
+        <sl-express-common.version>1.2-SNAPSHOT</sl-express-common.version>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>com.sl-express.common</groupId>
+            <artifactId>sl-express-common</artifactId>
+            <version>${sl-express-common.version}</version>
+        </dependency>
+        <!--SDN依赖-->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-neo4j</artifactId>
+        </dependency>
+    </dependencies>
+
+</project>
 \ No newline at end of file
+package com.sl.sdn;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+@SpringBootApplication
+public class SDNApplication {
+
+    public static void main(String[] args) {
+        SpringApplication.run(SDNApplication.class, args);
+    }
+}
 \ No newline at end of file
+package com.sl.sdn.dto;
+
+import cn.hutool.core.annotation.Alias;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+/**
+ * 机构数据对象，网点、一级转运、二级转运都是看作是机构
+ * BaseEntity中的location无法序列化，需要将经纬度拆开封装对象
+ */
+@Data
+public class OrganDTO {
+
+    @Alias("bid") //业务id作为id进行封装
+    @ApiModelProperty(value = "机构id", required = true)
+    private Long id;
+    @ApiModelProperty(value = "名称", required = true)
+    private String name;
+    @ApiModelProperty(value = "类型，1:一级转运，2：二级转运，3:网点", required = true)
+    private Integer type;
+    @ApiModelProperty(value = "电话", required = true)
+    private String phone;
+    @ApiModelProperty(value = "地址", required = true)
+    private String address;
+    @ApiModelProperty(value = "纬度", required = true)
+    private Double latitude;
+    @ApiModelProperty(value = "经度", required = true)
+    private Double longitude;
+
+}
+package com.sl.sdn.dto;
+
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * 运输路线对象
+ */
+@Data
+public class TransportLineNodeDTO {
+
+    @ApiModelProperty(value = "节点列表", required = true)
+    private List<OrganDTO> nodeList = new ArrayList<>();
+
+    @ApiModelProperty(value = "路线成本", required = true)
+    private Double cost = 0d;
+
+}
+package com.sl.sdn.entity.line;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+/**
+ * 运输路线实体
+ */
+@Data
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class TransportLine {
+
+    private Long id;
+    private Double cost; //成本
+
+}
+package com.sl.sdn.entity.node;
+
+
+import com.sl.sdn.enums.OrganTypeEnum;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.ToString;
+import lombok.experimental.SuperBuilder;
+import org.springframework.data.neo4j.core.schema.Node;
+
+/**
+ * 网点实体
+ */
+@Node("AGENCY")
+@Data
+@ToString(callSuper = true)
+@SuperBuilder(toBuilder = true)
+@NoArgsConstructor
+public class AgencyEntity extends BaseEntity {
+
+    @Override
+    public OrganTypeEnum getAgencyType() {
+        return OrganTypeEnum.AGENCY;
+    }
+}
+package com.sl.sdn.entity.node;
+
+import com.sl.sdn.enums.OrganTypeEnum;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+import org.neo4j.driver.types.Point;
+import org.springframework.data.neo4j.core.schema.GeneratedValue;
+import org.springframework.data.neo4j.core.schema.Id;
+
+@Data
+@SuperBuilder(toBuilder = true)
+@NoArgsConstructor
+@AllArgsConstructor
+public abstract class BaseEntity {
+
+    @Id
+    @GeneratedValue
+    @ApiModelProperty(value = "Neo4j ID", hidden = true)
+    private Long id;
+    @ApiModelProperty(value = "业务id", required = true)
+    private Long bid;
+    @ApiModelProperty(value = "名称", required = true)
+    private String name;
+    @ApiModelProperty(value = "电话", required = true)
+    private String phone;
+    @ApiModelProperty(value = "地址", required = true)
+    private String address;
+    @ApiModelProperty(value = "位置坐标, x: 纬度，y: 经度", required = true)
+    private Point location;
+
+    //机构类型
+    public abstract OrganTypeEnum getAgencyType();
+
+}
 \ No newline at end of file
+package com.sl.sdn.entity.node;
+
+import com.sl.sdn.enums.OrganTypeEnum;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.ToString;
+import lombok.experimental.SuperBuilder;
+import org.springframework.data.neo4j.core.schema.Node;
+
+/**
+ * 一级转运中心实体 (OneLevelTransportEntity)
+ */
+@Node("OLT")
+@Data
+@ToString(callSuper = true)
+@SuperBuilder(toBuilder = true)
+@NoArgsConstructor
+public class OLTEntity extends BaseEntity {
+
+    @Override
+    public OrganTypeEnum getAgencyType() {
+        return OrganTypeEnum.OLT;
+    }
+}
+package com.sl.sdn.entity.node;
+
+import com.sl.sdn.enums.OrganTypeEnum;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.ToString;
+import lombok.experimental.SuperBuilder;
+import org.springframework.data.neo4j.core.schema.Node;
+
+/**
+ * 二级转运中心实体(TwoLevelTransportEntity)
+ */
+@Node("TLT")
+@Data
+@ToString(callSuper = true)
+@SuperBuilder(toBuilder = true)
+@NoArgsConstructor
+public class TLTEntity extends BaseEntity {
+
+    @Override
+    public OrganTypeEnum getAgencyType() {
+        return OrganTypeEnum.TLT;
+    }
+}
 \ No newline at end of file
+package com.sl.sdn.enums;
+
+import cn.hutool.core.util.EnumUtil;
+import com.sl.transport.common.enums.BaseEnum;
+
+/**
+ * 机构类型枚举
+ */
+public enum OrganTypeEnum implements BaseEnum {
+
+    OLT(1, "一级转运中心"),
+    TLT(2, "二级转运中心"),
+    AGENCY(3, "网点");
+
+    /**
+     * 类型编码
+     */
+    private final Integer code;
+
+    /**
+     * 类型值
+     */
+    private final String value;
+
+    OrganTypeEnum(Integer code, String value) {
+        this.code = code;
+        this.value = value;
+    }
+
+    public Integer getCode() {
+        return code;
+    }
+
+    public String getValue() {
+        return value;
+    }
+
+    public static OrganTypeEnum codeOf(Integer code) {
+        return EnumUtil.getBy(OrganTypeEnum::getCode, code);
+    }
+}
+package com.sl.sdn.repository;
+
+import com.sl.sdn.entity.node.AgencyEntity;
+import org.springframework.data.neo4j.repository.Neo4jRepository;
+import org.springframework.stereotype.Repository;
+
+/**
+ * 网点操作
+ */
+@Repository
+public interface AgencyRepository extends Neo4jRepository<AgencyEntity,Long> {
+    /**
+     * 根据bid查询
+     * @param bid
+     * @return
+     */
+    AgencyEntity findByBid(Long bid);
+
+    /**
+     * 根据bid删除
+     * @param bid
+     * @return
+     */
+    Long deleteByBid(Long bid);
+
+}
+package com.sl.sdn.repository;
+
+import com.sl.sdn.entity.node.OLTEntity;
+import org.springframework.data.neo4j.repository.Neo4jRepository;
+import org.springframework.stereotype.Repository;
+
+/**
+ * 一级转运中心数据操作
+ */
+@Repository
+public interface OLTRepository extends Neo4jRepository<OLTEntity, Long> {
+
+    /**
+     * 根据bid查询
+     * @param bid
+     * @return
+     */
+    OLTEntity findByBid(Long bid);
+
+    /**
+     * 根据bid删除
+     * @param bid
+     * @return
+     */
+    Long deleteByBid(Long bid);
+}
+package com.sl.sdn.repository;
+
+import com.sl.sdn.dto.OrganDTO;
+
+import java.util.List;
+
+/**
+ * 通过机构查询
+ */
+public interface OrganRepository {
+
+    /**
+     * 无需指定type，根据id查询
+     * @param bid
+     * @return
+     */
+    OrganDTO findByBid(Long bid);
+
+
+    /**
+     * 查询所有的机构，如果name不为空的按照name模糊查询
+     * @param name
+     * @return
+     */
+    List<OrganDTO> findAll(String name);
+}
+package com.sl.sdn.repository;
+
+import com.sl.sdn.entity.node.TLTEntity;
+import org.springframework.data.neo4j.repository.Neo4jRepository;
+import org.springframework.stereotype.Repository;
+
+/**
+ * 二级转运中心数据操作
+ */
+@Repository
+public interface TLTRepository extends Neo4jRepository<TLTEntity,Long> {
+
+    /**
+     * 根据Bid查询
+     * @param bid
+     * @return
+     */
+    TLTEntity findByBid(Long bid);
+
+
+    /**
+     * 根据bid删除
+     * @param bid
+     * @return
+     */
+    Long deleteByBid(Long bid);
+
+}
+package com.sl.sdn.repository;
+
+import com.sl.sdn.dto.TransportLineNodeDTO;
+import com.sl.sdn.entity.node.AgencyEntity;
+
+/**
+ * 运输路线相关操作
+ */
+public interface TransportLineRepository {
+
+    /**
+     * 查询两个网点之间最短的路线，查询深度为：10
+     * @param start
+     * @param end
+     * @return
+     */
+    TransportLineNodeDTO findShortTestPath(AgencyEntity start, AgencyEntity end);
+}
+package com.sl.sdn.repository.impl;
+
+import cn.hutool.core.collection.CollUtil;
+import com.sl.sdn.dto.OrganDTO;
+import com.sl.sdn.enums.OrganTypeEnum;
+import com.sl.sdn.repository.OrganRepository;
+import com.sl.transport.common.util.BeanUtil;
+import org.neo4j.driver.internal.value.NodeValue;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.neo4j.core.Neo4jClient;
+import org.springframework.stereotype.Repository;
+
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+
+/**
+ * 通过机构查询
+ */
+@Repository
+public class OrganRepositoryImpl implements OrganRepository {
+
+    @Autowired
+    private Neo4jClient neo4jClient;
+
+    /**
+     * 无需指定type，根据id查询
+     * @param bid
+     * @return
+     */
+    @Override
+    public OrganDTO findByBid(Long bid) {
+        String cypherQuery = "MATCH(n) where n.bid=$bid RETURN n";
+
+        Optional<OrganDTO> organDTOOptional = neo4jClient.query(cypherQuery)
+                .bind(bid).to("bid")
+                .fetchAs(OrganDTO.class)
+                .mappedBy(((typeSystem, record) -> {
+                    NodeValue nodeValue = (NodeValue) record.get(0);
+                    Map<String, Object> map = nodeValue.asMap();
+                    OrganDTO organDTO = BeanUtil.toBeanIgnoreError(map, OrganDTO.class);
+                    organDTO.setType(OrganTypeEnum.valueOf(CollUtil.getFirst(nodeValue.asNode().labels())).getCode());
+                    organDTO.setLatitude(BeanUtil.getProperty(map.get("location"), "y"));
+                    organDTO.setLongitude(BeanUtil.getProperty(map.get("location"), "x"));
+                    return organDTO;
+                })).one();
+        return organDTOOptional.orElse(null);
+    }
+
+    /**
+     * 查询所有的机构，如果name不为空的按照name模糊查询
+     * @param name
+     * @return
+     */
+    @Override
+    public List<OrganDTO> findAll(String name) {
+        String cypherQuery = "MATCH(n) where n.name CONTAINS $name RETURN n";
+
+        Collection<OrganDTO> organDTOCollection = neo4jClient.query(cypherQuery)
+                .bind(name).to("name")
+                .fetchAs(OrganDTO.class)
+                .mappedBy(((typeSystem, record) -> {
+                    NodeValue nodeValue = (NodeValue) record.get(0);
+                    Map<String, Object> map = nodeValue.asMap();
+                    OrganDTO organDTO = BeanUtil.toBeanIgnoreError(map, OrganDTO.class);
+                    organDTO.setType(OrganTypeEnum.valueOf(CollUtil.getFirst(nodeValue.asNode().labels())).getCode());
+                    organDTO.setLatitude(BeanUtil.getProperty(map.get("location"), "y"));
+                    organDTO.setLongitude(BeanUtil.getProperty(map.get("location"), "x"));
+                    return organDTO;
+                })).all();
+        return (List<OrganDTO>) organDTOCollection;
+    }
+}
+package com.sl.sdn.repository.impl;
+
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.NumberUtil;
+import cn.hutool.core.util.StrUtil;
+import com.sl.sdn.dto.OrganDTO;
+import com.sl.sdn.dto.TransportLineNodeDTO;
+import com.sl.sdn.entity.node.AgencyEntity;
+import com.sl.sdn.enums.OrganTypeEnum;
+import com.sl.sdn.repository.TransportLineRepository;
+import com.sl.transport.common.util.BeanUtil;
+import org.neo4j.driver.internal.value.PathValue;
+import org.neo4j.driver.types.Path;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.neo4j.core.Neo4jClient;
+import org.springframework.data.neo4j.core.schema.Node;
+import org.springframework.stereotype.Repository;
+
+import java.util.Map;
+import java.util.Optional;
+
+/**
+ * 运输路线相关操作
+ */
+@Repository
+public class TransportLineRepositoryImpl implements TransportLineRepository {
+
+    @Autowired
+    private Neo4jClient neo4jClient;
+
+
+    /**
+     * 查询两个网点之间最短的路线，查询深度为：10
+     * @param start
+     * @param end
+     * @return
+     */
+    @Override
+    public TransportLineNodeDTO findShortTestPath(AgencyEntity start, AgencyEntity end) {
+//        获取网点数据在Neo4j中的类型
+        String type = AgencyEntity.class.getAnnotation(Node.class).value()[0];
+
+//        构造查询语句
+        String cypherQuery = StrUtil.format("MATCH path = shortestPath((start:{}) -[*..10]-> (end:{}))\n" +
+                "WHERE start.bid = $startId AND end.bid = $endId \n" +
+                "RETURN path", type, type);
+
+//        执行查询语句
+        Optional<TransportLineNodeDTO> transportLineNodeDTOOptional = neo4jClient.query(cypherQuery)
+                .bind(start.getBid()).to("startId") // 设置参数
+                .bind(end.getBid()).to("endId") // 设置参数
+                .fetchAs(TransportLineNodeDTO.class) // 设置想要数据类型
+                .mappedBy(((typeSystem, record) -> { // 对响应结果进行处理
+                    PathValue pathValue = (PathValue) record.get(0);
+                    Path path = pathValue.asPath();
+                    TransportLineNodeDTO transportLineNodeDTO = new TransportLineNodeDTO();
+
+//                    读取节点数据
+                    path.nodes().forEach(node -> {
+                        Map<String, Object> map = node.asMap();
+                        OrganDTO organDTO = BeanUtil.toBeanIgnoreError(map, OrganDTO.class);
+
+//                        读取第一个标签作为类型
+                        organDTO.setType(OrganTypeEnum.valueOf(CollUtil.getFirst(node.labels())).getCode());
+
+//                        查询处理的数据源，x:经度，y：纬度
+                        organDTO.setLatitude(BeanUtil.getProperty(map.get("location"), "y"));
+                        organDTO.setLongitude(BeanUtil.getProperty(map.get("location"), "x"));
+                        transportLineNodeDTO.getNodeList().add(organDTO);
+                    });
+
+//                    取2位小数
+                    transportLineNodeDTO.setCost(NumberUtil.round(transportLineNodeDTO.getCost(), 2).doubleValue());
+                    return transportLineNodeDTO;
+
+                })).one();
+
+
+        return transportLineNodeDTOOptional.orElse(null);
+    }
+
+}
+server:
+  port: 9902
+logging:
+  level:
+    org.springframework.data.neo4j: debug
+spring:
+  application:
+    name: sl-express-sdn
+  mvc:
+    pathmatch:
+      #解决异常：swagger Failed to start bean 'documentationPluginsBootstrapper'; nested exception is java.lang.NullPointerException
+      #因为Springfox使用的路径匹配是基于AntPathMatcher的，而Spring Boot 2.6.X使用的是PathPatternMatcher
+      matching-strategy: ant_path_matcher
+  data:
+    neo4j:
+      database: neo4j
+  neo4j:
+    authentication:
+      username: neo4j
+      password: neo4j123
+    uri: neo4j://192.168.150.101:7687
 \ No newline at end of file
+package com.sl.sdn.repository;
+
+import com.sl.sdn.entity.node.AgencyEntity;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageRequest;
+import org.springframework.data.domain.Sort;
+
+import java.util.List;
+
+@SpringBootTest
+class AgencyRepositoryTest {
+
+    @Autowired
+    private AgencyRepository agencyRepository;
+
+    @Test
+    void findByBid() {
+        AgencyEntity agencyEntity = agencyRepository.findByBid(9001L);
+        System.out.println(agencyEntity);
+    }
+
+    @Test
+    void deleteByBid() {
+        AgencyEntity agencyEntity = new AgencyEntity();
+        agencyEntity.setBid(9001L);
+        agencyEntity.setName("测试节点");
+        agencyEntity.setPhone("122222222222");
+        agencyEntity.setAddress("测试数据地址");
+
+        agencyRepository.save(agencyEntity);
+        System.out.println(agencyEntity);
+
+    }
+
+    @Test
+    public void testUpdate() {
+        AgencyEntity agencyEntity = agencyRepository.findByBid(9001L);
+        agencyEntity.setName("测试节点1");
+
+        agencyRepository.save(agencyEntity);
+        System.out.println(agencyEntity);
+
+    }
+
+    @Test
+    public void testDeleteByBid() {
+        Long count = agencyRepository.deleteByBid(9001L);
+        System.out.println(count);
+    }
+
+
+    @Test
+    public void testFindAll() {
+        List<AgencyEntity> list = agencyRepository.findAll();
+        for (AgencyEntity agencyEntity : list) {
+            System.out.println(agencyEntity);
+        }
+    }
+
+
+    @Test
+    public void testPage() {
+        PageRequest pageRequest = PageRequest.of(1, 2, Sort.by(Sort.Direction.DESC, "bid"));
+        Page<AgencyEntity> page = agencyRepository.findAll(pageRequest);
+        page.getContent().forEach(System.out::println);
+    }
+
+}
 \ No newline at end of file
+package com.sl.sdn.repository.impl;
+
+import com.sl.sdn.dto.OrganDTO;
+import com.sl.sdn.repository.OrganRepository;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+@SpringBootTest
+class OrganRepositoryImplTest {
+
+    @Autowired
+    private OrganRepository organRepository;
+
+    @Test
+    void findByBid() {
+        OrganDTO byBid = organRepository.findByBid(100280L);
+        System.out.println(byBid);
+    }
+
+    @Test
+    void findAll() {
+        List<OrganDTO> organRepositoryAll = organRepository.findAll("北京");
+        System.out.println(organRepositoryAll);
+    }
+}
 \ No newline at end of file
+package com.sl.sdn.repository.impl;
+
+import com.sl.sdn.dto.TransportLineNodeDTO;
+import com.sl.sdn.entity.node.AgencyEntity;
+import com.sl.sdn.repository.TransportLineRepository;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+
+@SpringBootTest
+class TransportLineRepositoryImplTest {
+
+    @Autowired
+    private TransportLineRepository transportLineRepository;
+
+    @Test
+    void findShortTestPath() {
+        AgencyEntity start = AgencyEntity.builder().bid(100280L).build();
+        AgencyEntity end = AgencyEntity.builder().bid(210057L).build();
+
+        TransportLineNodeDTO shortTestPath = transportLineRepository.findShortTestPath(start, end);
+        System.out.println(shortTestPath);
+    }
+}
 \ No newline at end of file