io.streamnative.pulsar.handlers.kop.schemaregistry.model.impl.MD5 Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pulsar-kafka-schema-registry Show documentation

Kafka Compatible Schema Registry

There is a newer version: 4.0.0.4

Show newest version

/**
 * Copyright (c) 2019 - 2024 StreamNative, Inc.. All Rights Reserved.
 */
/**
 * 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.streamnative.pulsar.handlers.kop.schemaregistry.model.impl;

import io.streamnative.pulsar.handlers.kop.schemaregistry.model.Schema;
import io.streamnative.pulsar.handlers.kop.schemaregistry.model.SchemaValue;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;

/**
 * Simple wrapper for 16 byte MD5 hash.
 */
public class MD5 {

    private final byte[] md5;

    public MD5(byte[] md5) {
        if (md5 == null) {
            throw new IllegalArgumentException("Tried to instantiate MD5 object with null byte array.");
        }
        if (md5.length != 16) {
            throw new IllegalArgumentException("Tried to instantiate MD5 object with invalid byte array.");
        }
        this.md5 = md5.clone();
    }

    public byte[] bytes() {
        return md5.clone();
    }

    public static MD5 ofSchema(Schema schema) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            schema.updateHash(md);
            return new MD5(md.digest());
        } catch (NoSuchAlgorithmException e) {
            throw new RuntimeException(e);
        }
    }

    public static MD5 ofSchema(SchemaValue schema) {
        byte[] bytes = schema.getMd5Bytes();
        return bytes != null ? new MD5(bytes) : ofSchema(schema.toSchema());
    }

    @Override
    public int hashCode() {
        return Arrays.hashCode(this.md5);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }

        if (o == null) {
            return false;
        }

        if (!(o instanceof MD5)) {
            return false;
        }

        MD5 otherMd5 = (MD5) o;
        return Arrays.equals(this.md5, otherMd5.md5);
    }
}