vamossagar12 commented on code in PR #14093: URL: https://github.com/apache/kafka/pull/14093#discussion_r1311113838
########## connect/runtime/src/main/java/org/apache/kafka/connect/converters/BooleanConverter.java: ########## @@ -0,0 +1,107 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.kafka.connect.converters; + +import java.util.HashMap; +import java.util.Map; +import org.apache.kafka.common.config.ConfigDef; +import org.apache.kafka.common.errors.SerializationException; +import org.apache.kafka.common.serialization.BooleanDeserializer; +import org.apache.kafka.common.serialization.BooleanSerializer; +import org.apache.kafka.common.utils.Utils; +import org.apache.kafka.connect.data.Schema; +import org.apache.kafka.connect.data.Schema.Type; +import org.apache.kafka.connect.data.SchemaAndValue; +import org.apache.kafka.connect.errors.DataException; +import org.apache.kafka.connect.storage.Converter; +import org.apache.kafka.connect.storage.ConverterConfig; +import org.apache.kafka.connect.storage.ConverterType; +import org.apache.kafka.connect.storage.HeaderConverter; + +/** + * {@link Converter} and {@link HeaderConverter} implementation that supports serializing to and + * deserializing from Boolean values. + * <p> + * When converting from bytes to Kafka Connect format, the converter will always return an optional + * BOOLEAN schema. + */ +public class BooleanConverter implements Converter, HeaderConverter { + + private static final ConfigDef CONFIG_DEF = ConverterConfig.newConfigDef(); + + private final BooleanSerializer serializer = new BooleanSerializer(); + private final BooleanDeserializer deserializer = new BooleanDeserializer(); + + @Override + public ConfigDef config() { + return CONFIG_DEF; + } + + @Override + public void configure(Map<String, ?> configs) { + BooleanConverterConfig conf = new BooleanConverterConfig(configs); + boolean isKey = conf.type() == ConverterType.KEY; + serializer.configure(configs, isKey); + deserializer.configure(configs, isKey); + } + + @Override + public void configure(Map<String, ?> configs, boolean isKey) { + Map<String, Object> conf = new HashMap<>(configs); + conf.put(ConverterConfig.TYPE_CONFIG, + isKey ? ConverterType.KEY.getName() : ConverterType.VALUE.getName()); + configure(conf); + } + + @Override + public byte[] fromConnectData(String topic, Schema schema, Object value) { + if (schema != null && schema.type() != Type.BOOLEAN) + throw new DataException("Invalid schema type for BooleanConverter: " + schema.type().toString()); + + try { + return serializer.serialize(topic, (Boolean) value); + } catch (ClassCastException e) { Review Comment: Should we be catching `SerializationException` as well? I see `NumberConverter` doing the same which also uses `serialize` method here. Not sure how that can be thrown though. -- 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: jira-unsubscr...@kafka.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
