Собственный регистр mysql udf

Question

Я пытаюсь написать пользовательскую собственную пользовательскую функцию для mysql, поэтому в качестве примера я взял функцию str_ucwords из http://www.mysqludf.org/index.php.

my_bool str_ucwords_init(UDF_INIT *initid, UDF_ARGS *args, char *message)
{
    /* make sure user has provided exactly one string argument */
    if (args->arg_count != 1 || args->arg_type[0] != STRING_RESULT || args->args[0] == NULL)
    {
        strcpy(message,"str_ucwords requires one string argument");
        return 1;
    }

    /* str_ucwords() will not be returning null */
    initid->maybe_null=0;

    return 0;
}

char *str_ucwords(UDF_INIT *initid, UDF_ARGS *args,
    char *result, unsigned long *res_length, 
    char *null_value, char *error)
{
    int i;
    int new_word = 0;

    // copy the argument string into result
    strncpy(result, args->args[0], args->lengths[0]);

    *res_length = args->lengths[0];

    // capitalize the first character of each word in the string
    for (i = 0; i < *res_length; i++)
    {
        if (my_isalpha(&my_charset_latin1, result[i]))
        {
            if (!new_word)
            {
                new_word = 1;
                result[i] = my_toupper(&my_charset_latin1, result[i]);
            }
        }
        else
        {
            new_word = 0;
        }
    }

    return result;
}

Это нормально работает, если я пытаюсь select str_ucwords("test string");, но если я пытаюсь выбрать поля из базы данных, например select str_ucwords(name) from name;, я ничего не получаю.

Как изменить эту функцию, чтобы она позволяла извлекать данные из полей базы данных?

Я уже пытался удалить args->arg_type[0] != STRING_RESULT из функции init.

Answer 1

Проблема не в типе аргумента, а в том, что NULL, когда вызывается str_ucwords_init (так как str_ucwords_init вызывается до получения каких-либо строк).Чтобы заставить str_ucwords работать с полями, вы должны поддерживать значения NULL, установив initid->maybe_null в 1 в функции _init и установив *null_value в 1 (и result в NULL, хотя это может быть и не так.необходимо) в str_ucwords, когда фактический аргумент равен нулю.

my_bool str_ucwords_init(UDF_INIT *initid, UDF_ARGS *args, char *message) {
    unsigned long res_length;

    /* make sure user has provided exactly one argument */
    if (args->arg_count != 1) {
        snprintf(message, MYSQL_ERRMSG_SIZE, "wrong number of arguments: str_ucwords requires one string argument, got %d arguments", args->arg_count);
        return 1;
    }
    /* make sure user has provided a string argument */
    if (args->arg_type[0] != STRING_RESULT) {
        snprintf(message, MYSQL_ERRMSG_SIZE, "str_ucwords requires one string argument (expected type %d, got type %d)", STRING_RESULT, args->arg_type[0]);
        return 1;
    }

    res_length = args->lengths[0];

    if (SIZE_MAX < res_length) {
        snprintf(message, MYSQL_ERRMSG_SIZE, "res_length (%lu) cannot be greater than SIZE_MAX (%zu)", res_length, (size_t) (SIZE_MAX));
        return 1;
    }

    initid->ptr = NULL;

    if (res_length > 255) {
        char *tmp = (char *) malloc((size_t) res_length); /* This is a safe cast because res_length <= SIZE_MAX. */
        if (tmp == NULL) {
            snprintf(message, MYSQL_ERRMSG_SIZE, "malloc() failed to allocate %zu bytes of memory", (size_t) res_length);
            return 1;
        }
        initid->ptr = tmp;
    }

    initid->maybe_null = 1;
    initid->max_length = res_length;
    return 0;
}

char *str_ucwords(UDF_INIT *initid, UDF_ARGS *args,
            char *result, unsigned long *res_length,
            char *null_value, char *error)
{
    int i;
    int new_word = 1;

    if (args->args[0] == NULL) {
        result = NULL;
        *res_length = 0;
        *null_value = 1;
    } else {
        if (initid->ptr != NULL) {
            result = initid->ptr;
        }

        // copy the argument string into result
        memcpy(result, args->args[0], args->lengths[0]);
        *res_length = args->lengths[0];

        // capitalize the first character of each word in the string
        for (i = 0; i < *res_length; i++) {
            if (my_isalpha(&my_charset_latin1, result[i])) {
                if (new_word) {
                    new_word = 0;
                    result[i] = my_toupper(&my_charset_latin1, result[i]);
                } else {
                    result[i] = my_tolower(&my_charset_latin1, result[i]);
                }
            } else {
                new_word = 1;
            }
        }
    }
    return result;
}

Более поздние выпуски lib_mysqludf_str должны поддерживать значения NULL в функциях без изменений, что означает, что они также должны работать со столбцами таблицы.