API with NestJS #83. Text search with tsvector and raw SQL

November 14, 2022

It is very common to implement a feature of searching through the contents of the database. In one of the previous articles, we learned how to implement it in a simple way using pattern matching.

Today we take it a step further and learn about the data types explicitly designed for full-text search.

Text Search Types

PostgreSQL provides two data types that help us implement full-text search. They allow us to search through a collection of texts and find the ones that match a given query the most.

tsvector

The tsvector column stores the text in a format optimized for search. To parse a string into the tsvector format, we need the to_tsvector function.

When we look at the result of the above query, we notice a set of optimizations. One of the most apparent is grouping duplicates. Thanks to using the English dictionary, PostgreSQL noticed that “quick” and “quickly” are two variants of the same word.

Also, using the tsvector type can help us filter out stop words. They are very common, appear in almost every sentence, and don’t have much value when searching through text. Since we used the English dictionary in the above example, PostgreSQL filtered out the words “the” and “over”.

tsquery

The tsquery data type stores the text we want to search for. To transform a string into the tsquery format, we can use the to_tsquery function.

To check if a certain tsvector matches the tsquery, we need to use the @@ operator.

When doing the above, we can play with the &, |, and ! boolean operators. For example, we can use the ! operator to make sure a given text does not contain a particular word.

Check out the official documentation for a good explanation of all available operators.

Another handy function is plainto_tsquery. It takes an unformatted phrase and inserts the & operator between words. Because of that, it is an excellent choice to handle the input from the user.

Transforming the existing data

Let’s take a look at our posts table.

Unfortunately, it does not contain a tsvector column. The most straightforward solution to the above problem is to convert our data to tsvector on the fly.

We can take the above even further and combine the contents of the title and post_content columns to search through both.

The crucial issue with the above approach is that it causes PostgreSQL to transform the text from every record of the posts database, which can take a substantial amount of time.

Instead, I suggest defining a generated column that contains the data transformed into the tsvector format.

If you want to know moure about generated columns, check out Defining generated columns with PostgreSQL and TypeORM

Since we use the STORED keyword, we define a stored generated column that is saved in our database. PostgreSQL updates it automatically every time we modify the post_content and title columns.

We can now use our generated column when making a SELECT query to improve its performance drastically.

Ordering the results

So far, we haven’t paid attention to the order of the results of our SELECT query. Sorting the search results based on relevance could help the users quite a bit.

For example, we can indicate that the text from the title column is more important than the post_content column. To do that, let’s change how we create our text_tsvector column and use the setweight function.

Let’s compare the two following posts after modifying the text_tsvector column:

The combined value of the title and post_content is the same in both posts. However, the text_tsvector takes into account that the title column is more important.

Thanks to the above, we can now use the ts_rank function to order our results based on the weight of each column.

Implementing full-text search with NestJS

Let’s create a migration first to implement the above functionalities in our NestJS project.

20221113211441_add_post_tsvector.ts

The crucial thing to notice above is that we are creating a Generalized Inverted Index (GIN). It works well with text searching and is appropriate when a column contains more than one value. Doing that can speed up our SELECT queries very significantly.

If you want to know more about indexes, check out API with NestJS #82. Introduction to indexes with raw SQL queries

In one of the previous parts of this series, we implemented the support for the search query parameter.

posts.controller.ts

Finally, we need to modify the SQL queries that we make in our repository.

posts.repository.ts

Above, we use the keyset pagination that prevents us from sorting the results in a straightforward way. If you want to know more, check out API with NestJS #77. Offset and keyset pagination with raw SQL queries

Summary

In this article, we’ve gone through implementing full-text search in PostgreSQL. To do that, we had to learn about the tsvector and tsquery data types. In addition, we’ve created a stored generated column and a Generalized Inverted Index to improve the performance. By doing all of the above, we’ve created a fast search mechanism that is a good fit for many applications.