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

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.model.CompatibilityChecker;
import io.streamnative.pulsar.handlers.kop.schemaregistry.model.SchemaStorage;
import io.streamnative.pulsar.handlers.kop.schemaregistry.model.SchemaStorageAccessor;
import io.streamnative.pulsar.handlers.kop.schemaregistry.utils.SubjectUtils;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import lombok.AllArgsConstructor;
import lombok.Data;

public class ConfigResource extends AbstractResource {

    public ConfigResource(SchemaStorageAccessor schemaStorageAccessor,
                          String defaultNamespace) {
        super(schemaStorageAccessor, defaultNamespace);
    }

    @Override
    public void register(SchemaRegistryHandler schemaRegistryHandler) {
        schemaRegistryHandler.addProcessor(new PutConfig());
        schemaRegistryHandler.addProcessor(new GetSubjectConfig());
        schemaRegistryHandler.addProcessor(new GetConfig());
        schemaRegistryHandler.addProcessor(new GetMode());
    }

    @Data
    @AllArgsConstructor
    public static final class GetConfigResponse {
        private String compatibilityLevel;
    }

    // GET /config
    public static class GetConfig extends HttpJsonRequestProcessor {

        public GetConfig() {
            super(Void.class, "/config", GET);
        }

        @Override
        protected CompletableFuture processRequest(Void payload, List patternGroups,
                                                                      FullHttpRequest request,
                                                                      Map> queryParams,
                                                                      String currentTenant)
                throws Exception {
            return CompletableFuture.completedFuture(new GetConfigResponse(CompatibilityChecker.Mode.NONE.name()));
        }

    }

    @Data
    @AllArgsConstructor
    public static final class GetModeResponse {
        private String mode;
    }

    // GET /mode
    public static class GetMode extends HttpJsonRequestProcessor {

        public GetMode() {
            super(Void.class, "/mode", GET);
        }

        @Override
        protected CompletableFuture processRequest(Void payload, List patternGroups,
                                                                    FullHttpRequest request,
                                                                    Map> queryParams,
                                                                    String currentTenant)
                throws Exception {
            return CompletableFuture.completedFuture(new GetModeResponse("READWRITE"));
        }

    }

    @Data
    public static final class PutConfigRequest {
        private String compatibility;
    }

    // GET /config/${subject}
    public class GetSubjectConfig extends HttpJsonRequestProcessor {

        public GetSubjectConfig() {
            super(Void.class, "/config/" + STRING_PATTERN, GET);
        }

        @Override
        protected CompletableFuture processRequest(Void payload, List patternGroups,
                                                                      FullHttpRequest request,
                                                                      Map> queryParams,
                                                                      String currentTenant)
                throws Exception {
            SchemaStorage schemaStorage = getSchemaStorage(currentTenant);
            final String subject = SubjectUtils.normalize(getString(0, patternGroups), defaultNamespace);
            return schemaStorage.getCompatibilityMode(subject).thenApply(r -> new GetConfigResponse(r.name()));
        }

    }

    // PUT /config/${subject}
    public class PutConfig extends HttpJsonRequestProcessor {

        public PutConfig() {
            super(PutConfigRequest.class, "/config/" + STRING_PATTERN, PUT);
        }

        @Override
        protected CompletableFuture processRequest(PutConfigRequest payload,
                                                                      List patternGroups,
                                                                      FullHttpRequest request,
                                                                      Map> queryParams,
                                                                      String currentTenant)
                throws Exception {
            SchemaStorage schemaStorage = getSchemaStorage(currentTenant);
            final String subject = SubjectUtils.normalize(getString(0, patternGroups), defaultNamespace);
            CompatibilityChecker.Mode mode = CompatibilityChecker.Mode.valueOf(payload.compatibility);
            return schemaStorage.setCompatibilityMode(subject, mode)
                    .thenApply(r -> new GetConfigResponse(payload.compatibility));
        }

    }

}