2

sunshine

2024-11-05 69fd283ab500241947e8eac9192c39f4a6092b51

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package com.nova.sankuai.controller.v1;
 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.nova.sankuai.domain.entity.ProtocolInfo;
import com.nova.sankuai.domain.enums.SourceTypeEnum;
import com.nova.sankuai.domain.vo.ProtocolInfoVo;
import com.nova.sankuai.infra.constants.ResultCode;
import com.nova.sankuai.infra.utils.ResultJson;
import com.nova.sankuai.service.IProtocolInfoService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;
 
import java.util.ArrayList;
import java.util.List;
 
/**
 * @author weikangdi
 * @create 2021/10/29
 */
@Api(tags = "协议管理")
@RestController
@RequestMapping(value = "/api/v1/protocolInfo")
@AllArgsConstructor
public class ProtocolInfoController {
 
    private final IProtocolInfoService protocolInfoService;
 
    @GetMapping(value = "/getProtocol")
    @ApiOperation(value = "获取协议")
    public ResultJson<ProtocolInfo> getProtocol(String code) {
        ProtocolInfo protocolInfo = protocolInfoService.lambdaQuery().eq(ProtocolInfo::getCode, code).one();
        if (protocolInfo == null) {
            return ResultJson.failure(ResultCode.SERVER_ERROR, "没有找到对应的协议");
        }
        return ResultJson.ok(protocolInfo);
    }
 
 
    @PostMapping(value = "/edit")
    @ApiOperation(value = "编辑协议")
    @PreAuthorize("hasAuthority('protocolInfo.edit')")
    public ResultJson<Void> edit(@RequestBody ProtocolInfo protocolInfo) {
        protocolInfoService.edit(protocolInfo);
        return ResultJson.success();
    }
 
 
    @PostMapping(value = "/getProtocolList")
    @ApiOperation(value = "获取协议列表")
    @PreAuthorize("hasAuthority('protocolInfo.getProtocolList')")
    public ResultJson<List<ProtocolInfo>> getProtocolList(@RequestBody ProtocolInfoVo protocolInfoVo) {
        LambdaQueryWrapper<ProtocolInfo> wrapper = new LambdaQueryWrapper<>();
        wrapper.like(StringUtils.isNotBlank(protocolInfoVo.getName()), ProtocolInfo::getName, protocolInfoVo.getName());
        wrapper.like(StringUtils.isNotBlank(protocolInfoVo.getAlias()), ProtocolInfo::getAlias, protocolInfoVo.getAlias());
        List<ProtocolInfo> result = new ArrayList<>();
        List<ProtocolInfo> resultList = protocolInfoService.list(wrapper);
        for (ProtocolInfo protocolInfo : resultList) {
            protocolInfo.setTypeString(SourceTypeEnum.value(protocolInfo.getType()));
            result.add(protocolInfo);
        }
        return ResultJson.ok(result);
    }
 
}