✨ feat: add full text search endpoint close #9

Author: nmdra

api/handlers.go

@@ -17,6 +17,7 @@ type AddBookRequest struct {
 	Isbn        string `json:"isbn"`
 }
 
+// POST /books
 func (h *BookHandler) AddBook(c echo.Context) error {
 	ctx := c.Request().Context()
 	var req AddBookRequest
@@ -30,14 +31,15 @@ func (h *BookHandler) AddBook(c echo.Context) error {
 	default:
 	}
 
-	err := h.Service.AddBook(c.Request().Context(), req.Isbn, req.Title, req.Description)
+	err := h.Service.AddBook(ctx, req.Isbn, req.Title, req.Description)
 	if err != nil {
 		return c.JSON(http.StatusInternalServerError, echo.Map{"error": err.Error()})
 	}
 
 	return c.JSON(http.StatusCreated, echo.Map{"status": "book added"})
 }
 
+// GET /search/vector?q=
 func (h *BookHandler) SearchBooks(c echo.Context) error {
 	ctx := c.Request().Context()
 	query := c.QueryParam("q")
@@ -61,3 +63,28 @@ func (h *BookHandler) SearchBooks(c echo.Context) error {
 
 	return c.JSON(http.StatusOK, results)
 }
+
+// GET /search/text?q=
+func (h *BookHandler) FullTextSearch(c echo.Context) error {
+	ctx := c.Request().Context()
+	query := c.QueryParam("q")
+	if query == "" {
+		return c.JSON(http.StatusBadRequest, echo.Map{"error": "missing query"})
+	}
+
+	select {
+	case <-ctx.Done():
+		return c.JSON(http.StatusRequestTimeout, echo.Map{"error": "request canceled or timed out"})
+	default:
+	}
+
+	results, err := h.Service.FullTextSearch(ctx, query)
+	if err != nil {
+		if ctx.Err() != nil {
+			return c.JSON(http.StatusRequestTimeout, echo.Map{"error": "request timed out"})
+		}
+		return c.JSON(http.StatusInternalServerError, echo.Map{"error": err.Error()})
+	}
+
+	return c.JSON(http.StatusOK, results)
+}

cmd/main.go

@@ -98,7 +98,8 @@ func main() {
 	e.GET("/ping", func(c echo.Context) error {
 		return c.String(200, "pong")
 	})
-	e.GET("/search", bookHandler.SearchBooks)
+	e.GET("/search/semantic", bookHandler.SearchBooks)
+	e.GET("/search/text", bookHandler.FullTextSearch)
 	e.POST("/books", bookHandler.AddBook)
 
 	// e.Logger.Fatal(e.Start(fmt.Sprintf(":%d", cfg.port)))

db/books.sql

@@ -11,4 +11,11 @@ LIMIT 5;
 -- name: GetBookByISBN :one
 SELECT id, isbn
 FROM books
-WHERE isbn = $1;
+WHERE isbn = $1;
+
+-- name: SearchBooksByText :many
+SELECT id, isbn, title, description
+FROM books
+WHERE tsv @@ plainto_tsquery('english', $1)
+ORDER BY ts_rank(tsv, plainto_tsquery('english', $1)) DESC
+LIMIT 10;

db/migrations/000005_full-text-search.down.sql

@@ -0,0 +1,4 @@
+DROP INDEX IF EXISTS idx_books_tsv;
+
+ALTER TABLE books
+DROP COLUMN IF EXISTS tsv;

db/migrations/000005_full-text-search.up.sql

@@ -0,0 +1,6 @@
+ALTER TABLE books
+ADD COLUMN tsv tsvector GENERATED ALWAYS AS (
+    to_tsvector('english', coalesce(title, '') || ' ' || coalesce(description, ''))
+) STORED;
+
+CREATE INDEX idx_books_tsv ON books USING GIN (tsv);

internal/repository/books.sql.go

@@ -124,3 +124,23 @@ func cosineSimilarity(a, b []float32) (float64, error) {
 
 	return dot / (math.Sqrt(normA) * math.Sqrt(normB)), nil
 }
+
+func (s *BookService) FullTextSearch(ctx context.Context, query string) ([]repository.SearchBooksByTextRow, error) {
+	select {
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	default:
+	}
+
+	if query == "" {
+		return nil, fmt.Errorf("search query cannot be empty")
+	}
+
+	books, err := s.Repository.SearchBooksByText(ctx, query)
+	if err != nil {
+		s.Logger.Error("Full-text search failed", "query", query, "error", err)
+		return nil, fmt.Errorf("full-text search failed: %w", err)
+	}
+
+	return books, nil
+}