io.streamnative.pulsar.handlers.kop.schemaregistry.resources.SchemaResource 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
/**
 * 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.resources;

import com.fasterxml.jackson.annotation.JsonInclude;
import io.netty.handler.codec.http.FullHttpRequest;
import io.streamnative.pulsar.handlers.kop.schemaregistry.HttpJsonRequestProcessor;
import io.streamnative.pulsar.handlers.kop.schemaregistry.SchemaRegistryHandler;
import io.streamnative.pulsar.handlers.kop.schemaregistry.SchemaRegistryRequestAuthenticator;
import io.streamnative.pulsar.handlers.kop.schemaregistry.model.Schema;
import io.streamnative.pulsar.handlers.kop.schemaregistry.model.SchemaStorage;
import io.streamnative.pulsar.handlers.kop.schemaregistry.model.SchemaStorageAccessor;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Collectors;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.apache.commons.collections4.CollectionUtils;

public class SchemaResource extends AbstractResource {

    public SchemaResource(SchemaStorageAccessor schemaStorageAccessor,
                          SchemaRegistryRequestAuthenticator schemaRegistryRequestAuthenticator) {
        super(schemaStorageAccessor, schemaRegistryRequestAuthenticator);
    }

    /**
     * Register all processors.
     * @param schemaRegistryHandler
     */
    public void register(SchemaRegistryHandler schemaRegistryHandler) {
        schemaRegistryHandler.addProcessor(new GetSchemaById());
        schemaRegistryHandler.addProcessor(new GetSchemaStringById());
        schemaRegistryHandler.addProcessor(new GetSchemaTypes());
        schemaRegistryHandler.addProcessor(new GetSchemaAliases());
    }

    @Data
    @NoArgsConstructor
    @AllArgsConstructor
    @JsonInclude(JsonInclude.Include.NON_EMPTY)
    public static final class GetSchemaResponse {
        private String schema;
        private String schemaType;
        //todo: references, metadata, ruleSet, maxId

        public String getSchemaType() {
            return "AVRO".equals(schemaType) ? null : schemaType;
        }
    }

    // /schemas/types
    public static class GetSchemaTypes extends HttpJsonRequestProcessor> {

        public GetSchemaTypes() {
            super(Void.class, "/schemas/types", GET);
        }

        @Override
        protected CompletableFuture> processRequest(Void payload, List patternGroups,
                                                                 FullHttpRequest request) {
            return CompletableFuture.completedFuture(Schema.getAllTypes());
        }

    }

    @Data
    @NoArgsConstructor
    @AllArgsConstructor
    public static final class SubjectVersionPair {
        private String subject;
        private int version;
    }

    // GET /schemas/ids/{int: id}
    public class GetSchemaById extends HttpJsonRequestProcessor {

        public GetSchemaById() {
            super(Void.class, "/schemas/ids/" + INT_PATTERN, GET);
        }

        @Override
        protected CompletableFuture processRequest(Void payload,
                                                                      List patternGroups,
                                                                      FullHttpRequest request)
                throws Exception {
            int id = getInt(0, patternGroups);
            SchemaStorage schemaStorage = getSchemaStorage(request);
            CompletableFuture> schemasFuture = schemaStorage.findSchemaById(id);
            return schemasFuture.thenApply(list -> {
                if (CollectionUtils.isEmpty(list)) {
                    return null;
                }
                return new GetSchemaResponse(list.get(0).getSchemaDefinition(), list.get(0).getType());
            });
        }
    }

    // GET /schemas/ids/{int: id}/schema     Get one schema string
    public class GetSchemaStringById extends HttpJsonRequestProcessor {

        public GetSchemaStringById() {
            super(Void.class, "/schemas/ids/" + INT_PATTERN + "/schema", GET);
        }

        @Override
        protected CompletableFuture processRequest(Void payload, List patternGroups,
                                                                      FullHttpRequest request)
                throws Exception {
            int id = getInt(0, patternGroups);
            SchemaStorage schemaStorage = getSchemaStorage(request);
            CompletableFuture> schemasFuture = schemaStorage.findSchemaById(id);
            return schemasFuture.thenApply(list -> {
                if (CollectionUtils.isEmpty(list)) {
                    return null;
                }
                return list.get(0).getSchemaDefinition();
            });
        }

    }

    // /schemas/ids/{int: id}/versions
    public class GetSchemaAliases extends HttpJsonRequestProcessor> {

        public GetSchemaAliases() {
            super(Void.class, "/schemas/ids/" + INT_PATTERN + "/versions", GET);
        }

        @Override
        protected CompletableFuture> processRequest(Void payload,
                                                                             List patternGroups,
                                                                             FullHttpRequest request)
                throws Exception {
            SchemaStorage schemaStorage = getSchemaStorage(request);
            int id = getInt(0, patternGroups);
            CompletableFuture> schemasFuture = schemaStorage.findSchemaById(id);
            return schemasFuture.thenApply(list -> {
                if (CollectionUtils.isEmpty(list)) {
                    return null;
                }
                return list.stream()
                        .map(s -> new SubjectVersionPair(s.getSubject(), s.getVersion()))
                        .collect(Collectors.toList());
            });
        }

    }


}