JinkunLiu commented on code in PR #27330:
URL: https://github.com/apache/flink/pull/27330#discussion_r2702309749
##########
flink-table/flink-table-planner/src/main/scala/org/apache/flink/table/planner/codegen/calls/ScalarOperatorGens.scala:
##########
@@ -1668,6 +1668,108 @@ object ScalarOperatorGens {
generateUnaryOperatorIfNotNull(ctx, resultType, map)(_ =>
s"${map.resultTerm}.size()")
}
+ def generateVariantGet(
+ ctx: CodeGeneratorContext,
+ variant: GeneratedExpression,
+ key: GeneratedExpression): GeneratedExpression = {
+ val Seq(resultTerm, nullTerm) = newNames(ctx, "result", "isNull")
+ val binaryVariantTerm = newName(ctx, "binaryVariantTerm")
+
+ val variantType = variant.resultType.asInstanceOf[VariantType]
+ val keyType = key.resultType
+
+ val variantTypeTerm = primitiveTypeTermForType(variantType)
+ val variantDefault = primitiveDefaultValue(variantType)
+ val variantTerm = variant.resultTerm
+ val keyTerm = key.resultTerm
+
+ val accessCode = if (isInteger(keyType)) {
+ generateIntegerKeyAccess(
+ variantTerm,
+ binaryVariantTerm,
+ keyTerm,
+ resultTerm,
+ nullTerm
+ )
+ } else if (isCharacterString(keyType)) {
+ generateCharacterStringKeyAccess(
+ variantTerm,
+ binaryVariantTerm,
+ keyTerm,
+ resultTerm,
+ nullTerm
+ )
+ } else {
+ throw new CodeGenException(s"Unsupported key type for variant: $keyType")
+ }
+
+ val finalCode =
+ s"""
+ |${variant.code}
+ |${key.code}
+ |boolean $nullTerm = (${variant.nullTerm}|| ${key.nullTerm});
+ |$variantTypeTerm $resultTerm = $variantDefault;
+ |if (!$nullTerm) {
+ | $accessCode
+ |}
+ """.stripMargin
+
+ GeneratedExpression(resultTerm, nullTerm, finalCode, variantType)
+ }
+
+ private def generateCharacterStringKeyAccess(
+ variantTerm: String,
+ binaryVariantTerm: String,
+ keyTerm: String,
+ resultTerm: String,
+ nullTerm: String): String = {
+ s"""
+ | if ($variantTerm.isObject()){
+ | if ($variantTerm instanceof $BINARY_VARIANT) {
+ | $BINARY_VARIANT $binaryVariantTerm = ($BINARY_VARIANT)
$variantTerm;
+ |
+ | String keyStr = null;
+ | if ($keyTerm instanceof $BINARY_STRING) {
Review Comment:
Thank you for the suggestion.
I have replaced `BinaryVariant` with `Variant` and improved the
implementation for obtaining fieldName.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
