[Bug][Connector-V2][ES]Fix es field type not support binary(#4240) (#…

…4274) Co-authored-by: iture123 <iture123@163.com>
apache · Mar 10, 2023 · 84f10f2 · 84f10f2
1 parent 6b7c81d
commit 84f10f2
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/...che/seatunnel/connectors/seatunnel/elasticsearch/constant/EsTypeMappingSeaTunnelType.java b/...che/seatunnel/connectors/seatunnel/elasticsearch/constant/EsTypeMappingSeaTunnelType.java
@@ -20,6 +20,8 @@
 import org.apache.seatunnel.api.table.type.BasicType;
 import org.apache.seatunnel.api.table.type.LocalTimeType;
 import org.apache.seatunnel.api.table.type.SeaTunnelDataType;
+import org.apache.seatunnel.connectors.seatunnel.elasticsearch.exception.ElasticsearchConnectorErrorCode;
+import org.apache.seatunnel.connectors.seatunnel.elasticsearch.exception.ElasticsearchConnectorException;
 
 import java.util.HashMap;
 import java.util.Map;
@@ -32,6 +34,7 @@ public class EsTypeMappingSeaTunnelType {
                     put("string", BasicType.STRING_TYPE);
                     put("keyword", BasicType.STRING_TYPE);
                     put("text", BasicType.STRING_TYPE);
+                    put("binary", BasicType.STRING_TYPE);
                     put("boolean", BasicType.BOOLEAN_TYPE);
                     put("byte", BasicType.BYTE_TYPE);
                     put("short", BasicType.SHORT_TYPE);
@@ -44,7 +47,19 @@ public class EsTypeMappingSeaTunnelType {
                 }
             };
 
+    /**
+     * if not find the mapping SeaTunnelDataType will throw runtime exception
+     *
+     * @param esType
+     * @return
+     */
     public static SeaTunnelDataType getSeaTunnelDataType(String esType) {
-        return MAPPING.get(esType);
+        SeaTunnelDataType seaTunnelDataType = MAPPING.get(esType);
+        if (seaTunnelDataType == null) {
+            throw new ElasticsearchConnectorException(
+                    ElasticsearchConnectorErrorCode.ES_FIELD_TYPE_NOT_SUPPORT,
+                    String.format("elasticsearch type is %s", esType));
+        }
+        return seaTunnelDataType;
     }
 }
diff --git a/...atunnel/connectors/seatunnel/elasticsearch/exception/ElasticsearchConnectorErrorCode.java b/...atunnel/connectors/seatunnel/elasticsearch/exception/ElasticsearchConnectorErrorCode.java
@@ -28,6 +28,7 @@ public enum ElasticsearchConnectorErrorCode implements SeaTunnelErrorCode {
     LIST_INDEX_FAILED("ELASTICSEARCH-05", "List elasticsearch index failed"),
     DROP_INDEX_FAILED("ELASTICSEARCH-06", "Drop elasticsearch index failed"),
     CREATE_INDEX_FAILED("ELASTICSEARCH-07", "Create elasticsearch index failed"),
+    ES_FIELD_TYPE_NOT_SUPPORT("ELASTICSEARCH-08", "Not support the elasticsearch field type");
     ;
 
     private final String code;