java-insagent-pc/src/main/java/com/ydtech/modules/message/controller/SystemMessageController.java

package com.ydtech.modules.message.controller;

import com.github.pagehelper.PageInfo;
import com.ydtech.core.page.HttpResult;
import com.ydtech.modules.base.controller.BaseController;
import com.ydtech.modules.message.entity.SystemMessageVo;
import com.ydtech.modules.message.entity.dto.SystemMessageDTO;
import com.ydtech.modules.message.service.SystemMessageService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;


@Slf4j
@Api(tags = "站内信")
@RestController
@RequestMapping("/sys/message")
@RequiredArgsConstructor
public class SystemMessageController {

    @Autowired
    private SystemMessageService systemMessageService;

    @PostMapping("/get")
    @ApiOperation("获取站内信信息")
    public HttpResult<PageInfo<SystemMessageDTO>> getUserMessage(@RequestBody SystemMessageVo systemMessageVo) {
        String userId = BaseController.getUserId();
        systemMessageVo.setUserId(userId);
        try {
            PageInfo<SystemMessageDTO> systemMessageByUser = systemMessageService.getSystemMessageByUser(systemMessageVo);
            return HttpResult.ok("success", systemMessageByUser);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return HttpResult.error("接口异常请重试!");
        }
    }

    @PostMapping("/update")
    @ApiOperation("修改站内信信息")
    public HttpResult updateUserMessage(@RequestParam Integer id) {
        try {
            systemMessageService.updateSystemMessageByUser(id);
            return HttpResult.ok("success");
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return HttpResult.error("接口异常请重试!");
        }
    }
}