From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from localhost (localhost [127.0.0.1]) by turing.freelists.org (Avenir Technologies Mail Multiplex) with ESMTP id 9993229050 for ; Mon, 22 Apr 2019 12:45:35 -0400 (EDT) Received: from turing.freelists.org ([127.0.0.1]) by localhost (turing.freelists.org [127.0.0.1]) (amavisd-new, port 10024) with ESMTP id zmK2O9WSvEU0 for ; Mon, 22 Apr 2019 12:45:35 -0400 (EDT) Received: from smtpng3.m.smailru.net (smtpng3.m.smailru.net [94.100.177.149]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by turing.freelists.org (Avenir Technologies Mail Multiplex) with ESMTPS id 93F7F29027 for ; Mon, 22 Apr 2019 12:45:34 -0400 (EDT) Content-Type: text/plain; charset=utf-8 Mime-Version: 1.0 (Mac OS X Mail 12.2 \(3445.102.3\)) Subject: [tarantool-patches] Re: [PATCH] sql: modify TRIM() function signature From: Roman Khabibov In-Reply-To: Date: Mon, 22 Apr 2019 19:45:31 +0300 Content-Transfer-Encoding: quoted-printable Message-Id: References: <20190411173326.36120-1-roman.habibov@tarantool.org> <07e149a4-679a-53f1-ccf9-78219cc0ec47@tarantool.org> <49c88b8e-95ff-1a37-9ec3-fc16597891e3@tarantool.org> <16D82235-D881-4AC2-AB80-E12344E33CD7@tarantool.org> <2f3fd95d-228e-4d76-e067-8297bd91afb4@tarantool.org> <14F67DBB-9F0B-4965-A5A2-04E639614265@tarantool.org> Sender: tarantool-patches-bounce@freelists.org Errors-to: tarantool-patches-bounce@freelists.org Reply-To: tarantool-patches@freelists.org List-Help: List-Unsubscribe: List-software: Ecartis version 1.0.0 List-Id: tarantool-patches List-Subscribe: List-Owner: List-post: List-Archive: To: tarantool-patches@freelists.org Cc: Vladislav Shpilevoy Hi! Thanks for the review. > On Apr 21, 2019, at 10:36 PM, Vladislav Shpilevoy = wrote: >=20 > Hi! Thanks for the patch! See 8 comments below. >=20 >>> 4. You again ignored my comment about NULL. Please, find all other >>> places and fix it finally. I said it already 1000 times in 1000 >>> reviews - we do not use 0 for pointers. It is a simple rule. Just >>> follow it. Write it down somewhere in a list of code style rules >>> and check them all before sending a patch. >>>=20 >>> Seeing how many my comments you repeatedly ignore, I think that >>> probably you should reconsider the way how you do self-reviews. If >>> you do it via just looking a couple of seconds at the code in the >>> text editor, then it is definitely a bad way. >>>=20 >>> First of all, use 'git diff/show' in console to look only at the >>> patch changes, not at the entire files and functions. If you do not >>> like console, and it is ok, then you can use Sublime Merge >>> desktop program or Sublime Git package for the editor. When you >>> look at the diff only, it is much simpler to notice such violations >>> and even bugs. >> + int char_cnt =3D sql_utf8_char_count(z, trim_set_sz); >> + if (char_cnt =3D=3D 0) >>=20 >> + unsigned char **ind_chars =3D >> + contextMalloc(context, >> + char_cnt * (sizeof(unsigned char *) + 1)); >> + if (ind_chars =3D=3D NULL) >>=20 >> + int i =3D 0; >> + char_cnt =3D 0; >>=20 >> + if ((input_str =3D sql_value_text(argv[0])) =3D=3D NULL) >> + return; >>=20 >> + if ((input_str =3D sql_value_text(argv[1])) =3D=3D NULL) >> + return; >>=20 >> + } else if ((trim_set =3D sql_value_text(argv[0])) !=3D NULL) { >> + int trim_set_sz =3D sql_value_bytes(argv[0]); >>=20 >> + if ((input_str =3D sql_value_text(argv[2])) =3D=3D NULL || >> + (trim_set =3D sql_value_text(argv[1])) =3D=3D NULL) >> + return; >=20 > 1. Sorry, but I do not understand this bunch of '+' and random lines > above. What is it? It is not a diff - I do not see function names, > '-', line numbers. Please, next time provide *full* normal diff > obtained by 'git --no-pager diff' command. >=20 > The same about all other 'diff's below. Understood. >>=20 >>> 5. In our code style we do not use 'char' to represent numbers, we >>> use 'uint8_t' or 'int8_t' when we want to use one-byte numbers. It >>> is the same as 'char'/'unsigned char', but looks shorter and it >>> becomes obvious that these values are used as numbers, not text. >>> Firstly I thought that char_len was an array of characters, but >>> it emerged being an array of symbol sizes. In the summary, I >>> suggest to use 'uint8_t *' for char_len array. >> + uint8_t *char_len =3D (uint8_t *)&ind_chars[char_cnt]; >>=20 >>>> + /* Individual characters in the character set. */ >>>> + char unsigned **ind_chars =3D 0; >>>=20 >>> 6. If you declare it as 'const char unsigned **', then you >>> can remove unnecessary type cast from line 1330. >=20 >> If you meant the following line "ind_chars[char_cnt] =3D (unsigned = char *)(z + i);=E2=80=9D >> then it isn=E2=80=99t compiled without the cast, because of assigning = to "'unsigned char *=E2=80=99 >> from 'const unsigned char *=E2=80=99=E2=80=9D. >=20 > In this concrete case I think 'goto's are ok. >=20 >>>> + */ >>>> +static void >>>> +trim_func_one_arg(sql_context * context, int argc, sql_value = **argv) >>>=20 >>> 13. In new code we use explicit 'struct' keyword for struct >>> types - sql_context and sql_value. Also, we do not put whitepaces > 6. How is it possible, that first time sql_utf8_char_count() > returned not 0, but now we got 0 on the same string using the > same methods (SQL_UTF8_FWD_1)? Has it disappeared with your diff? >=20 > 7. We never declare labels like that. Please, look at other places > where and how we declare labels. Also it is stated that we use > Linux Kernel code style, as I remember, and it means, that you > can lookup doubtful places on sites like this: > = https://www.kernel.org/doc/html/v4.10/process/coding-style.html#centralize= d-exiting-of-functions diff --git a/src/box/sql/func.c b/src/box/sql/func.c index abeecefa1..0ec359425 100644 --- a/src/box/sql/func.c +++ b/src/box/sql/func.c @@ -1286,108 +1286,198 @@ replaceFunc(sql_context * context, int argc, = sql_value ** argv) sql_result_text(context, (char *)zOut, j, sql_free); } =20 -/* - * Implementation of the TRIM(), LTRIM(), and RTRIM() functions. - * The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both. +/** + * Remove characters included in @a trim_set from @a input_str + * until encounter a character that doesn't belong to @a trim_set. + * Remove from the side specified by @a flags. + * @param context SQL context. + * @param flags Trim specification: left, right or both. + * @param trim_set The set of characters for trimming. + * @param char_len Lengths of each UTF-8 character in @a trim_set. + * @param char_cnt A number of UTF-8 characters in @a trim_set. + * @param input_str Input string for trimming. + * @param input_str_sz Input string size in bytes. */ static void -trimFunc(sql_context * context, int argc, sql_value ** argv) +trim_procedure(struct sql_context *context, enum trim_side_mask flags, + const unsigned char *trim_set, uint8_t *char_len, int = char_cnt, + const unsigned char *input_str, int input_str_sz) { - const unsigned char *zIn; /* Input string */ - const unsigned char *zCharSet; /* Set of characters to trim */ - int nIn; /* Number of bytes in input */ - int flags; /* 1: trimleft 2: trimright 3: trim */ - int i; /* Loop counter */ - unsigned char *aLen =3D 0; /* Length of each character in = zCharSet */ - unsigned char **azChar =3D 0; /* Individual characters in = zCharSet */ - int nChar; /* Number of characters in zCharSet */ - - if (sql_value_type(argv[0]) =3D=3D SQL_NULL) { - return; - } - zIn =3D sql_value_text(argv[0]); - if (zIn =3D=3D 0) - return; - nIn =3D sql_value_bytes(argv[0]); - assert(zIn =3D=3D sql_value_text(argv[0])); - if (argc =3D=3D 1) { - static const unsigned char lenOne[] =3D { 1 }; - static unsigned char *const azOne[] =3D { (u8 *) " " }; - nChar =3D 1; - aLen =3D (u8 *) lenOne; - azChar =3D (unsigned char **)azOne; - zCharSet =3D 0; - } else if ((zCharSet =3D sql_value_text(argv[1])) =3D=3D 0) { - return; - } else { - const unsigned char *z =3D zCharSet; - int trim_set_sz =3D sql_value_bytes(argv[1]); - /* - * Count the number of UTF-8 characters passing - * through the entire char set, but not up - * to the '\0' or X'00' character. This allows - * to handle trimming set containing such - * characters. - */ - nChar =3D sql_utf8_char_count(z, trim_set_sz); - if (nChar > 0) { - azChar =3D - contextMalloc(context, - ((i64) nChar) * (sizeof(char = *) + 1)); - if (azChar =3D=3D 0) { - return; - } - aLen =3D (unsigned char *)&azChar[nChar]; - z =3D zCharSet; - i =3D 0; - nChar =3D 0; - int handled_bytes_cnt =3D trim_set_sz; - while(handled_bytes_cnt > 0) { - azChar[nChar] =3D (unsigned char *)(z + = i); - SQL_UTF8_FWD_1(z, i, trim_set_sz); - aLen[nChar] =3D (u8) (z + i - = azChar[nChar]); - handled_bytes_cnt -=3D aLen[nChar]; - nChar++; - } - } - } - if (nChar > 0) { - flags =3D SQL_PTR_TO_INT(sql_user_data(context)); - if (flags & 1) { - while (nIn > 0) { - int len =3D 0; - for (i =3D 0; i < nChar; i++) { - len =3D aLen[i]; - if (len <=3D nIn - && memcmp(zIn, azChar[i], = len) =3D=3D 0) - break; - } - if (i >=3D nChar) + if (char_cnt =3D=3D 0) + goto finish; + int i =3D 0; + const unsigned char *z =3D NULL; + if ((flags & TRIM_LEADING) !=3D 0) { + while (input_str_sz > 0) { + int len =3D 0; + z =3D trim_set; + for (i =3D 0; i < char_cnt; ++i, z +=3D len) { + len =3D char_len[i]; + if (len <=3D input_str_sz + && memcmp(input_str, z, len) =3D=3D = 0) break; - zIn +=3D len; - nIn -=3D len; } + if (i >=3D char_cnt) + break; + input_str +=3D len; + input_str_sz -=3D len; } - if (flags & 2) { - while (nIn > 0) { - int len =3D 0; - for (i =3D 0; i < nChar; i++) { - len =3D aLen[i]; - if (len <=3D nIn - && memcmp(&zIn[nIn - len], - azChar[i], len) =3D=3D= 0) - break; - } - if (i >=3D nChar) + } + if ((flags & TRIM_TRAILING) !=3D 0) { + while (input_str_sz > 0) { + int len =3D 0; + z =3D trim_set; + for (i =3D 0; i < char_cnt; ++i, z +=3D len) { + len =3D char_len[i]; + if (len <=3D input_str_sz + && memcmp(&input_str[input_str_sz - = len], + z, len) =3D=3D 0) break; - nIn -=3D len; } + if (i >=3D char_cnt) + break; + input_str_sz -=3D len; } - if (zCharSet) { - sql_free(azChar); - } } - sql_result_text(context, (char *)zIn, nIn, SQL_TRANSIENT); + +finish: + sql_result_text(context, (char *)input_str, input_str_sz, + SQL_TRANSIENT); +} >=20 > 8. Now the function looks much more clean, and it is easy to see that > its performance for the most common case is strongly down. Before > your patch when there was no 'trim character set' argument, = whitespaces > where used by default, but they were not allocated on new memory via > contextMalloc() - they were declared on stack as an optimization. = Please, > refactor that function so as it does not allocate anything and takes > already somehow obtained trim char set and trim char lengths. >=20 > These arguments should only be allocated by trim_func_two_args() and > trim_func_three_args(), and only on demand, when not-default trim set > is passed. And obviously, you should not do it twice in both > places. Create a special function for that which takes > 'const unsigned char *trim_set', and returns the prepared arguments. +/** + * Prepare arguments for trimming procedure. Allocate memory for + * @a char_len (array of lengths each character in @a trim_set) + * and fill it. + * + * @param context SQL context. + * @param trim_set The set of characters for trimming. + * @param[out] char_len Lengths of each character in @ trim_set. + * @retval >=3D0 A number of UTF-8 characters in @a trim_set. + * @retval -1 Memory allocation error. + */ +static int +prepare_char_len(struct sql_context *context, const unsigned char = *trim_set, + int trim_set_sz, uint8_t **char_len) +{ + /* + * Count the number of UTF-8 characters passing through + * the entire char set, but not up to the '\0' or X'00' + * character. This allows to handle trimming set + * containing such characters. + */ + int char_cnt =3D sql_utf8_char_count(trim_set, trim_set_sz); + if (char_cnt =3D=3D 0) + return 0; + + if ((*char_len =3D (uint8_t *)contextMalloc(context, char_cnt)) = =3D=3D NULL) + return -1; + + const unsigned char *z =3D trim_set; + int i =3D 0, j =3D 0; + while(j < char_cnt) { + int old_i =3D i; + SQL_UTF8_FWD_1(z, i, trim_set_sz); + (*char_len)[j++] =3D i - old_i; + } + + return char_cnt; +} commit c35e8fd261ca6ab41fbe5e48935923314bef014a Author: Roman Khabibov Date: Thu Mar 28 14:01:33 2019 +0300 sql: modify TRIM() function signature =20 According to the ANSI standard, ltrim, rtrim and trim should be merged into one unified TRIM() function. The specialization of trimming (left, right or both and trimming characters) determined in arguments of this function. =20 Closes #3879 diff --git a/extra/mkkeywordhash.c b/extra/mkkeywordhash.c index be7bd5545..76e3265e7 100644 --- a/extra/mkkeywordhash.c +++ b/extra/mkkeywordhash.c @@ -278,6 +278,10 @@ static Keyword aKeywordTable[] =3D { { "WHILE", "TK_STANDARD", RESERVED, true = }, { "TEXT", "TK_TEXT", RESERVED, true = }, { "TRUNCATE", "TK_TRUNCATE", ALWAYS, true = }, + { "TRIM", "TK_TRIM", ALWAYS, true = }, + { "LEADING", "TK_LEADING", ALWAYS, true = }, + { "TRAILING", "TK_TRAILING", ALWAYS, true = }, + { "BOTH", "TK_BOTH", ALWAYS, true = }, }; =20 /* Number of keywords */ diff --git a/src/box/sql/func.c b/src/box/sql/func.c index abeecefa1..0ec359425 100644 --- a/src/box/sql/func.c +++ b/src/box/sql/func.c @@ -1286,108 +1286,198 @@ replaceFunc(sql_context * context, int argc, = sql_value ** argv) sql_result_text(context, (char *)zOut, j, sql_free); } =20 -/* - * Implementation of the TRIM(), LTRIM(), and RTRIM() functions. - * The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both. +/** + * Remove characters included in @a trim_set from @a input_str + * until encounter a character that doesn't belong to @a trim_set. + * Remove from the side specified by @a flags. + * @param context SQL context. + * @param flags Trim specification: left, right or both. + * @param trim_set The set of characters for trimming. + * @param char_len Lengths of each UTF-8 character in @a trim_set. + * @param char_cnt A number of UTF-8 characters in @a trim_set. + * @param input_str Input string for trimming. + * @param input_str_sz Input string size in bytes. */ static void -trimFunc(sql_context * context, int argc, sql_value ** argv) +trim_procedure(struct sql_context *context, enum trim_side_mask flags, + const unsigned char *trim_set, uint8_t *char_len, int = char_cnt, + const unsigned char *input_str, int input_str_sz) { - const unsigned char *zIn; /* Input string */ - const unsigned char *zCharSet; /* Set of characters to trim */ - int nIn; /* Number of bytes in input */ - int flags; /* 1: trimleft 2: trimright 3: trim */ - int i; /* Loop counter */ - unsigned char *aLen =3D 0; /* Length of each character in = zCharSet */ - unsigned char **azChar =3D 0; /* Individual characters in = zCharSet */ - int nChar; /* Number of characters in zCharSet */ - - if (sql_value_type(argv[0]) =3D=3D SQL_NULL) { - return; - } - zIn =3D sql_value_text(argv[0]); - if (zIn =3D=3D 0) - return; - nIn =3D sql_value_bytes(argv[0]); - assert(zIn =3D=3D sql_value_text(argv[0])); - if (argc =3D=3D 1) { - static const unsigned char lenOne[] =3D { 1 }; - static unsigned char *const azOne[] =3D { (u8 *) " " }; - nChar =3D 1; - aLen =3D (u8 *) lenOne; - azChar =3D (unsigned char **)azOne; - zCharSet =3D 0; - } else if ((zCharSet =3D sql_value_text(argv[1])) =3D=3D 0) { - return; - } else { - const unsigned char *z =3D zCharSet; - int trim_set_sz =3D sql_value_bytes(argv[1]); - /* - * Count the number of UTF-8 characters passing - * through the entire char set, but not up - * to the '\0' or X'00' character. This allows - * to handle trimming set containing such - * characters. - */ - nChar =3D sql_utf8_char_count(z, trim_set_sz); - if (nChar > 0) { - azChar =3D - contextMalloc(context, - ((i64) nChar) * (sizeof(char = *) + 1)); - if (azChar =3D=3D 0) { - return; - } - aLen =3D (unsigned char *)&azChar[nChar]; - z =3D zCharSet; - i =3D 0; - nChar =3D 0; - int handled_bytes_cnt =3D trim_set_sz; - while(handled_bytes_cnt > 0) { - azChar[nChar] =3D (unsigned char *)(z + = i); - SQL_UTF8_FWD_1(z, i, trim_set_sz); - aLen[nChar] =3D (u8) (z + i - = azChar[nChar]); - handled_bytes_cnt -=3D aLen[nChar]; - nChar++; - } - } - } - if (nChar > 0) { - flags =3D SQL_PTR_TO_INT(sql_user_data(context)); - if (flags & 1) { - while (nIn > 0) { - int len =3D 0; - for (i =3D 0; i < nChar; i++) { - len =3D aLen[i]; - if (len <=3D nIn - && memcmp(zIn, azChar[i], = len) =3D=3D 0) - break; - } - if (i >=3D nChar) + if (char_cnt =3D=3D 0) + goto finish; + int i =3D 0; + const unsigned char *z =3D NULL; + if ((flags & TRIM_LEADING) !=3D 0) { + while (input_str_sz > 0) { + int len =3D 0; + z =3D trim_set; + for (i =3D 0; i < char_cnt; ++i, z +=3D len) { + len =3D char_len[i]; + if (len <=3D input_str_sz + && memcmp(input_str, z, len) =3D=3D = 0) break; - zIn +=3D len; - nIn -=3D len; } + if (i >=3D char_cnt) + break; + input_str +=3D len; + input_str_sz -=3D len; } - if (flags & 2) { - while (nIn > 0) { - int len =3D 0; - for (i =3D 0; i < nChar; i++) { - len =3D aLen[i]; - if (len <=3D nIn - && memcmp(&zIn[nIn - len], - azChar[i], len) =3D=3D= 0) - break; - } - if (i >=3D nChar) + } + if ((flags & TRIM_TRAILING) !=3D 0) { + while (input_str_sz > 0) { + int len =3D 0; + z =3D trim_set; + for (i =3D 0; i < char_cnt; ++i, z +=3D len) { + len =3D char_len[i]; + if (len <=3D input_str_sz + && memcmp(&input_str[input_str_sz - = len], + z, len) =3D=3D 0) break; - nIn -=3D len; } + if (i >=3D char_cnt) + break; + input_str_sz -=3D len; } - if (zCharSet) { - sql_free(azChar); - } } - sql_result_text(context, (char *)zIn, nIn, SQL_TRANSIENT); + +finish: + sql_result_text(context, (char *)input_str, input_str_sz, + SQL_TRANSIENT); +} + +/** + * Prepare arguments for trimming procedure. Allocate memory for + * @a char_len (array of lengths each character in @a trim_set) + * and fill it. + * + * @param context SQL context. + * @param trim_set The set of characters for trimming. + * @param[out] char_len Lengths of each character in @ trim_set. + * @retval >=3D0 A number of UTF-8 characters in @a trim_set. + * @retval -1 Memory allocation error. + */ +static int +prepare_char_len(struct sql_context *context, const unsigned char = *trim_set, + int trim_set_sz, uint8_t **char_len) +{ + /* + * Count the number of UTF-8 characters passing through + * the entire char set, but not up to the '\0' or X'00' + * character. This allows to handle trimming set + * containing such characters. + */ + int char_cnt =3D sql_utf8_char_count(trim_set, trim_set_sz); + if (char_cnt =3D=3D 0) + return 0; + + if ((*char_len =3D (uint8_t *)contextMalloc(context, char_cnt)) = =3D=3D NULL) + return -1; + + const unsigned char *z =3D trim_set; + int i =3D 0, j =3D 0; + while(j < char_cnt) { + int old_i =3D i; + SQL_UTF8_FWD_1(z, i, trim_set_sz); + (*char_len)[j++] =3D i - old_i; + } + + return char_cnt; +} + +/** + * Normalize args from @a argv input array when it has one arg + * only. + * + * Case: TRIM() + * Call trimming procedure with TRIM_BOTH as the flags and " " as + * the trimming set. + */ +static void +trim_func_one_arg(struct sql_context *context, int argc, sql_value = **argv) +{ + assert(argc =3D=3D 1); + (void) argc; + + const unsigned char *input_str; + if ((input_str =3D sql_value_text(argv[0])) =3D=3D NULL) + return; + + int input_str_sz =3D sql_value_bytes(argv[0]); + static uint8_t len_one[] =3D { 1 }; + trim_procedure(context, TRIM_BOTH, (const unsigned char *) " ", + len_one, 1, input_str, input_str_sz); +} + +/** + * Normalize args from @a argv input array when it has two args. + * + * Case: TRIM( FROM ) + * If user has specified only, call trimming + * procedure with TRIM_BOTH as the flags and that trimming set. + * + * Case: TRIM(LEADING/TRAILING/BOTH FROM ) + * If user has specified side keyword only, then call trimming + * procedure with the specified side and " " as the trimming set. + */ +static void +trim_func_two_args(struct sql_context *context, int argc, sql_value = **argv) +{ + assert(argc =3D=3D 2); + (void) argc; + + const unsigned char *input_str; + if ((input_str =3D sql_value_text(argv[1])) =3D=3D NULL) + return; + + int input_str_sz =3D sql_value_bytes(argv[1]); + const unsigned char *trim_set; + if (sql_value_type(argv[0]) =3D=3D SQL_INTEGER) { + static uint8_t len_one[] =3D { 1 }; + trim_procedure(context, sql_value_int(argv[0]), + (const unsigned char *) " ", len_one, 1, + input_str, input_str_sz); + } else if ((trim_set =3D sql_value_text(argv[0])) !=3D NULL) { + int trim_set_sz =3D sql_value_bytes(argv[0]); + uint8_t *char_len =3D NULL; + int char_cnt =3D 0; + if ((char_cnt =3D prepare_char_len(context, trim_set, = trim_set_sz, + &char_len)) =3D=3D -1) + return; + trim_procedure(context, TRIM_BOTH, trim_set, char_len, = char_cnt, + input_str, input_str_sz); + sql_free(char_len); + } +} + +/** + * Normalize args from @a argv input array when it has three args. + * + * Case: TRIM(LEADING/TRAILING/BOTH FROM ) + * If user has specified side keyword and , then + * call trimming procedure with that args. + */ +static void +trim_func_three_args(struct sql_context *context, int argc, sql_value = **argv) +{ + assert(argc =3D=3D 3); + (void) argc; + + assert(sql_value_type(argv[0]) =3D=3D SQL_INTEGER); + const unsigned char *input_str, *trim_set; + if ((input_str =3D sql_value_text(argv[2])) =3D=3D NULL || + (trim_set =3D sql_value_text(argv[1])) =3D=3D NULL) + return; + + int trim_set_sz =3D sql_value_bytes(argv[1]); + int input_str_sz =3D sql_value_bytes(argv[2]); + uint8_t *char_len =3D NULL; + int char_cnt =3D 0; + if ((char_cnt =3D prepare_char_len(context, trim_set, = trim_set_sz, + &char_len)) =3D=3D -1) + return; + trim_procedure(context, sql_value_int(argv[0]), trim_set, = char_len, + char_cnt, input_str, input_str_sz); + sql_free(char_len); } =20 #ifdef SQL_ENABLE_UNKNOWN_SQL_FUNCTION @@ -1818,12 +1908,9 @@ sqlRegisterBuiltinFunctions(void) FIELD_TYPE_INTEGER), FUNCTION2(likely, 1, 0, 0, noopFunc, SQL_FUNC_UNLIKELY, FIELD_TYPE_INTEGER), - FUNCTION_COLL(ltrim, 1, 1, 0, trimFunc), - FUNCTION_COLL(ltrim, 2, 1, 0, trimFunc), - FUNCTION_COLL(rtrim, 1, 2, 0, trimFunc), - FUNCTION_COLL(rtrim, 2, 2, 0, trimFunc), - FUNCTION_COLL(trim, 1, 3, 0, trimFunc), - FUNCTION_COLL(trim, 2, 3, 0, trimFunc), + FUNCTION_COLL(trim, 1, 3, 0, trim_func_one_arg), + FUNCTION_COLL(trim, 2, 3, 0, trim_func_two_args), + FUNCTION_COLL(trim, 3, 3, 0, trim_func_three_args), FUNCTION(min, -1, 0, 1, minmaxFunc, FIELD_TYPE_SCALAR), FUNCTION(min, 0, 0, 1, 0, FIELD_TYPE_SCALAR), AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize, diff --git a/src/box/sql/parse.y b/src/box/sql/parse.y index 099daf512..a56ce7a10 100644 --- a/src/box/sql/parse.y +++ b/src/box/sql/parse.y @@ -1032,6 +1032,55 @@ expr(A) ::=3D CAST(X) LP expr(E) AS typedef(T) = RP(Y). { sqlExprAttachSubtrees(pParse->db, A.pExpr, E.pExpr, 0); } %endif SQL_OMIT_CAST + +expr(A) ::=3D TRIM(X) LP trim_operands(Y) RP(E). { + A.pExpr =3D sqlExprFunction(pParse, Y, &X); + spanSet(&A, &X, &E); +} + +%type trim_operands {struct ExprList *} +%destructor trim_operands {sql_expr_list_delete(pParse->db, $$);} + +trim_operands(A) ::=3D trim_from_clause(F) expr(Y). { + A =3D sql_expr_list_append(pParse->db, F, Y.pExpr); +} + +trim_operands(A) ::=3D expr(Y). { + A =3D sql_expr_list_append(pParse->db, NULL, Y.pExpr); +} + +%type trim_from_clause {struct ExprList *} +%destructor trim_from_clause {sql_expr_list_delete(pParse->db, $$);} + +/* + * The following two rules cover three cases of keyword + * (LEADING/TRAILING/BOTH) and combination. + * The case when both of them are absent is disallowed. + */ +trim_from_clause(A) ::=3D expr(Y) FROM. { + A =3D sql_expr_list_append(pParse->db, NULL, Y.pExpr); +} + +trim_from_clause(A) ::=3D trim_specification(N) expr_optional(Y) FROM. = { + struct Expr *p =3D sql_expr_new_dequoted(pParse->db, TK_INTEGER, + &sqlIntTokens[N]); + A =3D sql_expr_list_append(pParse->db, NULL, p); + if (Y !=3D NULL) + A =3D sql_expr_list_append(pParse->db, A, Y); +} + +%type expr_optional {struct Expr *} +%destructor expr_optional {sql_expr_delete(pParse->db, $$, false);} + +expr_optional(A) ::=3D . { A =3D NULL; } +expr_optional(A) ::=3D expr(X). { A =3D X.pExpr; } + +%type trim_specification {enum trim_side_mask} + +trim_specification(A) ::=3D LEADING. { A =3D TRIM_LEADING; } +trim_specification(A) ::=3D TRAILING. { A =3D TRIM_TRAILING; } +trim_specification(A) ::=3D BOTH. { A =3D TRIM_BOTH; } + expr(A) ::=3D id(X) LP distinct(D) exprlist(Y) RP(E). { if( Y && Y->nExpr>pParse->db->aLimit[SQL_LIMIT_FUNCTION_ARG] ){ const char *err =3D @@ -1294,7 +1343,7 @@ expr(A) ::=3D EXISTS(B) LP select(Y) RP(E). { } =20 /* CASE expressions */ -expr(A) ::=3D CASE(C) case_operand(X) case_exprlist(Y) case_else(Z) = END(E). { +expr(A) ::=3D CASE(C) expr_optional(X) case_exprlist(Y) case_else(Z) = END(E). { spanSet(&A,&C,&E); /*A-overwrites-C*/ A.pExpr =3D sqlPExpr(pParse, TK_CASE, X, 0); if( A.pExpr ){ @@ -1319,10 +1368,6 @@ case_exprlist(A) ::=3D WHEN expr(Y) THEN expr(Z). = { %destructor case_else {sql_expr_delete(pParse->db, $$, false);} case_else(A) ::=3D ELSE expr(X). {A =3D X.pExpr;} case_else(A) ::=3D . {A =3D 0;}=20 -%type case_operand {Expr*} -%destructor case_operand {sql_expr_delete(pParse->db, $$, false);} -case_operand(A) ::=3D expr(X). {A =3D X.pExpr; = /*A-overwrites-X*/}=20 -case_operand(A) ::=3D . {A =3D 0;}=20 =20 %type exprlist {ExprList*} %destructor exprlist {sql_expr_list_delete(pParse->db, $$);} diff --git a/src/box/sql/parse_def.c b/src/box/sql/parse_def.c index 49c76a326..aa1323cb2 100644 --- a/src/box/sql/parse_def.c +++ b/src/box/sql/parse_def.c @@ -34,7 +34,9 @@ =20 const struct Token sqlIntTokens[] =3D { {"0", 1, false}, - {"1", 1, false} + {"1", 1, false}, + {"2", 1, false}, + {"3", 1, false}, }; =20 void diff --git a/src/box/sql/parse_def.h b/src/box/sql/parse_def.h index a1af2bacd..5899a7e4e 100644 --- a/src/box/sql/parse_def.h +++ b/src/box/sql/parse_def.h @@ -87,7 +87,7 @@ struct Token { bool isReserved; }; =20 -/** Constant tokens for values 0 and 1. */ +/** Constant tokens for integer values. */ extern const struct Token sqlIntTokens[]; =20 /** Generate a Token object from a string. */ diff --git a/src/box/sql/sqlInt.h b/src/box/sql/sqlInt.h index b322602dc..d5a3e15c1 100644 --- a/src/box/sql/sqlInt.h +++ b/src/box/sql/sqlInt.h @@ -1680,6 +1680,17 @@ struct FuncDestructor { * single query - might change = over time */ =20 +/* + * Trim side mask components. TRIM_LEADING means to trim left side + * only. TRIM_TRAILING is to trim right side only. TRIM_BOTH is to + * trim both sides. + */ +enum trim_side_mask { + TRIM_LEADING =3D 1, + TRIM_TRAILING =3D 2, + TRIM_BOTH =3D TRIM_LEADING | TRIM_TRAILING +}; + /* * The following three macros, FUNCTION(), LIKEFUNC() and AGGREGATE() = are * used to create the initializers for the FuncDef structures. diff --git a/test/sql-tap/badutf1.test.lua = b/test/sql-tap/badutf1.test.lua index d104efaa9..9079dfe25 100755 --- a/test/sql-tap/badutf1.test.lua +++ b/test/sql-tap/badutf1.test.lua @@ -302,7 +302,7 @@ test:do_test( test:do_test( "badutf-4.1", function() - return test:execsql2("SELECT = hex(trim('\x80\x80\x80\xf0\x80\x80\x80\xff','\x80\xff')) AS x") + return test:execsql2("SELECT hex(TRIM('\x80\xff' FROM = '\x80\x80\x80\xf0\x80\x80\x80\xff')) AS x") end, { -- "X", "F0" @@ -312,7 +312,7 @@ test:do_test( test:do_test( "badutf-4.2", function() - return test:execsql2("SELECT = hex(ltrim('\x80\x80\x80\xf0\x80\x80\x80\xff','\x80\xff')) AS x") + return test:execsql2("SELECT hex(TRIM(LEADING '\x80\xff' FROM = '\x80\x80\x80\xf0\x80\x80\x80\xff')) AS x") end, { -- "X", "F0808080FF" @@ -322,7 +322,7 @@ test:do_test( test:do_test( "badutf-4.3", function() - return test:execsql2("SELECT = hex(rtrim('\x80\x80\x80\xf0\x80\x80\x80\xff','\x80\xff')) AS x") + return test:execsql2("SELECT hex(TRIM(TRAILING '\x80\xff' FROM = '\x80\x80\x80\xf0\x80\x80\x80\xff')) AS x") end, { -- "X", "808080F0" @@ -332,7 +332,7 @@ test:do_test( test:do_test( "badutf-4.4", function() - return test:execsql2("SELECT = hex(trim('\x80\x80\x80\xf0\x80\x80\x80\xff','\xff\x80')) AS x") + return test:execsql2("SELECT hex(TRIM('\xff\x80' FROM = '\x80\x80\x80\xf0\x80\x80\x80\xff')) AS x") end, { -- "X", "808080F0808080FF" @@ -342,7 +342,7 @@ test:do_test( test:do_test( "badutf-4.5", function() - return test:execsql2("SELECT = hex(trim('\xff\x80\x80\xf0\x80\x80\x80\xff','\xff\x80')) AS x") + return test:execsql2("SELECT hex(TRIM('\xff\x80' FROM = '\xff\x80\x80\xf0\x80\x80\x80\xff')) AS x") end, { -- "X", "80F0808080FF" @@ -352,7 +352,7 @@ test:do_test( test:do_test( "badutf-4.6", function() - return test:execsql2("SELECT = hex(trim('\xff\x80\xf0\x80\x80\x80\xff','\xff\x80')) AS x") + return test:execsql2("SELECT hex(TRIM('\xff\x80' FROM = '\xff\x80\xf0\x80\x80\x80\xff')) AS x") end, { -- "X", "F0808080FF" @@ -362,7 +362,7 @@ test:do_test( test:do_test( "badutf-4.7", function() - return test:execsql2("SELECT = hex(trim('\xff\x80\xf0\x80\x80\x80\xff','\xff\x80\x80')) AS x") + return test:execsql2("SELECT hex(TRIM('\xff\x80\x80' FROM = '\xff\x80\xf0\x80\x80\x80\xff')) AS x") end, { -- "X", "FF80F0808080FF" diff --git a/test/sql-tap/func.test.lua b/test/sql-tap/func.test.lua index 251cc3534..fe9a98191 100755 --- a/test/sql-tap/func.test.lua +++ b/test/sql-tap/func.test.lua @@ -1,6 +1,6 @@ #!/usr/bin/env tarantool test =3D require("sqltester") -test:plan(14586) +test:plan(14590) =20 --!./tcltestrunner.lua -- 2001 September 15 @@ -1915,7 +1915,7 @@ test:do_catchsql_test( SELECT trim(1,2,3) ]], { -- - 1, "wrong number of arguments to function TRIM()" + 1, "Syntax error near ','" -- }) =20 @@ -1925,7 +1925,7 @@ test:do_catchsql_test( SELECT ltrim(1,2,3) ]], { -- - 1, "wrong number of arguments to function LTRIM()" + 1, "Function 'LTRIM' does not exist" -- }) =20 @@ -1935,7 +1935,7 @@ test:do_catchsql_test( SELECT rtrim(1,2,3) ]], { -- - 1, "wrong number of arguments to function RTRIM()" + 1, "Function 'RTRIM' does not exist" -- }) =20 @@ -1952,7 +1952,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.5", [[ - SELECT ltrim(' hi '); + SELECT TRIM(LEADING FROM ' hi '); ]], { -- "hi " @@ -1962,7 +1962,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.6", [[ - SELECT rtrim(' hi '); + SELECT TRIM(TRAILING FROM ' hi '); ]], { -- " hi" @@ -1972,7 +1972,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.7", [[ - SELECT trim(' hi ','xyz'); + SELECT TRIM('xyz' FROM ' hi '); ]], { -- " hi " @@ -1982,7 +1982,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.8", [[ - SELECT ltrim(' hi ','xyz'); + SELECT TRIM(LEADING 'xyz' FROM ' hi '); ]], { -- " hi " @@ -1992,7 +1992,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.9", [[ - SELECT rtrim(' hi ','xyz'); + SELECT TRIM(TRAILING 'xyz' FROM ' hi '); ]], { -- " hi " @@ -2002,7 +2002,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.10", [[ - SELECT trim('xyxzy hi zzzy','xyz'); + SELECT TRIM('xyz' FROM 'xyxzy hi zzzy'); ]], { -- " hi " @@ -2012,7 +2012,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.11", [[ - SELECT ltrim('xyxzy hi zzzy','xyz'); + SELECT TRIM(LEADING 'xyz' FROM 'xyxzy hi zzzy'); ]], { -- " hi zzzy" @@ -2022,7 +2022,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.12", [[ - SELECT rtrim('xyxzy hi zzzy','xyz'); + SELECT TRIM(TRAILING 'xyz' FROM 'xyxzy hi zzzy'); ]], { -- "xyxzy hi " @@ -2032,7 +2032,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.13", [[ - SELECT trim(' hi ',''); + SELECT TRIM('' FROM ' hi '); ]], { -- " hi " @@ -2043,7 +2043,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.14", [[ - SELECT hex(trim(x'c280e1bfbff48fbfbf6869',x'6162e1bfbfc280')) + SELECT hex(TRIM(x'6162e1bfbfc280' FROM = x'c280e1bfbff48fbfbf6869')) ]], { -- "F48FBFBF6869" @@ -2052,8 +2052,8 @@ test:do_execsql_test( =20 test:do_execsql_test( "func-22.15", - [[SELECT hex(trim(x'6869c280e1bfbff48fbfbf61', - x'6162e1bfbfc280f48fbfbf'))]], { + [[SELECT hex(TRIM(x'6162e1bfbfc280f48fbfbf' + FROM x'6869c280e1bfbff48fbfbf61'))]], { -- "6869" -- @@ -2062,7 +2062,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.16", [[ - SELECT hex(trim(x'ceb1ceb2ceb3',x'ceb1')); + SELECT hex(TRIM(x'ceb1' FROM x'ceb1ceb2ceb3')); ]], { -- "CEB2CEB3" @@ -2083,7 +2083,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.21", [[ - SELECT typeof(trim(NULL,'xyz')); + SELECT typeof(TRIM('xyz' FROM NULL)); ]], { -- "null" @@ -2093,7 +2093,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.22", [[ - SELECT typeof(trim('hello',NULL)); + SELECT typeof(TRIM(NULL FROM 'hello')); ]], { -- "null" @@ -2105,7 +2105,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.23", [[ - SELECT TRIM(X'004100', X'00'); + SELECT TRIM(X'00' FROM X'004100'); ]], { -- "A" @@ -2115,7 +2115,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.24", [[ - SELECT TRIM(X'004100', X'0000'); + SELECT TRIM(X'0000' FROM X'004100'); ]], { -- "A" @@ -2125,7 +2125,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.25", [[ - SELECT TRIM(X'004100', X'0042'); + SELECT TRIM(X'0042' FROM X'004100'); ]], { -- "A" @@ -2135,7 +2135,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.26", [[ - SELECT TRIM(X'00004100420000', X'00'); + SELECT TRIM(X'00' FROM X'00004100420000'); ]], { -- "A\0B" @@ -2145,7 +2145,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.27", [[ - SELECT LTRIM(X'004100', X'00'); + SELECT TRIM(LEADING X'00' FROM X'004100'); ]], { -- "A\0" @@ -2155,7 +2155,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.28", [[ - SELECT LTRIM(X'004100', X'0000'); + SELECT TRIM(LEADING X'0000' FROM X'004100'); ]], { -- "A\0" @@ -2165,7 +2165,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.29", [[ - SELECT LTRIM(X'004100', X'0042'); + SELECT TRIM(LEADING X'0042' FROM X'004100'); ]], { -- "A\0" @@ -2175,7 +2175,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.30", [[ - SELECT LTRIM(X'00004100420000', X'00'); + SELECT TRIM(LEADING X'00' FROM X'00004100420000'); ]], { -- "A\0B\0\0" @@ -2185,7 +2185,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.31", [[ - SELECT RTRIM(X'004100', X'00'); + SELECT TRIM(TRAILING X'00' FROM X'004100'); ]], { -- "\0A" @@ -2195,7 +2195,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.32", [[ - SELECT RTRIM(X'004100', X'0000'); + SELECT TRIM(TRAILING X'0000' FROM X'004100'); ]], { -- "\0A" @@ -2205,7 +2205,7 @@ test:do_execsql_test( test:do_execsql_test( "func-22.33", [[ - SELECT RTRIM(X'004100', X'0042'); + SELECT TRIM(TRAILING X'0042' FROM X'004100'); ]], { -- "\0A" @@ -2215,13 +2215,56 @@ test:do_execsql_test( test:do_execsql_test( "func-22.34", [[ - SELECT RTRIM(X'00004100420000', X'00'); + SELECT TRIM(TRAILING X'00' FROM X'00004100420000'); ]], { -- "\0\0A\0B" -- }) =20 +-- gh-3879 Check new TRIM() grammar, particularly BOTH keyword and +-- FROM without any agrs before. LEADING and TRAILING keywords is +-- checked above. + +test:do_execsql_test( + "func-22.35", + [[ + SELECT TRIM(BOTH FROM ' hi '); + ]], { + -- + "hi" + -- + }) +test:do_execsql_test( + "func-22.36", + [[ + SELECT TRIM(BOTH 'xyz' FROM ' hi '); + ]], { + -- + " hi " + -- + }) + +test:do_execsql_test( + "func-22.37", + [[ + SELECT TRIM(BOTH 'xyz' FROM 'xyxzy hi zzzy'); + ]], { + -- + " hi " + -- + }) + +test:do_catchsql_test( + "func-22.38", + [[ + SELECT TRIM(FROM 'xyxzy'); + ]], { + -- + 1, "Syntax error near 'FROM'" + -- + }) + -- This is to test the deprecated sql_aggregate_count() API. -- --test:do_test( @@ -2838,16 +2881,16 @@ test:do_execsql_test( "SELECT TRIM(CHAR(32,00,32,00,32));", {string.char(00,32,00)}) =20 --- LTRIM +-- LEFT TRIM test:do_execsql_test( "func-70", - "SELECT LTRIM(CHAR(32,00,32,00,32));", + "SELECT TRIM(LEADING FROM CHAR(32,00,32,00,32));", {string.char(00,32,00,32)}) =20 --- RTRIM +-- RIGHT TRIM test:do_execsql_test( "func-71", - "SELECT RTRIM(CHAR(32,00,32,00,32));", + "SELECT TRIM(TRAILING FROM CHAR(32,00,32,00,32));", {string.char(32,00,32,00)}) =20 -- GROUP_CONCAT diff --git a/test/sql-tap/with1.test.lua b/test/sql-tap/with1.test.lua index 495aa4ee4..ec45e5e76 100755 --- a/test/sql-tap/with1.test.lua +++ b/test/sql-tap/with1.test.lua @@ -550,7 +550,7 @@ test:do_execsql_test("8.1-mandelbrot", [[ SELECT group_concat( substr(' .+*#', 1+min(iter/7,4), 1), '')=20 FROM m2 GROUP BY cy ) - SELECT group_concat(rtrim(t),x'0a') FROM a; + SELECT group_concat(TRIM(TRAILING FROM t),x'0a') FROM a; ]], { -- <8.1-mandelbrot> [[ ....#