Busca textual

import pg from "pg";

const pool = new pg.Pool({ connectionString: process.env.DATABASE_URL });

// websearch_to_tsquery aceita o que o usuário digita: aspas, OR e -palavra
const BUSCA = `
  SELECT id, nome,
         ts_headline('pt', descricao, q, 'StartSel=<mark>, StopSel=</mark>') AS trecho,
         count(*) OVER () AS total
    FROM produtos, websearch_to_tsquery('pt', $1) AS q
   WHERE busca @@ q
   ORDER BY ts_rank(busca, q) DESC, id
   LIMIT $2`;

async function buscar(termo: string, limite = 3) {
  const { rows } = await pool.query(BUSCA, [termo, limite]);
  return { total: Number(rows[0]?.total ?? 0), resultados: rows.map(({ total, ...r }) => r) };
}

const termos = ["tenis corrida", "corredores treino", "mochila -infantil", "relogio \"sem fio\""];
for (const termo of termos) {
  console.log(termo, JSON.stringify(await buscar(termo)));
}
await pool.end();

import json
import os

import psycopg
from psycopg.rows import dict_row

conexao = psycopg.connect(os.environ["DATABASE_URL"], row_factory=dict_row)

# websearch_to_tsquery aceita o que o usuário digita: aspas, OR e -palavra
BUSCA = """
  SELECT id, nome,
         ts_headline('pt', descricao, q, 'StartSel=<mark>, StopSel=</mark>') AS trecho,
         count(*) OVER () AS total
    FROM produtos, websearch_to_tsquery('pt', %s) AS q
   WHERE busca @@ q
   ORDER BY ts_rank(busca, q) DESC, id
   LIMIT %s"""


def buscar(termo: str, limite: int = 3) -> dict:
    linhas = conexao.execute(BUSCA, (termo, limite)).fetchall()
    total = linhas[0]["total"] if linhas else 0
    resultados = [{k: v for k, v in linha.items() if k != "total"} for linha in linhas]
    return {"total": total, "resultados": resultados}


for termo in ["tenis corrida", "corredores treino", "mochila -infantil", 'relogio "sem fio"']:
    print(termo, json.dumps(buscar(termo), ensure_ascii=False))

package main

import (
	"context"
	"fmt"
	"os"

	"github.com/jackc/pgx/v5"
	"github.com/jackc/pgx/v5/pgxpool"
)

// websearch_to_tsquery aceita o que o usuário digita: aspas, OR e -palavra
const busca = `
  SELECT id, nome,
         ts_headline('pt', descricao, q, 'StartSel=<mark>, StopSel=</mark>') AS trecho,
         count(*) OVER () AS total
    FROM produtos, websearch_to_tsquery('pt', $1) AS q
   WHERE busca @@ q
   ORDER BY ts_rank(busca, q) DESC, id
   LIMIT $2`

type Resultado struct {
	ID     int
	Nome   string
	Trecho string
	Total  int64
}

func buscar(ctx context.Context, db *pgxpool.Pool, termo string, limite int) ([]Resultado, error) {
	linhas, err := db.Query(ctx, busca, termo, limite)
	if err != nil {
		return nil, err
	}
	return pgx.CollectRows(linhas, pgx.RowToStructByPos[Resultado])
}

func main() {
	ctx := context.Background()
	db, err := pgxpool.New(ctx, os.Getenv("DATABASE_URL"))
	if err != nil {
		panic(err)
	}
	defer db.Close()
	termos := []string{"tenis corrida", "corredores treino", "mochila -infantil", `relogio "sem fio"`}
	for _, termo := range termos {
		r, err := buscar(ctx, db, termo, 3)
		fmt.Println(termo, r, err)
	}
}

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class Busca {
    record Resultado(int id, String nome, String trecho, long total) {}

    // websearch_to_tsquery aceita o que o usuário digita: aspas, OR e -palavra
    static final String BUSCA = """
        SELECT id, nome,
               ts_headline('pt', descricao, q, 'StartSel=<mark>, StopSel=</mark>') AS trecho,
               count(*) OVER () AS total
          FROM produtos, websearch_to_tsquery('pt', ?) AS q
         WHERE busca @@ q
         ORDER BY ts_rank(busca, q) DESC, id
         LIMIT ?""";

    static List<Resultado> buscar(Connection db, String termo, int limite) throws SQLException {
        try (var comando = db.prepareStatement(BUSCA)) {
            comando.setString(1, termo);
            comando.setInt(2, limite);
            var resultados = new ArrayList<Resultado>();
            try (var linhas = comando.executeQuery()) {
                while (linhas.next()) {
                    resultados.add(new Resultado(linhas.getInt("id"), linhas.getString("nome"),
                        linhas.getString("trecho"), linhas.getLong("total")));
                }
            }
            return resultados;
        }
    }

    public static void main(String[] args) throws Exception {
        try (var db = DriverManager.getConnection(System.getenv("JDBC_URL"))) {
            var termos = List.of(
                "tenis corrida", "corredores treino", "mochila -infantil", "relogio \"sem fio\"");
            for (var termo : termos) {
                System.out.println(termo + " " + buscar(db, termo, 3));
            }
        }
    }
}

using Npgsql;

await using var db = NpgsqlDataSource.Create(Environment.GetEnvironmentVariable("DATABASE_URL")!);

// websearch_to_tsquery aceita o que o usuário digita: aspas, OR e -palavra
const string Busca = """
    SELECT id, nome,
           ts_headline('pt', descricao, q, 'StartSel=<mark>, StopSel=</mark>') AS trecho,
           count(*) OVER () AS total
      FROM produtos, websearch_to_tsquery('pt', $1) AS q
     WHERE busca @@ q
     ORDER BY ts_rank(busca, q) DESC, id
     LIMIT $2
    """;

async Task<List<Resultado>> Buscar(string termo, int limite = 3)
{
    await using var comando = db.CreateCommand(Busca);
    comando.Parameters.Add(new NpgsqlParameter { Value = termo });
    comando.Parameters.Add(new NpgsqlParameter { Value = limite });
    var resultados = new List<Resultado>();
    await using var linhas = await comando.ExecuteReaderAsync();
    while (await linhas.ReadAsync())
        resultados.Add(new(linhas.GetInt32(0), linhas.GetString(1),
            linhas.GetString(2), linhas.GetInt64(3)));
    return resultados;
}

string[] termos =
    ["tenis corrida", "corredores treino", "mochila -infantil", "relogio \"sem fio\""];
foreach (var termo in termos)
    Console.WriteLine($"{termo} [{string.Join(", ", await Buscar(termo))}]");

record Resultado(int Id, string Nome, string Trecho, long Total);