forked from opensearch-project/sql
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into feature/json-extract
Signed-off-by: Kenrick Yap <[email protected]> # Conflicts: # core/src/main/java/org/opensearch/sql/expression/DSL.java # core/src/main/java/org/opensearch/sql/expression/function/BuiltinFunctionName.java # core/src/main/java/org/opensearch/sql/expression/json/JsonFunctions.java # core/src/main/java/org/opensearch/sql/utils/JsonUtils.java # core/src/test/java/org/opensearch/sql/expression/json/JsonFunctionsTest.java # docs/user/ppl/functions/json.rst # integ-test/src/test/java/org/opensearch/sql/ppl/JsonFunctionsIT.java # integ-test/src/test/resources/json_test.json # ppl/src/main/antlr/OpenSearchPPLLexer.g4 # ppl/src/main/antlr/OpenSearchPPLParser.g4
- Loading branch information
Showing
79 changed files
with
600 additions
and
2,548 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
34 changes: 34 additions & 0 deletions
34
async-query-core/src/test/java/org/opensearch/sql/spark/data/type/SparkDataTypeTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,34 @@ | ||
| /* | ||
| * Copyright OpenSearch Contributors | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| */ | ||
|
|
||
| package org.opensearch.sql.spark.data.type; | ||
|
|
||
| import static org.junit.jupiter.api.Assertions.*; | ||
|
|
||
| import org.junit.jupiter.api.Test; | ||
|
|
||
| class SparkDataTypeTest { | ||
|
|
||
| @Test | ||
| void testTypeName() { | ||
| String expectedTypeName = "spark_string"; | ||
| SparkDataType sparkDataType = new SparkDataType(expectedTypeName); | ||
|
|
||
| assertEquals( | ||
| expectedTypeName, sparkDataType.typeName(), "Type name should match the expected value"); | ||
| } | ||
|
|
||
| @Test | ||
| void testEqualsAndHashCode() { | ||
| SparkDataType type1 = new SparkDataType("spark_integer"); | ||
| SparkDataType type2 = new SparkDataType("spark_integer"); | ||
| SparkDataType type3 = new SparkDataType("spark_double"); | ||
|
|
||
| assertEquals(type1, type2); | ||
| assertNotEquals(type1, type3); | ||
| assertEquals(type1.hashCode(), type2.hashCode()); | ||
| assertNotEquals(type1.hashCode(), type3.hashCode()); | ||
| } | ||
| } |
40 changes: 40 additions & 0 deletions
40
async-query-core/src/test/java/org/opensearch/sql/spark/data/value/SparkExprValueTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,40 @@ | ||
| /* | ||
| * Copyright OpenSearch Contributors | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| */ | ||
|
|
||
| package org.opensearch.sql.spark.data.value; | ||
|
|
||
| import static org.junit.jupiter.api.Assertions.*; | ||
|
|
||
| import org.junit.jupiter.api.Test; | ||
| import org.opensearch.sql.spark.data.type.SparkDataType; | ||
|
|
||
| class SparkExprValueTest { | ||
| private final SparkDataType sparkDataType = new SparkDataType("char"); | ||
|
|
||
| @Test | ||
| public void getters() { | ||
| SparkExprValue sparkExprValue = new SparkExprValue(sparkDataType, "str"); | ||
|
|
||
| assertEquals(sparkDataType, sparkExprValue.type()); | ||
| assertEquals("str", sparkExprValue.value()); | ||
| } | ||
|
|
||
| @Test | ||
| public void unsupportedCompare() { | ||
| SparkExprValue sparkExprValue = new SparkExprValue(sparkDataType, "str"); | ||
|
|
||
| assertThrows(UnsupportedOperationException.class, () -> sparkExprValue.compare(sparkExprValue)); | ||
| } | ||
|
|
||
| @Test | ||
| public void testEquals() { | ||
| SparkExprValue sparkExprValue1 = new SparkExprValue(sparkDataType, "str"); | ||
| SparkExprValue sparkExprValue2 = new SparkExprValue(sparkDataType, "str"); | ||
| SparkExprValue sparkExprValue3 = new SparkExprValue(sparkDataType, "other"); | ||
|
|
||
| assertTrue(sparkExprValue1.equal(sparkExprValue2)); | ||
| assertFalse(sparkExprValue1.equal(sparkExprValue3)); | ||
| } | ||
| } |
63 changes: 63 additions & 0 deletions
63
...rg/opensearch/sql/spark/functions/response/DefaultSparkSqlFunctionResponseHandleTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,63 @@ | ||
| /* | ||
| * Copyright OpenSearch Contributors | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| */ | ||
|
|
||
| package org.opensearch.sql.spark.functions.response; | ||
|
|
||
| import static org.junit.jupiter.api.Assertions.*; | ||
|
|
||
| import java.net.URL; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.Paths; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import org.json.JSONObject; | ||
| import org.junit.jupiter.api.Test; | ||
| import org.opensearch.sql.data.model.ExprBooleanValue; | ||
| import org.opensearch.sql.data.model.ExprByteValue; | ||
| import org.opensearch.sql.data.model.ExprDateValue; | ||
| import org.opensearch.sql.data.model.ExprDoubleValue; | ||
| import org.opensearch.sql.data.model.ExprFloatValue; | ||
| import org.opensearch.sql.data.model.ExprIntegerValue; | ||
| import org.opensearch.sql.data.model.ExprLongValue; | ||
| import org.opensearch.sql.data.model.ExprShortValue; | ||
| import org.opensearch.sql.data.model.ExprStringValue; | ||
| import org.opensearch.sql.data.model.ExprTimestampValue; | ||
| import org.opensearch.sql.data.model.ExprValue; | ||
| import org.opensearch.sql.executor.ExecutionEngine; | ||
| import org.opensearch.sql.executor.ExecutionEngine.Schema.Column; | ||
|
|
||
| class DefaultSparkSqlFunctionResponseHandleTest { | ||
|
|
||
| @Test | ||
| public void testConstruct() throws Exception { | ||
| DefaultSparkSqlFunctionResponseHandle handle = | ||
| new DefaultSparkSqlFunctionResponseHandle(readJson()); | ||
|
|
||
| assertTrue(handle.hasNext()); | ||
| ExprValue value = handle.next(); | ||
| Map<String, ExprValue> row = value.tupleValue(); | ||
| assertEquals(ExprBooleanValue.of(true), row.get("col1")); | ||
| assertEquals(new ExprLongValue(2), row.get("col2")); | ||
| assertEquals(new ExprIntegerValue(3), row.get("col3")); | ||
| assertEquals(new ExprShortValue(4), row.get("col4")); | ||
| assertEquals(new ExprByteValue(5), row.get("col5")); | ||
| assertEquals(new ExprDoubleValue(6.1), row.get("col6")); | ||
| assertEquals(new ExprFloatValue(7.1), row.get("col7")); | ||
| assertEquals(new ExprTimestampValue("2024-01-02 03:04:05.1234"), row.get("col8")); | ||
| assertEquals(new ExprDateValue("2024-01-03 04:05:06.1234"), row.get("col9")); | ||
| assertEquals(new ExprStringValue("some string"), row.get("col10")); | ||
|
|
||
| ExecutionEngine.Schema schema = handle.schema(); | ||
| List<Column> columns = schema.getColumns(); | ||
| assertEquals("col1", columns.get(0).getName()); | ||
| } | ||
|
|
||
| private JSONObject readJson() throws Exception { | ||
| final URL url = | ||
| DefaultSparkSqlFunctionResponseHandle.class.getResource( | ||
| "/spark_execution_result_test.json"); | ||
| return new JSONObject(Files.readString(Paths.get(url.toURI()))); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.