mirror of
https://github.com/usual2970/certimate.git
synced 2025-06-08 05:29:51 +00:00
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package nodeprocessor
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/usual2970/certimate/internal/domain"
|
|
"github.com/usual2970/certimate/internal/pkg/utils/certs"
|
|
"github.com/usual2970/certimate/internal/repository"
|
|
)
|
|
|
|
type uploadNode struct {
|
|
node *domain.WorkflowNode
|
|
outputRepo workflowOutputRepository
|
|
*nodeLogger
|
|
}
|
|
|
|
func NewUploadNode(node *domain.WorkflowNode) *uploadNode {
|
|
return &uploadNode{
|
|
node: node,
|
|
nodeLogger: NewNodeLogger(node),
|
|
outputRepo: repository.NewWorkflowOutputRepository(),
|
|
}
|
|
}
|
|
|
|
// Run 上传证书节点执行
|
|
// 包含上传证书的工作流，理论上应该手动执行，如果每天定时执行，也只是重新保存一下
|
|
func (n *uploadNode) Run(ctx context.Context) error {
|
|
n.AddOutput(ctx, n.node.Name, "进入上传证书节点")
|
|
|
|
nodeConfig := n.node.GetConfigForUpload()
|
|
|
|
// 查询上次执行结果
|
|
lastOutput, err := n.outputRepo.GetByNodeId(ctx, n.node.Id)
|
|
if err != nil && !domain.IsRecordNotFoundError(err) {
|
|
n.AddOutput(ctx, n.node.Name, "查询申请记录失败", err.Error())
|
|
return err
|
|
}
|
|
|
|
// 检查证书是否过期
|
|
// 如果证书过期，则直接返回错误
|
|
certX509, err := certs.ParseCertificateFromPEM(nodeConfig.Certificate)
|
|
if err != nil {
|
|
n.AddOutput(ctx, n.node.Name, "解析证书失败")
|
|
return err
|
|
}
|
|
if time.Now().After(certX509.NotAfter) {
|
|
n.AddOutput(ctx, n.node.Name, "证书已过期")
|
|
return errors.New("certificate is expired")
|
|
}
|
|
|
|
// 生成实体
|
|
certificate := &domain.Certificate{
|
|
Source: domain.CertificateSourceTypeUpload,
|
|
}
|
|
certificate.PopulateFromPEM(nodeConfig.Certificate, nodeConfig.PrivateKey)
|
|
|
|
// 保存执行结果
|
|
// TODO: 先保持一个节点始终只有一个输出，后续增加版本控制
|
|
currentOutput := &domain.WorkflowOutput{
|
|
WorkflowId: getContextWorkflowId(ctx),
|
|
NodeId: n.node.Id,
|
|
Node: n.node,
|
|
Succeeded: true,
|
|
Outputs: n.node.Outputs,
|
|
}
|
|
if lastOutput != nil {
|
|
currentOutput.Id = lastOutput.Id
|
|
}
|
|
if _, err := n.outputRepo.SaveWithCertificate(ctx, currentOutput, certificate); err != nil {
|
|
n.AddOutput(ctx, n.node.Name, "保存上传记录失败", err.Error())
|
|
return err
|
|
}
|
|
n.AddOutput(ctx, n.node.Name, "保存上传记录成功")
|
|
|
|
return nil
|
|
}
|