mysql udf 性能_适当的mysql udf

问题不在于参数的类型,而是在调用str_ucwords_init时它是NULL(因为在检索任何行之前调用了str_ucwords_init).要使str_ucwords与字段一起使用,您必须通过在_init函数中将initid-> maybe_null设置为1并在str_ucwords中将* null_value设置为1(并且结果为NULL,尽管这可能不是必需的)来支持NULL值.实际参数为null.

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值而不进行更改,这意味着它们也应该与表列一起使用.