diff --git a/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QA.java b/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QA.java index e9b0566..b3fc4e8 100644 --- a/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QA.java +++ b/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QA.java @@ -1,5 +1,8 @@ package io.github.timemachinelab.core.qatree; +import lombok.Data; + +@Data public abstract class QA { protected final QAType type; @@ -10,5 +13,14 @@ public abstract class QA { public QA(QAType type) { this.type = type; + this.createTime = System.currentTimeMillis(); + this.updateTime = this.createTime; + } + + /** + * 更新updateTime为当前时间戳 + */ + public void updateTimestamp() { + this.updateTime = System.currentTimeMillis(); } } diff --git a/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QaTreeDomain.java b/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QaTreeDomain.java index 6717c5c..2425b3f 100644 --- a/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QaTreeDomain.java +++ b/prompto-lab-app/src/main/java/io/github/timemachinelab/core/qatree/QaTreeDomain.java @@ -13,8 +13,8 @@ public QaTree createTree(String userStartQuestion) { return new QaTree(startNode); } - public QaTree appendNode(QaTree tree, String parentId, QaTreeNode node) { - tree.addNode(parentId, node); + public QaTree appendNode(QaTree tree, String parentId, QA qa) { + tree.addNode(parentId, new QaTreeNode(qa)); return tree; } }