Fix searching for very short (1 or 2 bytes) queries.

[plocate] / plocate-build.cpp

diff --git a/plocate-build.cpp b/plocate-build.cpp
index 658aea2c63f6784f6afe2286441f9943b458934e..c6804497248f4a9a828200d784b9450f8986ad72 100644 (file)
--- a/plocate-build.cpp
+++ b/plocate-build.cpp
@@ -377,6 +377,7 @@ void do_build(const char *infile, const char *outfile, int block_size)
        }
        corpus.flush_block();
        dprintf("Read %zu files from %s\n", corpus.num_files, infile);
+       hdr.num_docids = corpus.filename_blocks.size();
 
        // Stick an empty block at the end as sentinel.
        corpus.filename_blocks.push_back(ftell(outfp));