Browse Source

更新图谱高危对象信息

MarkHuang 4 years ago
parent
commit
03ee83e041

+ 4 - 0
src/main/java/com/diagbot/entity/node/Operation.java

@@ -7,6 +7,7 @@ import lombok.Getter;
 import lombok.Setter;
 import org.neo4j.ogm.annotation.Labels;
 import org.neo4j.ogm.annotation.NodeEntity;
+import org.neo4j.ogm.annotation.Property;
 import org.neo4j.ogm.annotation.Relationship;
 
 import java.io.Serializable;
@@ -18,6 +19,9 @@ import java.util.Set;
 @NodeEntity(label = "手术和操作")
 public class Operation extends BaseNode  {
 
+    @Property(name = "高危级别")
+    private String risklevel;
+
     @Relationship(type="疾病相关手术和操作", direction = Relationship.INCOMING)
     private Set<Disease> diseases;
 

+ 1 - 1
src/main/java/com/diagbot/repository/MedicineNode.java

@@ -198,7 +198,7 @@ public class MedicineNode {
 
             medicine = medicineRepository.findByNameIs(drugname);
 
-            if (null!=medicine.getRisklevel() && medicine.getRisklevel().trim().length()>0) {
+            if (null != medicine && null!=medicine.getRisklevel() && medicine.getRisklevel().trim().length()>0) {
                 highRiskNeoDTO.setIsHighrisk(true);
 
             }

+ 5 - 0
src/main/java/com/diagbot/repository/OperationNode.java

@@ -53,6 +53,11 @@ public class OperationNode {
             highRiskNeoDTO.setStandname(opname);
 
             operation = operationRepository.findByNameIs(opname);
+
+            if (null != operation && null!=operation.getRisklevel() && operation.getRisklevel().trim().length()>0) {
+                highRiskNeoDTO.setIsHighrisk(true);
+
+            }
         }
 
         return highRiskNeoDTOS;