Operations list shows ESs for each entity

This was about two hours of learning, and getting confused - and it
needs to be refactored. We need to be able to generate the list of EPs
for the Operation also.

At the moment this only prints the object representation. More work to
be done writing a method that gives us the id and textual representation
for the link in the table.

1 files changed, 21 insertions, 11 deletions

diff --git a/internal/models/engagement.go b/internal/models/engagement.go
index db1c2f5..26b468a 100644
--- a/internal/models/engagement.go
+++ b/internal/models/engagement.go
@@ -2,7 +2,6 @@ package models
 
 import (
 	"database/sql"
-	"errors"
 	"time"
 )
 
@@ -17,20 +16,31 @@ type EngagementStrategyModel struct {
 	DB *sql.DB
 }
 
-func (m *EngagementStrategyModel) GetForOperation(id int) (EngagementStrategy, error) {
+func (m *EngagementStrategyModel) GetForOperation(id int) ([]EngagementStrategy, error) {
 	stmt := `SELECT id, valid_from, valid_to FROM engagement_strategies
 WHERE operation_id = ?`
-	row := m.DB.QueryRow(stmt, id)
+	rows, err := m.DB.Query(stmt, id)
+	if err != nil {
+		return nil, err
+	}
 
-	var es EngagementStrategy
+	defer rows.Close()
 
-	err := row.Scan(&es.ID, &es.ValidFrom, &es.ValidTo)
-	if err != nil {
-		if errors.Is(err, sql.ErrNoRows) {
-			return EngagementStrategy{}, ErrNoRecord
-		} else {
-			return EngagementStrategy{}, err
+	var esses []EngagementStrategy
+
+	for rows.Next() {
+		var es EngagementStrategy
+		err = rows.Scan(&es.ID, &es.ValidFrom, &es.ValidTo)
+		if err != nil {
+			return nil, err
 		}
+
+		esses = append(esses, es)
+	}
+
+	if err = rows.Err(); err != nil {
+		return nil, err
 	}
-	return es, nil
+
+	return esses, nil
 }