io.confluent.connect.cdc.JsonChange 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.annotation.JsonProperty;
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 java.io.IOException;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

@JsonAutoDetect(fieldVisibility = JsonAutoDetect.Visibility.ANY, getterVisibility = JsonAutoDetect.Visibility.NONE, setterVisibility = JsonAutoDetect.Visibility.NONE)
@JsonInclude(JsonInclude.Include.NON_NULL)
public class JsonChange {
  Map metadata = new LinkedHashMap<>();
  Map sourcePartition = new LinkedHashMap<>();
  Map sourceOffset = new LinkedHashMap<>();
  String databaseName;
  String schemaName;
  String tableName;
  Change.ChangeType changeType;
  long timestamp;
  @JsonProperty
  List keyColumns = new ArrayList<>();
  @JsonProperty
  List valueColumns = new ArrayList<>();

  @Override
  public String toString() {
    return MoreObjects.toStringHelper(this.getClass())
        .add("schemaName", this.schemaName)
        .add("tableName", this.tableName)
        .add("changeType", this.changeType)
        .add("metadata", this.metadata)
        .add("sourcePartition", this.sourcePartition)
        .add("sourceOffset", this.sourceOffset)
        .add("timestamp", this.timestamp)
        .add("keyColumns", this.keyColumns)
        .add("valueColumns", this.valueColumns)
        .omitNullValues()
        .toString();
  }

  static class Serializer extends JsonSerializer {
    @Override
    public void serialize(Change change, JsonGenerator jsonGenerator, SerializerProvider serializerProvider) throws IOException, JsonProcessingException {
      JsonChange result = new JsonChange();
      result.databaseName = change.databaseName();
      result.schemaName = change.schemaName();
      result.tableName = change.tableName();
      result.changeType = change.changeType();
      result.timestamp = change.timestamp();
      result.metadata = change.metadata();
      result.sourceOffset = change.sourceOffset();
      result.sourcePartition = change.sourcePartition();
      result.keyColumns = change.keyColumns();
      result.valueColumns = change.valueColumns();
      jsonGenerator.writeObject(result);
    }
  }

  static class Deserializer extends JsonDeserializer {
    @Override
    public Change deserialize(JsonParser jsonParser, DeserializationContext deserializationContext) throws IOException, JsonProcessingException {
      JsonChange storage = jsonParser.readValueAs(JsonChange.class);
      Change result = mock(Change.class);
      when(result.databaseName()).thenReturn(storage.databaseName);
      when(result.schemaName()).thenReturn(storage.schemaName);
      when(result.tableName()).thenReturn(storage.tableName);
      when(result.changeType()).thenReturn(storage.changeType);
      when(result.timestamp()).thenReturn(storage.timestamp);
      when(result.metadata()).thenReturn(storage.metadata);
      when(result.sourceOffset()).thenReturn(storage.sourceOffset);
      when(result.sourcePartition()).thenReturn(storage.sourcePartition);
      when(result.keyColumns()).thenReturn(storage.keyColumns);
      when(result.valueColumns()).thenReturn(storage.valueColumns);
      return result;
    }
  }
}