增加任务接口逻辑

2024-12-11 21:15:59 +08:00
parent e2ba6924b8
commit 9f4fb0a9d0
24 changed files with 765 additions and 63 deletions
--- a/internal/svc/service_context.go
+++ b/internal/svc/service_context.go
@@ -6,13 +6,20 @@ import (
 )

 type ServiceContext struct {
-	Config    config.Config
-	TaskModel model.NhTaskModel
+	Config               config.Config
+	TaskModel            model.NhTaskModel
+	TaskAssetModel       model.NhTaskAssetModel
+	TaskAssetRecordModel model.NhTaskAssetRecordModel
+	TaskProgressModel    model.NhTaskProgressModel
 }

 func NewServiceContext(c config.Config) *ServiceContext {
+	dbConn := c.MySql.Conn()
 	return &ServiceContext{
-		Config:    c,
-		TaskModel: model.NewNhTaskModel(c.MySql.Conn()),
+		Config:               c,
+		TaskModel:            model.NewNhTaskModel(dbConn),
+		TaskAssetModel:       model.NewNhTaskAssetModel(dbConn),
+		TaskAssetRecordModel: model.NewNhTaskAssetRecordModel(dbConn),
+		TaskProgressModel:    model.NewNhTaskProgressModel(dbConn),
 	}
 }