fix(be): use ansible task repo

Author: fiftin

cli/cmd/root.go

@@ -73,7 +73,7 @@ func Execute() {
 func runService() {
 	store := createStore("root")
 	terraformStore := proFactory.NewTerraformStore(store)
-	ansibleTaskRepository := proFactory.NewAnsibleTaskRepository(store)
+	ansibleTaskRepo := proFactory.NewAnsibleTaskRepository(store)
 
 	projectService := server.NewProjectService(store, store)
 	encryptionService := server.NewAccessKeyEncryptionService(store, store, store)
@@ -93,6 +93,7 @@ func runService() {
 
 	taskPool := tasks.CreateTaskPool(
 		store,
+		ansibleTaskRepo,
 		inventoryService,
 		encryptionService,
 		accessKeyInstallationService,
@@ -130,7 +131,7 @@ func runService() {
 	route := api.Route(
 		store,
 		terraformStore,
-		ansibleTaskRepository,
+		ansibleTaskRepo,
 		&taskPool,
 		projectService,
 		integrationService,

services/tasks/TaskPool.go

@@ -58,6 +58,7 @@ type TaskPool struct {
 	logger chan logRecord
 
 	store                  db.Store
+	ansibleTaskRepo        db.AnsibleTaskRepository
 	logWriteService        pro_interfaces.LogWriteService
 	inventoryService       server.InventoryService
 	encryptionService      server.AccessKeyEncryptionService
@@ -70,6 +71,7 @@ type TaskPool struct {
 
 func CreateTaskPool(
 	store db.Store,
+	ansibleTaskRepo db.AnsibleTaskRepository,
 	inventoryService server.InventoryService,
 	encryptionService server.AccessKeyEncryptionService,
 	keyInstallationService server.AccessKeyInstallationService,
@@ -218,6 +220,7 @@ func (p *TaskPool) handleLogs() {
 
 			newStage, newState, err := stage_parsers.MoveToNextStage(
 				p.store,
+				p.ansibleTaskRepo,
 				p.logWriteService,
 				record.task.Template.App,
 				record.task.Task.ProjectID,

services/tasks/TaskRunner_test.go

@@ -55,9 +55,11 @@ func TestTaskRunnerRun(t *testing.T) {
 
 	pool := CreateTaskPool(
 		store,
+		nil,
 		&InventoryServiceMock{},
 		nil,
 		keyInstaller,
+		nil,
 	)
 
 	go pool.Run()