com.netflix.spinnaker.halyard.controllers.v1.PubsubController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of halyard-web Show documentation
Spinnaker Halyard
The newest version!
/*
 * Copyright 2017 Google, 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.spinnaker.halyard.controllers.v1;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.netflix.spinnaker.halyard.config.config.v1.HalconfigDirectoryStructure;
import com.netflix.spinnaker.halyard.config.config.v1.HalconfigParser;
import com.netflix.spinnaker.halyard.config.model.v1.node.Halconfig;
import com.netflix.spinnaker.halyard.config.model.v1.node.Pubsub;
import com.netflix.spinnaker.halyard.config.model.v1.node.Pubsubs;
import com.netflix.spinnaker.halyard.config.services.v1.PubsubService;
import com.netflix.spinnaker.halyard.core.tasks.v1.DaemonTask;
import com.netflix.spinnaker.halyard.models.v1.ValidationSettings;
import com.netflix.spinnaker.halyard.util.v1.GenericEnableDisableRequest;
import com.netflix.spinnaker.halyard.util.v1.GenericGetRequest;
import com.netflix.spinnaker.halyard.util.v1.GenericUpdateRequest;
import java.util.List;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;

@RestController
@RequiredArgsConstructor
@RequestMapping("/v1/config/deployments/{deploymentName:.+}/pubsubs")
public class PubsubController {
  private final HalconfigParser halconfigParser;
  private final PubsubService pubsubService;
  private final HalconfigDirectoryStructure halconfigDirectoryStructure;
  private final ObjectMapper objectMapper;

  @RequestMapping(value = "/{pubsubName:.+}", method = RequestMethod.GET)
  DaemonTask get(
      @PathVariable String deploymentName,
      @PathVariable String pubsubName,
      @ModelAttribute ValidationSettings validationSettings) {
    return GenericGetRequest.builder()
        .getter(() -> pubsubService.getPubsub(deploymentName, pubsubName))
        .validator(() -> pubsubService.validatePubsub(deploymentName, pubsubName))
        .description("Get the " + pubsubName + " pubsub")
        .build()
        .execute(validationSettings);
  }

  @RequestMapping(value = "/{pubsubName:.+}", method = RequestMethod.PUT)
  DaemonTask setPubsub(
      @PathVariable String deploymentName,
      @PathVariable String pubsubName,
      @ModelAttribute ValidationSettings validationSettings,
      @RequestBody Object rawPubsub) {
    Pubsub pubsub = objectMapper.convertValue(rawPubsub, Pubsubs.translatePubsubType(pubsubName));
    return GenericUpdateRequest.builder(halconfigParser)
        .stagePath(halconfigDirectoryStructure.getStagingPath(deploymentName))
        .updater(p -> pubsubService.setPubsub(deploymentName, p))
        .validator(() -> pubsubService.validatePubsub(deploymentName, pubsubName))
        .description("Edit the " + pubsubName + " pubsub")
        .build()
        .execute(validationSettings, pubsub);
  }

  @RequestMapping(value = "/{pubsubName:.+}/enabled", method = RequestMethod.PUT)
  DaemonTask setEnabled(
      @PathVariable String deploymentName,
      @PathVariable String pubsubName,
      @ModelAttribute ValidationSettings validationSettings,
      @RequestBody boolean enabled) {
    return GenericEnableDisableRequest.builder(halconfigParser)
        .updater(e -> pubsubService.setEnabled(deploymentName, pubsubName, e))
        .validator(() -> pubsubService.validatePubsub(deploymentName, pubsubName))
        .description("Edit the " + pubsubName + " pubsub")
        .build()
        .execute(validationSettings, enabled);
  }

  @RequestMapping(value = "/", method = RequestMethod.GET)
  DaemonTask> pubsubs(
      @PathVariable String deploymentName, @ModelAttribute ValidationSettings validationSettings) {
    return GenericGetRequest.>builder()
        .getter(() -> pubsubService.getAllPubsubs(deploymentName))
        .validator(() -> pubsubService.validateAllPubsubs(deploymentName))
        .description("Get all pubsubs")
        .build()
        .execute(validationSettings);
  }
}