com.spredfast.kafka.connect.s3.source.S3Partition Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kafka-connect-s3 Show documentation

S3 Source and Sink Connectors for Kafka Connect

There is a newer version: 0.5.0

package com.spredfast.kafka.connect.s3.source;

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

public class S3Partition {
	private final String bucket;
	private final String keyPrefix;
	private final String topic;
	private final int partition;

	public S3Partition(String bucket, String keyPrefix, String topic, int partition) {
		this.bucket = bucket;
		this.keyPrefix = normalizePrefix(keyPrefix);
		this.topic = topic;
		this.partition = partition;
	}

	public static S3Partition from(String bucket, String keyPrefix, String topic, int partition) {
		return new S3Partition(bucket, keyPrefix, topic, partition);
	}

	public static S3Partition from(Map map) {
		String bucket = (String) map.get("bucket");
		String keyPrefix = (String) map.get("keyPrefix");
		String topic = (String) map.get("topic");
		int partition = ((Number) map.get("kafkaPartition")).intValue();
		return from(bucket, keyPrefix, topic, partition);
	}

	public static String normalizePrefix(String keyPrefix) {
		return keyPrefix == null ? ""
			: keyPrefix.endsWith("/") ? keyPrefix : keyPrefix + "/";
	}

	public Map asMap() {
		Map map = new HashMap<>();
		map.put("bucket", bucket);
		map.put("keyPrefix", keyPrefix);
		map.put("topic", topic);
		map.put("kafkaPartition", partition);
		return map;
	}

	@Override
	public boolean equals(Object o) {
		if (this == o) return true;
		if (o == null || getClass() != o.getClass()) return false;
		S3Partition that = (S3Partition) o;
		return partition == that.partition &&
			Objects.equals(bucket, that.bucket) &&
			Objects.equals(keyPrefix, that.keyPrefix) &&
			Objects.equals(topic, that.topic);
	}

	@Override
	public int hashCode() {
		return Objects.hash(bucket, keyPrefix, topic, partition);
	}

	@Override
	public String toString() {
		return bucket + "/" + keyPrefix + "/" + topic  + "-" + partition;
	}
}