39 lines
791 B
Go
39 lines
791 B
Go
package cmd
|
|
|
|
import (
|
|
"framework_v2/internal/providers/config"
|
|
"framework_v2/internal/providers/ent"
|
|
"framework_v2/internal/providers/jobs"
|
|
"framework_v2/internal/providers/logger"
|
|
"framework_v2/internal/providers/redis"
|
|
"github.com/hibiken/asynq"
|
|
"github.com/spf13/cobra"
|
|
"log"
|
|
)
|
|
|
|
var workerCommand = &cobra.Command{
|
|
Use: "worker",
|
|
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
config.InitConfig()
|
|
logger.InitLogger()
|
|
ent.InitEnt()
|
|
redis.InitRedis()
|
|
jobs.InitAsynQServer()
|
|
|
|
runWorker()
|
|
},
|
|
}
|
|
|
|
func runWorker() {
|
|
mux := asynq.NewServeMux()
|
|
|
|
//mux.HandleFunc(tasks.DocumentChunkTask, tasks.HandleDocumentChunkTask)
|
|
//mux.HandleFunc(tasks.DocumentImportTask, tasks.HandleDocumentImportTask)
|
|
|
|
if err := jobs.AsynQServer.Run(mux); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
}
|