Code/user_center/src/main/java/com/huaxu/controller/TenantController.java

package com.huaxu.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.huaxu.common.FileUploadUtil;
import com.huaxu.entity.TenantEntity;
import com.huaxu.model.AjaxMessage;
import com.huaxu.model.Pagination;
import com.huaxu.model.ResultStatus;
import com.huaxu.service.TenantService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.List;

/**
 * 租户(Tenant)控制层
 *
 * @author makejava
 * @since 2020-10-20 16:56:39
 */
@RestController
@RequestMapping("/tenant")
@Api(tags = "租户接口")
public class TenantController {
    /**
     * 租户
     */
    @Autowired
    private TenantService tenantService;
    @Value("${UMIS.sys_config_path}")
    private String baseDir;

    /**
     * 通过主键查询单条数据
     *
     * @param id
     * @return 单条数据
     */
    @RequestMapping(value = "get", method = RequestMethod.POST)
    @ApiOperation(value = "根据id查询租户")
    public AjaxMessage<TenantEntity> selectOne(
            @ApiParam(value = "租户信息", required = true) Integer id) {
        TenantEntity result = tenantService.selectById(id);

        return new AjaxMessage<>(ResultStatus.OK, result);
    }

    /**
     * 查询租户信息
     *
     * @param tenantEntity 参数对象
     * @return 多条数据
     */
    @RequestMapping(value = "selectList", method = RequestMethod.POST)
    @ApiOperation(value = "查询租户信息")
    public AjaxMessage<List<TenantEntity>> selectList(
            @ApiParam(value = "租户信息", required = true) @RequestBody TenantEntity tenantEntity) {
        List<TenantEntity> result = tenantService.selectList(tenantEntity);

        return new AjaxMessage<>(ResultStatus.OK, result);
    }

    /**
     * 新增
     */
    @RequestMapping(value = "addTenantLogo", method = RequestMethod.POST)
    @ResponseBody
    @ApiOperation(value = "租户logo")
    public  AjaxMessage<String>  addUserPhoto(@ApiParam(value = "租户logo", required = true)@RequestParam("avatarfile") MultipartFile file) {
        String avatar = "";
        if (!file.isEmpty()) {
            try {
                avatar = FileUploadUtil.uploadWeb(baseDir, file);
            } catch (IOException e) {
                return new AjaxMessage<>(ResultStatus.ERROR, e.getMessage());
            }
        }
        return new AjaxMessage<>(ResultStatus.OK, avatar);
    }

    /**
     * 新增一条数据
     *
     * @param tenantEntity 实体类
     * @return Response对象
     */
    @RequestMapping(value = "insert", method = RequestMethod.POST)
    @ApiOperation(value = "插入租户信息")
    public AjaxMessage<Integer> insert(@ApiParam(value = "租户信息", required = true) @RequestBody TenantEntity tenantEntity) {
        int result = tenantService.insert(tenantEntity);

        return new AjaxMessage<>(ResultStatus.OK, result);
    }

    /**
     * 修改一条数据
     *
     * @param tenantEntity 实体类
     * @return Response对象
     */
    @RequestMapping(value = "update", method = RequestMethod.POST)
    @ApiOperation(value = "修改租户信息")
    public AjaxMessage<Integer> update(@ApiParam(value = "租户信息", required = true) @RequestBody TenantEntity tenantEntity) {
        int result = tenantService.update(tenantEntity);
        return new AjaxMessage<>(ResultStatus.OK, result);

    }

    /**
     * 单条删除数据
     *
     * @param id
     * @return Response对象
     */
    @RequestMapping(value = "delete", method = RequestMethod.POST)
    @ApiOperation(value = "删除租户信息")
    public AjaxMessage<Integer> delete(@ApiParam(value = "租户信息", required = true) Integer id) {
        int result = tenantService.deleteById(id);
        return new AjaxMessage<>(ResultStatus.OK, result);
    }


    /**
     * 分页查询
     *
     * @param pageNum  偏移
     * @param pageSize 条数
     * @return Response对象
     */
    @RequestMapping(value = "selectPage", method = RequestMethod.POST)
    @ApiOperation(value = "查询租户信息列表")
    public AjaxMessage<Pagination<TenantEntity>> selectPage(
            @ApiParam(value = "租户信息", required = true) @RequestBody TenantEntity tenantEntity,
            @ApiParam(value = "页数,非必传，默认第一页",  defaultValue = "1")  int pageNum,
            @ApiParam(value = "条数，非必传，默认15条",  defaultValue = "30")  int pageSize

    ) {
        IPage<TenantEntity> iPage = new Page<>(pageNum, pageSize);

        iPage = tenantService.selectPage(tenantEntity, iPage);

        Pagination<TenantEntity> pages = new Pagination<>(iPage);

        return new AjaxMessage<>(ResultStatus.OK, pages);
    }

}