test_lfind.c

Go to the documentation of this file.

/*--------------------------------------------------------------------------
 *
 * test_lfind.c
 *      Test correctness of optimized linear search functions.
 *
 * Copyright (c) 2022-2026, PostgreSQL Global Development Group
 *
 * IDENTIFICATION
 *      src/test/modules/test_lfind/test_lfind.c
 *
 * -------------------------------------------------------------------------
 */
 
#include "postgres.h"
 
#include "fmgr.h"
#include "port/pg_lfind.h"
 
/*
 * Convenience macros for testing both vector and scalar operations. The 2x
 * factor is to make sure iteration works
 */
#define LEN_NO_TAIL(vectortype) (2 * sizeof(vectortype))
#define LEN_WITH_TAIL(vectortype) (LEN_NO_TAIL(vectortype) + 3)
 
PG_MODULE_MAGIC;
 
/* workhorse for test_lfind8 */
static void
test_lfind8_internal(uint8 key)
{
    uint8       charbuf[LEN_WITH_TAIL(Vector8)];
    const int   len_no_tail = LEN_NO_TAIL(Vector8);
    const int   len_with_tail = LEN_WITH_TAIL(Vector8);
 
    memset(charbuf, 0xFF, len_with_tail);
    /* search tail to test one-byte-at-a-time path */
    charbuf[len_with_tail - 1] = key;
    if (key > 0x00 && pg_lfind8(key - 1, charbuf, len_with_tail))
        elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key - 1);
    if (key < 0xFF && !pg_lfind8(key, charbuf, len_with_tail))
        elog(ERROR, "pg_lfind8() did not find existing element '0x%x'", key);
    if (key < 0xFE && pg_lfind8(key + 1, charbuf, len_with_tail))
        elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key + 1);
 
    memset(charbuf, 0xFF, len_with_tail);
    /* search with vector operations */
    charbuf[len_no_tail - 1] = key;
    if (key > 0x00 && pg_lfind8(key - 1, charbuf, len_no_tail))
        elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key - 1);
    if (key < 0xFF && !pg_lfind8(key, charbuf, len_no_tail))
        elog(ERROR, "pg_lfind8() did not find existing element '0x%x'", key);
    if (key < 0xFE && pg_lfind8(key + 1, charbuf, len_no_tail))
        elog(ERROR, "pg_lfind8() found nonexistent element '0x%x'", key + 1);
}
 
PG_FUNCTION_INFO_V1(test_lfind8);
Datum
test_lfind8(PG_FUNCTION_ARGS)
{
    test_lfind8_internal(0);
    test_lfind8_internal(1);
    test_lfind8_internal(0x7F);
    test_lfind8_internal(0x80);
    test_lfind8_internal(0x81);
    test_lfind8_internal(0xFD);
    test_lfind8_internal(0xFE);
    test_lfind8_internal(0xFF);
 
    PG_RETURN_VOID();
}
 
/* workhorse for test_lfind8_le */
static void
test_lfind8_le_internal(uint8 key)
{
    uint8       charbuf[LEN_WITH_TAIL(Vector8)];
    const int   len_no_tail = LEN_NO_TAIL(Vector8);
    const int   len_with_tail = LEN_WITH_TAIL(Vector8);
 
    memset(charbuf, 0xFF, len_with_tail);
    /* search tail to test one-byte-at-a-time path */
    charbuf[len_with_tail - 1] = key;
    if (key > 0x00 && pg_lfind8_le(key - 1, charbuf, len_with_tail))
        elog(ERROR, "pg_lfind8_le() found nonexistent element <= '0x%x'", key - 1);
    if (key < 0xFF && !pg_lfind8_le(key, charbuf, len_with_tail))
        elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key);
    if (key < 0xFE && !pg_lfind8_le(key + 1, charbuf, len_with_tail))
        elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key + 1);
 
    memset(charbuf, 0xFF, len_with_tail);
    /* search with vector operations */
    charbuf[len_no_tail - 1] = key;
    if (key > 0x00 && pg_lfind8_le(key - 1, charbuf, len_no_tail))
        elog(ERROR, "pg_lfind8_le() found nonexistent element <= '0x%x'", key - 1);
    if (key < 0xFF && !pg_lfind8_le(key, charbuf, len_no_tail))
        elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key);
    if (key < 0xFE && !pg_lfind8_le(key + 1, charbuf, len_no_tail))
        elog(ERROR, "pg_lfind8_le() did not find existing element <= '0x%x'", key + 1);
}
 
PG_FUNCTION_INFO_V1(test_lfind8_le);
Datum
test_lfind8_le(PG_FUNCTION_ARGS)
{
    test_lfind8_le_internal(0);
    test_lfind8_le_internal(1);
    test_lfind8_le_internal(0x7F);
    test_lfind8_le_internal(0x80);
    test_lfind8_le_internal(0x81);
    test_lfind8_le_internal(0xFD);
    test_lfind8_le_internal(0xFE);
    test_lfind8_le_internal(0xFF);
 
    PG_RETURN_VOID();
}
 
PG_FUNCTION_INFO_V1(test_lfind32);
Datum
test_lfind32(PG_FUNCTION_ARGS)
{
#define TEST_ARRAY_SIZE 135
    uint32      test_array[TEST_ARRAY_SIZE] = {0};
 
    test_array[8] = 1;
    test_array[64] = 2;
    test_array[TEST_ARRAY_SIZE - 1] = 3;
 
    if (pg_lfind32(1, test_array, 4))
        elog(ERROR, "pg_lfind32() found nonexistent element");
    if (!pg_lfind32(1, test_array, TEST_ARRAY_SIZE))
        elog(ERROR, "pg_lfind32() did not find existing element");
 
    if (pg_lfind32(2, test_array, 32))
        elog(ERROR, "pg_lfind32() found nonexistent element");
    if (!pg_lfind32(2, test_array, TEST_ARRAY_SIZE))
        elog(ERROR, "pg_lfind32() did not find existing element");
 
    if (pg_lfind32(3, test_array, 96))
        elog(ERROR, "pg_lfind32() found nonexistent element");
    if (!pg_lfind32(3, test_array, TEST_ARRAY_SIZE))
        elog(ERROR, "pg_lfind32() did not find existing element");
 
    if (pg_lfind32(4, test_array, TEST_ARRAY_SIZE))
        elog(ERROR, "pg_lfind32() found nonexistent element");
 
    PG_RETURN_VOID();
}