diff --git a/internal/database/claims.go b/internal/database/claims.go index b568596..fc41339 100644 --- a/internal/database/claims.go +++ b/internal/database/claims.go @@ -3,6 +3,8 @@ package database import ( "context" "vegan-barcode/internal/models" + + "github.com/vingarcia/ksql" ) func (database *Database) FindClaimsByProductID(product_id int) (claims []models.Claim, err error) { @@ -55,3 +57,16 @@ func (database *Database) FindClaimsByProductID(product_id int) (claims []models } return } + +func (database *Database) FindUserClaimById(claim_id int) (*UserClaim, error) { + ctx := context.Background() + + var claim UserClaim + + err := database.db.QueryOne(ctx, claim, "FROM user_claims WHERE id = ?", claim_id) + + if err == ksql.ErrRecordNotFound { + return nil, err + } + return &claim, err +}