Manage errors

Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>

pkg/coredata/measure.go

@@ -16,6 +16,7 @@ package coredata
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"maps"
 	"time"
@@ -23,6 +24,7 @@ import (
 	"github.com/getprobo/probo/pkg/gid"
 	"github.com/getprobo/probo/pkg/page"
 	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgconn"
 
 	"go.gearno.de/kit/pg"
 )
@@ -41,8 +43,24 @@ type (
 	}
 
 	Measures []*Measure
+
+	ErrMeasureNotFound struct {
+		Identifier string
+	}
+
+	ErrMeasureAlreadyExists struct {
+		message string
+	}
 )
 
+func (e ErrMeasureNotFound) Error() string {
+	return fmt.Sprintf("measure not found: %q", e.Identifier)
+}
+
+func (e ErrMeasureAlreadyExists) Error() string {
+	return e.message
+}
+
 func (m Measure) CursorKey(orderBy MeasureOrderField) page.CursorKey {
 	switch orderBy {
 	case MeasureOrderFieldCreatedAt:
@@ -395,6 +413,10 @@ LIMIT 1;
 
 	measure, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[Measure])
 	if err != nil {
+		if errors.Is(err, pgx.ErrNoRows) {
+			return &ErrMeasureNotFound{Identifier: measureID.String()}
+		}
+
 		return fmt.Errorf("cannot collect measures: %w", err)
 	}
 
@@ -525,7 +547,20 @@ VALUES (
 		"state":           m.State,
 	}
 	_, err := conn.Exec(ctx, q, args)
-	return err
+
+	if err != nil {
+		var pgErr *pgconn.PgError
+		if errors.As(err, &pgErr) {
+			if pgErr.Code == "23505" && pgErr.ConstraintName == "mitigations_org_ref_unique" {
+				return &ErrMeasureAlreadyExists{
+					message: fmt.Sprintf("measure with organization_id %s and reference_id %q already exists", m.OrganizationID, m.ReferenceID),
+				}
+			}
+		}
+		return fmt.Errorf("cannot insert measure: %w", err)
+	}
+
+	return nil
 }
 
 func (m *Measure) Update(