org.apache.inlong.manager.web.controller.DataNodeController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of manager-web
The newest version!
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.inlong.manager.web.controller;

import org.apache.inlong.manager.common.enums.OperationTarget;
import org.apache.inlong.manager.common.enums.OperationType;
import org.apache.inlong.manager.common.enums.TenantUserTypeEnum;
import org.apache.inlong.manager.common.validation.SaveValidation;
import org.apache.inlong.manager.common.validation.UpdateByIdValidation;
import org.apache.inlong.manager.common.validation.UpdateByKeyValidation;
import org.apache.inlong.manager.pojo.common.PageResult;
import org.apache.inlong.manager.pojo.common.Response;
import org.apache.inlong.manager.pojo.common.UpdateResult;
import org.apache.inlong.manager.pojo.node.DataNodeInfo;
import org.apache.inlong.manager.pojo.node.DataNodePageRequest;
import org.apache.inlong.manager.pojo.node.DataNodeRequest;
import org.apache.inlong.manager.pojo.user.LoginUserUtils;
import org.apache.inlong.manager.service.node.DataNodeService;
import org.apache.inlong.manager.service.operationlog.OperationLog;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

/**
 * Data node controller
 */
@RestController
@RequestMapping("/api")
@Api(tags = "Data-Node-API")
public class DataNodeController {

    @Autowired
    private DataNodeService dataNodeService;

    @PostMapping(value = "/node/save")
    @ApiOperation(value = "Save node")
    @OperationLog(operation = OperationType.CREATE, operationTarget = OperationTarget.NODE)
    public Response save(@Validated(SaveValidation.class) @RequestBody DataNodeRequest request) {
        String currentUser = LoginUserUtils.getLoginUser().getName();
        return Response.success(dataNodeService.save(request, currentUser));
    }

    @GetMapping(value = "/node/get/{id}")
    @ApiOperation(value = "Get node by id")
    @ApiImplicitParam(name = "id", value = "Data node ID", dataTypeClass = Integer.class, required = true)
    public Response get(@PathVariable Integer id) {
        String currentUser = LoginUserUtils.getLoginUser().getName();
        return Response.success(dataNodeService.get(id, currentUser));
    }

    @PostMapping(value = "/node/getByKey")
    @ApiOperation(value = "Get data node by key")
    @OperationLog(operation = OperationType.GET, operationTarget = OperationTarget.NODE)
    @ApiImplicitParams({
            @ApiImplicitParam(name = "name", value = "Data node name", dataTypeClass = String.class, required = true),
            @ApiImplicitParam(name = "type", value = "Data node type", dataTypeClass = String.class, required = true)
    })
    public Response getByKey(@RequestParam String name, @RequestParam String type) {
        return Response.success(dataNodeService.get(name, type));
    }

    @PostMapping(value = "/node/list")
    @ApiOperation(value = "List data node")
    public Response> list(@RequestBody DataNodePageRequest request) {
        request.setCurrentUser(LoginUserUtils.getLoginUser().getName());
        request.setIsAdminRole(
                LoginUserUtils.getLoginUser().getRoles().contains(TenantUserTypeEnum.TENANT_ADMIN.name()));
        return Response.success(dataNodeService.list(request));
    }

    @PostMapping(value = "/node/update")
    @OperationLog(operation = OperationType.UPDATE, operationTarget = OperationTarget.NODE)
    @ApiOperation(value = "Update data node")
    public Response update(@Validated(UpdateByIdValidation.class) @RequestBody DataNodeRequest request) {
        String username = LoginUserUtils.getLoginUser().getName();
        return Response.success(dataNodeService.update(request, username));
    }

    @PostMapping(value = "/node/updateByKey")
    @OperationLog(operation = OperationType.UPDATE, operationTarget = OperationTarget.NODE)
    @ApiOperation(value = "Update data node by key")
    public Response updateByKey(
            @Validated(UpdateByKeyValidation.class) @RequestBody DataNodeRequest request) {
        String username = LoginUserUtils.getLoginUser().getName();
        return Response.success(dataNodeService.updateByKey(request, username));
    }

    @DeleteMapping(value = "/node/delete/{id}")
    @ApiOperation(value = "Delete data node by id")
    @OperationLog(operation = OperationType.DELETE, operationTarget = OperationTarget.NODE)
    @ApiImplicitParam(name = "id", value = "Data node ID", dataTypeClass = Integer.class, required = true)
    public Response delete(@PathVariable Integer id) {
        return Response.success(dataNodeService.delete(id, LoginUserUtils.getLoginUser().getName()));
    }

    @DeleteMapping(value = "/node/deleteByKey")
    @ApiOperation(value = "Delete data node by key")
    @OperationLog(operation = OperationType.DELETE, operationTarget = OperationTarget.NODE)
    @ApiImplicitParams({
            @ApiImplicitParam(name = "name", value = "Data node name", dataTypeClass = String.class, required = true),
            @ApiImplicitParam(name = "type", value = "Data node type", dataTypeClass = String.class, required = true)
    })
    public Response deleteByKey(@RequestParam String name, @RequestParam String type) {
        return Response.success(dataNodeService.deleteByKey(name, type,
                LoginUserUtils.getLoginUser().getName()));
    }

    @PostMapping("/node/testConnection")
    @ApiOperation(value = "Test connection for data node")
    public Response testConnection(@Validated @RequestBody DataNodeRequest request) {
        return Response.success(dataNodeService.testConnection(request));
    }

}