io.streamnative.pulsar.handlers.kop.schemaregistry.model.Schema 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
The 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;

import com.fasterxml.jackson.annotation.JsonProperty;
import io.streamnative.pulsar.handlers.kop.schemaregistry.model.rest.SchemaReference;
import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NoArgsConstructor;


// This class is migrated from io.confluent.kafka.schemaregistry.client.rest.entities.Schema
@Data
@AllArgsConstructor
@Builder()
@EqualsAndHashCode
@NoArgsConstructor
public final class Schema {

    public static final String TYPE_AVRO = "AVRO";
    public static final String TYPE_JSON = "JSON";
    public static final String TYPE_PROTOBUF = "PROTOBUF";
    private static final List ALL_TYPES =
        Collections.unmodifiableList(Arrays.asList(TYPE_AVRO, TYPE_JSON, TYPE_PROTOBUF));

    private String tenant;
    @JsonProperty("id")
    private Integer id;
    @JsonProperty("version")
    private Integer version;
    @JsonProperty("schema")
    private String schema;
    @JsonProperty("subject")
    private String subject;
    @JsonProperty("type")
    @Builder.Default
    private String type = TYPE_AVRO;
    @JsonProperty("references")
    @Builder.Default
    private List references = Collections.emptyList();

    public static List getAllTypes() {
        return ALL_TYPES;
    }

    public Schema copy() {
        return new Schema(tenant, id, version, schema, subject, type, references);
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append("{subject=" + this.subject + ",");
        sb.append("version=" + this.version + ",");
        sb.append("id=" + this.id + ",");
        sb.append("schemaType=" + this.type + ",");
        sb.append("references=" + this.references + ",");
        return sb.toString();
    }

    public void updateHash(MessageDigest md) {
        if (schema != null) {
            md.update(schema.getBytes(StandardCharsets.UTF_8));
        }
        if (references != null) {
            references.forEach(r -> r.updateHash(md));
        }
    }

}