io.confluent.connect.cdc.JsonConnectSchema Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kafka-connect-cdc-test Show documentation
The newest version!
/**
 * Copyright © 2017 Jeremy Custenborder ([email protected])
 *
 * Licensed 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 io.confluent.connect.cdc;

import com.fasterxml.jackson.annotation.JsonAutoDetect;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.JsonDeserializer;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.google.common.base.MoreObjects;
import com.google.common.base.Preconditions;
import com.google.common.base.Strings;
import org.apache.kafka.connect.data.Field;
import org.apache.kafka.connect.data.Schema;
import org.apache.kafka.connect.data.SchemaBuilder;

import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.Map;

@JsonAutoDetect(fieldVisibility = JsonAutoDetect.Visibility.ANY, getterVisibility = JsonAutoDetect.Visibility.NONE, setterVisibility = JsonAutoDetect.Visibility.NONE)
@JsonInclude(JsonInclude.Include.NON_NULL)
public class JsonConnectSchema {
  String name;
  String doc;
  Schema.Type type;
  Object defaultValue;
  Integer version;
  Map parameters;
  boolean isOptional;
  Schema keySchema;
  Schema valueSchema;
  Map fieldSchemas;

  public JsonConnectSchema() {

  }

  public JsonConnectSchema(Schema schema) {
    this.name = schema.name();
    this.doc = schema.doc();
    this.type = schema.type();
    this.defaultValue = schema.defaultValue();
    this.version = schema.version();
    this.parameters = schema.parameters();
    this.isOptional = schema.isOptional();

    if (Schema.Type.MAP == this.type) {
      this.keySchema = schema.keySchema();
      this.valueSchema = schema.valueSchema();
    } else if (Schema.Type.ARRAY == this.type) {
      this.keySchema = null;
      this.valueSchema = schema.valueSchema();
    } else if (Schema.Type.STRUCT == this.type) {
      this.fieldSchemas = new LinkedHashMap<>();
      for (Field field : schema.fields()) {
        this.fieldSchemas.put(field.name(), field.schema());
      }
    }
  }

  public Schema build() {
    SchemaBuilder builder;

    switch (this.type) {
      case MAP:
        Preconditions.checkNotNull(this.keySchema, "keySchema cannot be null.");
        Preconditions.checkNotNull(this.valueSchema, "valueSchema cannot be null.");
        builder = SchemaBuilder.map(this.keySchema, this.valueSchema);
        break;
      case ARRAY:
        Preconditions.checkNotNull(this.valueSchema, "valueSchema cannot be null.");
        builder = SchemaBuilder.array(this.valueSchema);
        break;
      default:
        builder = SchemaBuilder.type(this.type);
        break;
    }

    if (Schema.Type.STRUCT == this.type) {
      for (Map.Entry kvp : this.fieldSchemas.entrySet()) {
        builder.field(kvp.getKey(), kvp.getValue());
      }
    }

    if (!Strings.isNullOrEmpty(this.name)) {
      builder.name(this.name);
    }

    if (!Strings.isNullOrEmpty(this.doc)) {
      builder.doc(this.doc);
    }

    if (null != this.defaultValue) {
      builder.defaultValue(this.defaultValue);
    }

    if (null != this.parameters) {
      builder.parameters(this.parameters);
    }

    if (this.isOptional) {
      builder.optional();
    }

    if (null != this.version) {
      builder.version(this.version);
    }

    return builder.build();
  }

  @Override
  public String toString() {
    return MoreObjects.toStringHelper(this.getClass())
        .add("type", this.type)
        .add("name", this.name)
        .add("isOptional", this.isOptional)
        .add("version", this.version)
        .add("keySchema", this.keySchema)
        .add("valueSchema", this.valueSchema)
        .add("parameters", this.parameters)
        .add("doc", this.doc)
        .add("defaultValue", this.defaultValue)
        .omitNullValues()
        .toString();
  }

  @Override
  public int hashCode() {
    //TODO: FIx this
    return super.hashCode();
  }

  @Override
  public boolean equals(Object obj) {
    if (!(obj instanceof JsonConnectSchema)) {
      return false;
    }

    JsonConnectSchema that = (JsonConnectSchema) obj;

    if (this.name != that.name) {
      return false;
    }
    if (this.type != that.type) {
      return false;
    }
    if (this.doc != that.doc) {
      return false;
    }
    if (this.isOptional != that.isOptional) {
      return false;
    }
    if (this.version != that.version) {
      return false;
    }
    if (this.defaultValue != that.defaultValue) {
      return false;
    }

    if (Schema.Type.MAP == this.type) {
      if (this.keySchema.equals(that.keySchema)) {
        return false;
      }
      if (this.valueSchema.equals(that.keySchema)) {
        return false;
      }
    } else if (Schema.Type.ARRAY == this.type) {
      if (this.valueSchema.equals(that.keySchema)) {
        return false;
      }
    }

    return true;
  }

  static class Serializer extends JsonSerializer {
    @Override
    public void serialize(Schema schema, JsonGenerator jsonGenerator, SerializerProvider serializerProvider) throws IOException, JsonProcessingException {
      JsonConnectSchema jsonConnectSchema = new JsonConnectSchema(schema);
      jsonGenerator.writeObject(jsonConnectSchema);
    }
  }

  static class Deserializer extends JsonDeserializer {

    @Override
    public Schema deserialize(JsonParser jsonParser, DeserializationContext deserializationContext) throws IOException {
      JsonConnectSchema jsonConnectSchema = jsonParser.readValueAs(JsonConnectSchema.class);
      return jsonConnectSchema.build();
    }
  }
}