Skip to content

Commit

Permalink
Support SQLServer convert function parse (#29138)
Browse files Browse the repository at this point in the history
  • Loading branch information
@zihaoAK47
zihaoAK47 authored Nov 23, 2023
1 parent 518e2dc commit 8806813
Show file tree
Hide file tree
Showing 5 changed files with 327 additions and 4 deletions.
6 changes: 5 additions & 1 deletion parser/sql/dialect/sqlserver/src/main/antlr4/imports/sqlserver/BaseRule.g4
View file
Original file line number Diff line number Diff line change
Expand Up @@ -300,13 +300,17 @@ distinct
;

specialFunction
: castFunction | charFunction
: castFunction | charFunction | convertFunction
;

castFunction
: CAST LP_ expr AS dataType RP_
;

convertFunction
: CONVERT LP_ dataType COMMA_ expr (COMMA_ NUMBER_)? RP_
;

charFunction
: CHAR LP_ expr (COMMA_ expr)* (USING ignoredIdentifier)? RP_
;
Expand Down
2 changes: 1 addition & 1 deletion parser/sql/dialect/sqlserver/src/main/antlr4/imports/sqlserver/Literals.g4
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ IDENTIFIER_
;

DELIMITED_IDENTIFIER_
: (LBT_ | DQ_) [a-zA-Z0-9@$#_.\\/\u0080-\uFFFF ]+ (DQ_ | RBT_)
: (LBT_ | DQ_) [a-zA-Z0-9@$#_,.\\/\u0080-\uFFFF ]+ (DQ_ | RBT_)
;

STRING_
Expand Down
15 changes: 15 additions & 0 deletions ...ache/shardingsphere/sql/parser/sqlserver/visitor/statement/SQLServerStatementVisitor.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@
import org.apache.shardingsphere.sql.parser.autogen.SQLServerStatementParser.ColumnNamesContext;
import org.apache.shardingsphere.sql.parser.autogen.SQLServerStatementParser.ColumnNamesWithSortContext;
import org.apache.shardingsphere.sql.parser.autogen.SQLServerStatementParser.ConstraintNameContext;
import org.apache.shardingsphere.sql.parser.autogen.SQLServerStatementParser.ConvertFunctionContext;
import org.apache.shardingsphere.sql.parser.autogen.SQLServerStatementParser.CreateTableAsSelectClauseContext;
import org.apache.shardingsphere.sql.parser.autogen.SQLServerStatementParser.CteClauseContext;
import org.apache.shardingsphere.sql.parser.autogen.SQLServerStatementParser.CteClauseSetContext;
Expand Down Expand Up @@ -598,6 +599,9 @@ public final ASTNode visitSpecialFunction(final SpecialFunctionContext ctx) {
if (null != ctx.castFunction()) {
return visit(ctx.castFunction());
}
if (null != ctx.convertFunction()) {
return visit(ctx.convertFunction());
}
if (null != ctx.charFunction()) {
return visit(ctx.charFunction());
}
Expand All @@ -618,6 +622,17 @@ public final ASTNode visitCastFunction(final CastFunctionContext ctx) {
return result;
}

@Override
public ASTNode visitConvertFunction(final ConvertFunctionContext ctx) {
FunctionSegment result = new FunctionSegment(ctx.getStart().getStartIndex(), ctx.getStop().getStopIndex(), ctx.CONVERT().getText(), getOriginalText(ctx));
result.getParameters().add((DataTypeSegment) visit(ctx.dataType()));
result.getParameters().add((ExpressionSegment) visit(ctx.expr()));
if (null != ctx.NUMBER_()) {
result.getParameters().add(new LiteralExpressionSegment(ctx.getStart().getStartIndex(), ctx.getStop().getStopIndex(), ctx.NUMBER_().getText()));
}
return result;
}

@Override
public final ASTNode visitCharFunction(final CharFunctionContext ctx) {
calculateParameterCount(ctx.expr());
Expand Down
295 changes: 294 additions & 1 deletion test/it/parser/src/main/resources/case/dml/select.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3771,7 +3771,7 @@
</where>
</select>

<select sql-case-id="select_with_convert_function">
<select sql-case-id="select_with_convert_function1">
<from>
<simple-table name="t_order" start-index="70" stop-index="76" />
</from>
Expand All @@ -3797,6 +3797,299 @@
</where>
</select>

<select sql-case-id="select_with_convert_function2">
<projections start-index="7" stop-index="246">
<column-projection name="sequence_name" start-index="7" stop-index="19" />
<column-projection name="sequence_catalog" start-index="22" stop-index="37" />
<column-projection name="sequence_schema" start-index="40" stop-index="54" />
<expression-projection alias="start_value" text="convert( bigint, start_value )" start-index="57" stop-index="101">
<expr>
<function function-name="convert" text="convert( bigint, start_value )" start-index="57" stop-index="86">
<parameter>
<data-type value="bigint" start-index="66" stop-index="71" />
</parameter>
<parameter>
<column name="start_value" start-index="74" stop-index="84" />
</parameter>
</function>
</expr>
</expression-projection>
<expression-projection alias="minimum_value" text="convert( bigint, minimum_value )" start-index="104" stop-index="152">
<expr>
<function function-name="convert" text="convert( bigint, minimum_value )" start-index="104" stop-index="135">
<parameter>
<data-type value="bigint" start-index="113" stop-index="118" />
</parameter>
<parameter>
<column name="minimum_value" start-index="121" stop-index="133" />
</parameter>
</function>
</expr>
</expression-projection>
<expression-projection alias="maximum_value" text="convert( bigint, maximum_value )" start-index="155" stop-index="203">
<expr>
<function function-name="convert" text="convert( bigint, maximum_value )" start-index="155" stop-index="186">
<parameter>
<data-type value="bigint" start-index="164" stop-index="169" />
</parameter>
<parameter>
<column name="maximum_value" start-index="172" stop-index="184" />
</parameter>
</function>
</expr>
</expression-projection>
<expression-projection alias="increment" text="convert( bigint, increment )" start-index="206" stop-index="246">
<expr>
<function function-name="convert" text="convert( bigint, increment )" start-index="206" stop-index="233">
<parameter>
<data-type value="bigint" start-index="215" stop-index="220" />
</parameter>
<parameter>
<column name="increment" start-index="223" stop-index="231" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
<from>
<simple-table name="SEQUENCES" start-index="253" stop-index="280">
<owner name="INFORMATION_SCHEMA" start-index="253" stop-index="270" />
</simple-table>
</from>
</select>

<select sql-case-id="select_with_convert_function3">
<projections start-index="7" stop-index="37">
<expression-projection text="CONVERT(NVARCHAR, GETDATE(), 0)" start-index="7" stop-index="37">
<expr>
<function function-name="CONVERT" text="CONVERT(NVARCHAR, GETDATE(), 0)" start-index="7" stop-index="37">
<parameter>
<data-type value="NVARCHAR" start-index="15" stop-index="22" />
</parameter>
<parameter>
<function function-name="GETDATE" text="GETDATE()" start-index="25" stop-index="33" />
</parameter>
<parameter>
<literal-expression value="0" start-index="7" stop-index="37" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_convert_function4">
<projections start-index="7" stop-index="61">
<expression-projection text="CONVERT(DECIMAL(10, 5), CONVERT(VARBINARY(20), @myval))" start-index="7" stop-index="61">
<expr>
<function function-name="CONVERT" text="CONVERT(DECIMAL(10, 5), CONVERT(VARBINARY(20), @myval))" start-index="7" stop-index="61">
<parameter>
<data-type value="DECIMAL" start-index="15" stop-index="28" />
</parameter>
<parameter>
<function function-name="CONVERT" text="CONVERT(VARBINARY(20), @myval)" start-index="31" stop-index="60">
<parameter>
<data-type value="VARBINARY" start-index="39" stop-index="51" />
</parameter>
<parameter>
<column name="@myval" start-index="54" stop-index="59" />
</parameter>
</function>
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_convert_function5">
<projections start-index="7" stop-index="69">
<expression-projection alias="Style 0, character to binary" text="CONVERT(BINARY(8), 'Name', 0)" start-index="7" stop-index="69">
<expr>
<function function-name="CONVERT" text="CONVERT(BINARY(8), 'Name', 0)" start-index="7" stop-index="35">
<parameter>
<data-type value="BINARY" start-index="15" stop-index="23" />
</parameter>
<parameter>
<literal-expression value="Name" start-index="26" stop-index="31" />
</parameter>
<parameter>
<literal-expression value="0" start-index="7" stop-index="35" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_convert_xml1">
<projections start-index="7" stop-index="43">
<expression-projection text="CONVERT(XML, '&lt;root&gt;&lt;child/&gt;&lt;/root&gt;')" start-index="7" stop-index="43">
<expr>
<function function-name="CONVERT" text="CONVERT(XML, '&lt;root&gt;&lt;child/&gt;&lt;/root&gt;')" start-index="7" stop-index="43">
<parameter>
<data-type value="XML" start-index="15" stop-index="17" />
</parameter>
<parameter>
<literal-expression value="&lt;root&gt;&lt;child/&gt;&lt;/root&gt;" start-index="20" stop-index="42" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_convert_xml2">
<projections start-index="7" stop-index="65">
<expression-projection text="CONVERT(XML, '&lt;root&gt; &lt;child/&gt; &lt;/root&gt;', 1)" start-index="7" stop-index="65">
<expr>
<function function-name="CONVERT" text="CONVERT(XML, '&lt;root&gt; &lt;child/&gt; &lt;/root&gt;', 1)" start-index="7" stop-index="65">
<parameter>
<data-type value="XML" start-index="15" stop-index="17" />
</parameter>
<parameter>
<literal-expression value="&lt;root&gt; &lt;child/&gt; &lt;/root&gt;" start-index="20" stop-index="61" />
</parameter>
<parameter>
<literal-expression value="1" start-index="7" stop-index="65" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_cast_convert_function">
<projections start-index="7" stop-index="149">
<expression-projection alias="UnconvertedDateTime" text="GETDATE()" start-index="7" stop-index="38">
<expr>
<function function-name="GETDATE" text="GETDATE()" start-index="7" stop-index="15" />
</expr>
</expression-projection>
<expression-projection alias="UsingCast" text="CAST(GETDATE() AS NVARCHAR(30))" start-index="41" stop-index="84">
<expr>
<function function-name="CAST" text="CAST(GETDATE() AS NVARCHAR(30))" start-index="41" stop-index="71">
<parameter>
<data-type value="NVARCHAR" start-index="59" stop-index="70" />
</parameter>
</function>
</expr>
</expression-projection>
<expression-projection alias="UsingConvertTo_ISO8601" text="CONVERT(NVARCHAR(30), GETDATE(), 126)" start-index="87" stop-index="149">
<expr>
<function function-name="CONVERT" text="CONVERT(NVARCHAR(30), GETDATE(), 126)" start-index="87" stop-index="123">
<parameter>
<data-type value="NVARCHAR" start-index="95" stop-index="106" />
</parameter>
<parameter>
<function function-name="GETDATE" text="GETDATE()" start-index="109" stop-index="117" />
</parameter>
<parameter>
<literal-expression value="126" start-index="87" stop-index="123" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_cast_convert_datetime">
<projections start-index="7" stop-index="187">
<expression-projection alias="UnconvertedText" text="2006-04-25T15:50:59.997" start-index="7" stop-index="50">
<expr>
<literal-expression value="2006-04-25T15:50:59.997" start-index="7" stop-index="31" />
</expr>
</expression-projection>
<expression-projection alias="UsingCast" text="CAST('2006-04-25T15:50:59.997' AS DATETIME)" start-index="53" stop-index="108">
<expr>
<function function-name="CAST" text="CAST('2006-04-25T15:50:59.997' AS DATETIME)" start-index="53" stop-index="95">
<parameter>
<literal-expression value="2006-04-25T15:50:59.997" start-index="58" stop-index="82" />
</parameter>
<parameter>
<data-type value="DATETIME" start-index="87" stop-index="94" />
</parameter>
</function>
</expr>
</expression-projection>
<expression-projection alias="UsingConvertFrom_ISO8601" text="CONVERT(DATETIME, '2006-04-25T15:50:59.997', 126)" start-index="111" stop-index="187">
<expr>
<function function-name="CONVERT" text="CONVERT(DATETIME, '2006-04-25T15:50:59.997', 126)" start-index="111" stop-index="159">
<parameter>
<data-type value="DATETIME" start-index="119" stop-index="126" />
</parameter>
<parameter>
<literal-expression value="2006-04-25T15:50:59.997" start-index="129" stop-index="153" />
</parameter>
<parameter>
<literal-expression value="126" start-index="111" stop-index="159" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_convert_hex1">
<projections start-index="7" stop-index="71">
<expression-projection alias="Style 0, binary to character" text="CONVERT(CHAR(8), 0x4E616d65, 0)" start-index="7" stop-index="71">
<expr>
<function function-name="CONVERT" text="CONVERT(CHAR(8), 0x4E616d65, 0)" start-index="7" stop-index="37">
<parameter>
<data-type value="CHAR" start-index="15" stop-index="21" />
</parameter>
<parameter>
<common-expression literal-text="0x4E616d65" start-index="24" stop-index="33" />
</parameter>
<parameter>
<literal-expression value="0" start-index="7" stop-index="37" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_convert_hex2">
<projections start-index="7" stop-index="71">
<expression-projection alias="Style 1, binary to character" text="CONVERT(CHAR(8), 0x4E616d65, 1)" start-index="7" stop-index="71">
<expr>
<function function-name="CONVERT" text="CONVERT(CHAR(8), 0x4E616d65, 1)" start-index="7" stop-index="37">
<parameter>
<data-type value="CHAR" start-index="15" stop-index="21" />
</parameter>
<parameter>
<common-expression literal-text="0x4E616d65" start-index="24" stop-index="33" />
</parameter>
<parameter>
<literal-expression value="1" start-index="7" stop-index="37" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_convert_hex3">
<projections start-index="7" stop-index="75">
<expression-projection alias="Style 1, character to binary" text="CONVERT(BINARY(4), '0x4E616D65', 1)" start-index="7" stop-index="75">
<expr>
<function function-name="CONVERT" text="CONVERT(BINARY(4), '0x4E616D65', 1)" start-index="7" stop-index="41">
<parameter>
<data-type value="BINARY" start-index="15" stop-index="23" />
</parameter>
<parameter>
<literal-expression value="0x4E616D65" start-index="26" stop-index="37" />
</parameter>
<parameter>
<literal-expression value="1" start-index="7" stop-index="41" />
</parameter>
</function>
</expr>
</expression-projection>
</projections>
</select>

<select sql-case-id="select_with_json_extract">
<from>
<simple-table name="tb_content_json" start-index="70" stop-index="84" />
Expand Down
Loading

0 comments on commit 8806813

Please sign in to comment.