viljarb/LazyPM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make tasks use map instead of slice

Browse Source
This commit is contained in:
Robin Olsen
2026-02-23 13:13:34 +01:00
parent 2c2fbac3a3
commit 8decd74952
5 changed files with 28 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
cmd/survey/init.go
View File

@@ -26,15 +26,15 @@ func initInterfaces() map[string]task.Interface {
return interfaces
}
func initTasks(app *service.App) []task.Tasker {
var taskList []task.Tasker
func initTasks(app *service.App) map[string]task.Tasker {
taskMap := make(map[string]task.Tasker)
for _, name := range task.ListTasks() {
t, err := task.GetTasks(name, app)
t, err := task.GetTask(name, app)
if err != nil {
continue
}
taskList = append(taskList, t)
taskMap[name] = t
}
return taskList
return taskMap
}

23
cmd/survey/runner.go
View File

@@ -32,15 +32,14 @@ func runStartCmd(cmd *cobra.Command, args []string) error {
}
}
if cmd.Flags().Changed("stage") {
if surveyCmd.Task < 1 || surveyCmd.Task > len(surveyTasks) {
return fmt.Errorf("invalid stage")
if cmd.Flags().Changed("task") {
if surveyTask := surveyTasks[surveyCmd.Task]; surveyTask == nil {
return fmt.Errorf("invalid task, valid are %v", task.ListTasks())
}
if err := task.RunTask(cmd.Context(), surveyTasks[surveyCmd.Task-1],
interfaces[surveyCmd.InterfaceType], tasks.InterfaceToType(interfaces[surveyCmd.InterfaceType])); err != nil {
return err
surveyTasks = map[string]task.Tasker{
surveyCmd.Task: surveyTasks[surveyCmd.Task],
}
return nil
}
if err := newIntroModel().Run(); err != nil {
@@ -53,7 +52,7 @@ func runStartCmd(cmd *cobra.Command, args []string) error {
return nil
}
func taskLoop(ctx context.Context, surveyTasks []task.Tasker, interfaces map[string]task.Interface) error {
func taskLoop(ctx context.Context, surveyTasks map[string]task.Tasker, interfaces map[string]task.Interface) error {
var iNames []string
for name := range interfaces {
iNames = append(iNames, name)
@@ -63,14 +62,16 @@ func taskLoop(ctx context.Context, surveyTasks []task.Tasker, interfaces map[str
iNames[i], iNames[j] = iNames[j], iNames[i]
})
for i, t := range surveyTasks {
idx := i % len(iNames)
selected := interfaces[iNames[idx]]
idx := 0
for _, t := range surveyTasks {
iIdx := idx % len(iNames)
selected := interfaces[iNames[iIdx]]
if err := task.RunTask(ctx, t, selected, tasks.InterfaceToType(selected)); err != nil {
return err
}
}
idx++
return nil
}