com.proofpoint.discovery.DynamicAnnouncementResource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of discovery-server Show documentation
Discovery server
The newest version!
/*
 * Copyright 2010 Proofpoint, 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.proofpoint.discovery;

import com.google.common.base.Joiner;
import com.proofpoint.node.NodeInfo;

import javax.inject.Inject;
import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.Consumes;
import javax.ws.rs.DELETE;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.util.HashSet;
import java.util.Set;

import static java.lang.String.format;
import static java.util.Objects.requireNonNullElse;
import static javax.ws.rs.core.Response.Status.ACCEPTED;
import static javax.ws.rs.core.Response.Status.BAD_REQUEST;
import static javax.ws.rs.core.Response.Status.FORBIDDEN;

@Path("/v1/announcement/{node_id}")
public class DynamicAnnouncementResource
{
    private final NodeInfo nodeInfo;
    private final DynamicStore dynamicStore;
    private final Set proxyTypes;
    private final AuthManager authManager;

    @Inject
    public DynamicAnnouncementResource(DynamicStore dynamicStore, NodeInfo nodeInfo, DiscoveryConfig discoveryConfig, AuthManager authManager)
    {
        this.dynamicStore = dynamicStore;
        this.nodeInfo = nodeInfo;
        proxyTypes = discoveryConfig.getProxyProxiedTypes();
        this.authManager = authManager;
    }

    @PUT
    @Consumes(MediaType.APPLICATION_JSON)
    public Response put(@PathParam("node_id") Id nodeId, DynamicAnnouncement announcement, @Context HttpServletRequest request)
    {
        authManager.checkAuthAnnounce(nodeId, announcement, request);
        if (!nodeInfo.getEnvironment().equals(announcement.getEnvironment())) {
            return Response.status(BAD_REQUEST)
                    .entity(format("Environment mismatch. Expected: %s, Provided: %s", nodeInfo.getEnvironment(), announcement.getEnvironment()))
                    .build();
        }

        if (!proxyTypes.isEmpty()) {
            Set forbiddenTypes = new HashSet<>();
            for (DynamicServiceAnnouncement serviceAnnouncement : announcement.getServiceAnnouncements()) {
                String type = serviceAnnouncement.getType();
                if (proxyTypes.contains(type)) {
                    forbiddenTypes.add(type);
                }
            }
            if (!forbiddenTypes.isEmpty()) {
                return Response.status(FORBIDDEN)
                        .entity(format("Cannot announce proxied type%s %s", forbiddenTypes.size() == 1 ? "" : "s", Joiner.on(',').join(forbiddenTypes)))
                        .build();
            }
        }

        String location = requireNonNullElse(announcement.getLocation(), "/somewhere/" + nodeId.toString());

        DynamicAnnouncement announcementWithLocation = DynamicAnnouncement.copyOf(announcement)
                .setLocation(location)
                .setAnnouncer(request.getRemoteAddr())
                .build();

        dynamicStore.put(nodeId, announcementWithLocation);

        return Response.status(ACCEPTED).build();
    }

    @DELETE
    public void delete(@PathParam("node_id") Id nodeId, @Context HttpServletRequest request)
    {
        authManager.checkAuthDelete(nodeId, request);
        dynamicStore.delete(nodeId);
    }
}