Toggle INT or BIGINT in metadata

Author: ekawinataa

src/main/java/com/gotocompany/depot/maxcompute/util/MetadataUtil.java

@@ -36,7 +36,7 @@ public MetadataUtil(MaxComputeSinkConfig maxComputeSinkConfig) {
         this.maxComputeTimestampDataType = maxComputeSinkConfig.getMaxComputeProtoTimestampToMaxcomputeType();
         this.zoneId = maxComputeSinkConfig.getZoneId();
         metadataTypeMap = ImmutableMap.<String, TypeInfo>builder()
-                .put(INTEGER, TypeInfoFactory.INT)
+                .put(INTEGER, maxComputeSinkConfig.isProtoIntegerTypesToBigintEnabled() ? TypeInfoFactory.BIGINT : TypeInfoFactory.INT)
                 .put(LONG, TypeInfoFactory.BIGINT)
                 .put(FLOAT, TypeInfoFactory.FLOAT)
                 .put(DOUBLE, TypeInfoFactory.DOUBLE)
@@ -45,7 +45,12 @@ public MetadataUtil(MaxComputeSinkConfig maxComputeSinkConfig) {
                 .put(TIMESTAMP, maxComputeTimestampDataType.getTypeInfo())
                 .build();
         metadataMapperMap = ImmutableMap.<String, Function<Object, Object>>builder()
-                .put(INTEGER, obj -> ((Number) obj).intValue())
+                .put(INTEGER, obj -> {
+                    if (maxComputeSinkConfig.isProtoIntegerTypesToBigintEnabled()) {
+                        return ((Number) obj).longValue();
+                    }
+                    return ((Number) obj).intValue();
+                })
                 .put(LONG, obj -> ((Number) obj).longValue())
                 .put(FLOAT, obj -> ((Number) obj).floatValue())
                 .put(DOUBLE, obj -> ((Number) obj).doubleValue())

src/test/java/com/gotocompany/depot/maxcompute/util/MetadataUtilTest.java

@@ -26,6 +26,7 @@ public void setup() {
         MaxComputeSinkConfig maxComputeSinkConfig = Mockito.mock(MaxComputeSinkConfig.class);
         Mockito.when(maxComputeSinkConfig.getMaxComputeProtoTimestampToMaxcomputeType()).thenReturn(MaxComputeTimestampDataType.TIMESTAMP_NTZ);
         Mockito.when(maxComputeSinkConfig.getZoneId()).thenReturn(ZoneId.of("UTC"));
+        Mockito.when(maxComputeSinkConfig.isProtoIntegerTypesToBigintEnabled()).thenReturn(false);
         this.metadataUtil = new MetadataUtil(maxComputeSinkConfig);
     }