数据管理模块 公司接口

src/main/java/com/sky/ioc/controller/data/CompanyDataController.java

@@ -0,0 +1,39 @@
+package com.sky.ioc.controller.data;
+
+import com.sky.ioc.entity.domain.data.Building;
+import com.sky.ioc.entity.domain.data.CompanyData;
+import com.sky.ioc.service.data.CompanyDataService;
+import com.sky.ioc.tool.ReturnMsg;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+
+@Api(tags = "数据管理--公司")
+@Slf4j
+@RestController
+@RequestMapping("/data")
+public class CompanyDataController {
+
+    @Autowired
+    CompanyDataService companyDataService;
+
+    @ApiOperation("修改公司信息")
+    @PutMapping(value = "/company")
+    public ReturnMsg editCompanyData(CompanyData companyData) {
+        return companyDataService.editCompanyData(companyData);
+    }
+
+    @ApiOperation("添加公司信息")
+    @PostMapping(value = "/company")
+    public ReturnMsg addCompanyData(CompanyData companyData) {
+        return companyDataService.addCompanyData(companyData);
+    }
+
+    @ApiOperation("获取公司详情")
+    @GetMapping(value = "/company")
+    public ReturnMsg getCompanyDataList() {
+        return companyDataService.getList();
+    }
+}

src/main/java/com/sky/ioc/mapper/data/CompanyDataMapper.java

@@ -2,6 +2,8 @@ package com.sky.ioc.mapper.data;
 
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.sky.ioc.entity.domain.data.CompanyData;
+import org.apache.ibatis.annotations.Mapper;
 
+@Mapper
 public interface CompanyDataMapper extends BaseMapper<CompanyData> {
 }

src/main/java/com/sky/ioc/service/data/CompanyDataService.java

@@ -0,0 +1,14 @@
+package com.sky.ioc.service.data;
+
+import com.sky.ioc.entity.domain.data.Building;
+import com.sky.ioc.entity.domain.data.CompanyData;
+import com.sky.ioc.tool.ReturnMsg;
+
+public interface CompanyDataService {
+
+    ReturnMsg addCompanyData(CompanyData companyData);
+
+    ReturnMsg editCompanyData(CompanyData companyData);
+
+    ReturnMsg getList();
+}

src/main/java/com/sky/ioc/service/data/impl/CompanyDataServiceImpl.java

@@ -0,0 +1,39 @@
+package com.sky.ioc.service.data.impl;
+
+import com.sky.ioc.entity.domain.data.CompanyData;
+import com.sky.ioc.mapper.data.CompanyDataMapper;
+import com.sky.ioc.service.data.CompanyDataService;
+import com.sky.ioc.tool.ReturnMsg;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+@Service
+public class CompanyDataServiceImpl implements CompanyDataService {
+
+    @Autowired
+    CompanyDataMapper companyDataMapper;
+
+    @Override
+    public ReturnMsg addCompanyData(CompanyData companyData) {
+        companyDataMapper.insert(companyData);
+        return ReturnMsg.ok();
+    }
+
+    @Override
+    public ReturnMsg editCompanyData(CompanyData companyData) {
+        if(companyData.getId()!=null){
+            companyDataMapper.updateById(companyData);
+            return ReturnMsg.ok();
+        }else{
+          return ReturnMsg.fail("公司id不能为空");
+        }
+    }
+
+    @Override
+    public ReturnMsg getList() {
+        List<CompanyData> lists = companyDataMapper.selectList(null);
+        return ReturnMsg.ok(lists);
+    }
+}