virtual-patient/virtual-patient-graph/src/main/java/com/supervision/node/TreatmentPlanNode.java

package com.supervision.node;

import lombok.Data;
import org.neo4j.ogm.annotation.*;

import java.util.HashSet;
import java.util.Set;

@Data
@NodeEntity("处置措施")
public class TreatmentPlanNode {

    @Id
    @GeneratedValue
    private Long id;

    @Property("处置计划")
    private String treatmentPlanName;

    @Relationship("所属药品")
    private Set<DrugNode> drugNodeSet;

    public TreatmentPlanNode(String treatmentPlanName) {
        this.treatmentPlanName = treatmentPlanName;
        drugNodeSet = new HashSet<>();
    }
}