diff options
Diffstat (limited to 'internal/models/engagement.go')
-rw-r--r-- | internal/models/engagement.go | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/internal/models/engagement.go b/internal/models/engagement.go new file mode 100644 index 0000000..db1c2f5 --- /dev/null +++ b/internal/models/engagement.go @@ -0,0 +1,36 @@ +package models + +import ( + "database/sql" + "errors" + "time" +) + +type EngagementStrategy struct { + ID int + ValidFrom time.Time + ValidTo time.Time + Operation Operation +} + +type EngagementStrategyModel struct { + DB *sql.DB +} + +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) + + var es EngagementStrategy + + 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 + } + } + return es, nil +} |