39 lines
821 B
Go
39 lines
821 B
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/babariviere/short/internal/db"
|
||
|
|
"github.com/babariviere/short/internal/oas"
|
||
|
|
"github.com/jackc/pgx/v5"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ oas.Handler = (*handler)(nil)
|
||
|
|
|
||
|
|
type handler struct {
|
||
|
|
oas.UnimplementedHandler
|
||
|
|
queries *db.Queries
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewHandler(queries *db.Queries) *handler {
|
||
|
|
return &handler{
|
||
|
|
queries: queries,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *handler) RedirectLongURL(ctx context.Context, params oas.RedirectLongURLParams) (oas.RedirectLongURLRes, error) {
|
||
|
|
res, err := h.queries.GetURLByHash(ctx, params.Hash)
|
||
|
|
if err != nil {
|
||
|
|
if errors.Is(err, pgx.ErrNoRows) {
|
||
|
|
return &oas.RedirectLongURLNotFound{}, nil
|
||
|
|
}
|
||
|
|
return nil, fmt.Errorf("unable to fetch URL by hash: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &oas.RedirectLongURLTemporaryRedirect{
|
||
|
|
Location: oas.NewOptString(res.LongUrl),
|
||
|
|
}, nil
|
||
|
|
}
|