All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.netflix.eureka2.server.transport.http.registration.RegistrationHttpRequestRouter Maven / Gradle / Ivy

    /*
 * Copyright 2014 Netflix, Inc.
 *
 * 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 com.netflix.eureka2.server.transport.http.registration;

import java.util.concurrent.ConcurrentHashMap;

import com.google.inject.Inject;
import com.netflix.eureka2.protocol.registration.Register;
import com.netflix.eureka2.protocol.registration.Update;
import com.netflix.eureka2.registry.InstanceInfo;
import com.netflix.eureka2.server.service.ServerChannelFactory;
import com.netflix.eureka2.service.RegistrationChannel;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.HttpResponseStatus;
import io.reactivex.netty.protocol.http.server.HttpServerRequest;
import io.reactivex.netty.protocol.http.server.HttpServerResponse;
import netflix.karyon.transport.RequestRouter;
import rx.Observable;
import rx.functions.Func1;

/**
 * FIXME Implement heartbeat mechanism.
 *
 * @author Tomasz Bak
 */
public class RegistrationHttpRequestRouter implements RequestRouter, HttpServerResponse> {

    private final String appsBaseURI;
    private final ServerChannelFactory eurekaServerService;

    private final ConcurrentHashMap activeRegistrationChannels = new ConcurrentHashMap();

    @Inject
    public RegistrationHttpRequestRouter(ServerChannelFactory eurekaServerService, String baseURI) {
        this.eurekaServerService = eurekaServerService;
        this.appsBaseURI = baseURI + "/apps";
    }

    @Override
    public Observable route(HttpServerRequest request, HttpServerResponse response) {
        String path = request.getPath();
        if (!path.startsWith(appsBaseURI)) {
            return notFound(response);
        }

        // POST /apps
        HttpMethod httpMethod = request.getHttpMethod();
        if (path.length() == appsBaseURI.length()) {
            if (httpMethod.equals(HttpMethod.POST)) {
                return handleRegistrationPost(request, response);
            }
            return notFound(response);
        }

        // Must be /apps/...
        if (path.charAt(appsBaseURI.length()) != '/' || path.length() < appsBaseURI.length() + 2) {
            return notFound(response);
        }

        String instanceId = path.substring(appsBaseURI.length() + 1);

        if (httpMethod.equals(HttpMethod.GET)) {
            return handleInstanceGet(instanceId, response);
        }
        if (httpMethod.equals(HttpMethod.PUT)) {
            return handleInstanceUpdate(instanceId, request, response);
        }
        if (httpMethod.equals(HttpMethod.DELETE)) {
            return handleInstanceDelete(instanceId);
        }

        return notFound(response);
    }

    private Observable handleRegistrationPost(HttpServerRequest request, final HttpServerResponse response) {
        return request.getContent().flatMap(new Func1>() {
            @Override
            public Observable call(Object o) {
                if (!(o instanceof Register)) {
                    response.setStatus(HttpResponseStatus.BAD_REQUEST);
                    return Observable.empty();
                }
                InstanceInfo instanceInfo = ((Register) o).getInstanceInfo();
                if (activeRegistrationChannels.contains(instanceInfo.getId())) {
                    activeRegistrationChannels.get(instanceInfo.getId()).close();
                }
                RegistrationChannel registrationChannel = eurekaServerService.newRegistrationChannel();
                activeRegistrationChannels.put(instanceInfo.getId(), registrationChannel);
                return registrationChannel.register(instanceInfo);
            }
        });
    }

    private Observable handleInstanceGet(String instanceId, HttpServerResponse response) {
        return Observable.error(new RuntimeException("not implemented yet"));
    }

    private Observable handleInstanceUpdate(final String instanceId, HttpServerRequest request, final HttpServerResponse response) {
        return request.getContent().flatMap(new Func1>() {
            @Override
            public Observable call(Object o) {
                if (!(o instanceof Update)) {
                    response.setStatus(HttpResponseStatus.BAD_REQUEST);
                    return Observable.empty();
                }
                InstanceInfo instanceInfo = ((Update) o).getInstanceInfo();
                if (activeRegistrationChannels.containsKey(instanceId)) {
                    return activeRegistrationChannels.get(instanceId).update(instanceInfo);
                }
                // Instead of returning error, we silently do regular registration.
                RegistrationChannel registrationChannel = eurekaServerService.newRegistrationChannel();
                activeRegistrationChannels.put(instanceInfo.getId(), registrationChannel);
                return registrationChannel.register(instanceInfo);
            }
        });
    }

    private Observable handleInstanceDelete(String instanceId) {
        RegistrationChannel toRemove = activeRegistrationChannels.remove(instanceId);
        if (toRemove != null) {
            toRemove.close();
        }
        return Observable.empty();
    }

    private static Observable notFound(HttpServerResponse response) {
        response.setStatus(HttpResponseStatus.NOT_FOUND);
        return Observable.empty();
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy