优化结构

pkg/module/module.go

@@ -6,6 +6,7 @@ package module
 import (
 	"github.com/gin-gonic/gin"
 	"kra/pkg/database/migration"
+	"kra/pkg/mq"
 	"kra/pkg/task"
 )
 
@@ -29,11 +30,12 @@ type TimedTask struct {
 }
 
 type Definition struct {
-	Name       string
-	Migrations []migration.Step
-	Surface    Surface
-	TimedTasks []TimedTask
-	Tasks      []task.Method
+	Name                     string
+	Migrations               []migration.Step
+	Surface                  Surface
+	TimedTasks               []TimedTask
+	Tasks                    []task.Method
+	SubscriptionContributors []mq.SubscriptionContributor
 }
 
 type Catalog struct {
@@ -73,6 +75,16 @@ func (c Catalog) TaskMethods() []task.Method {
 	return methods
 }
 
+// SubscriptionContributors returns the module declarations that are applied
+// to the process-wide MQ runtime during application startup.
+func (c Catalog) SubscriptionContributors() []mq.SubscriptionContributor {
+	var contributors []mq.SubscriptionContributor
+	for _, item := range c.Definitions {
+		contributors = append(contributors, item.SubscriptionContributors...)
+	}
+	return contributors
+}
+
 type RouteRegistrar interface {
 	RegisterRoutes(public, private *gin.RouterGroup, engine *gin.Engine)
 }