增加数据源模块

huimv-cattle/src/main/java/com/huimv/cattle/controller/DataSourceController.java

@@ -0,0 +1,39 @@
+package com.huimv.cattle.controller;
+
+
+import com.huimv.cattle.service.DataSourceService;
+import com.huimv.common.utils.Result;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+
+import java.text.ParseException;
+import java.util.Map;
+
+/**
+ * <p>
+ *  前端控制器
+ * </p>
+ *
+ * @author zn
+ * @since 2022-12-15
+ */
+@CrossOrigin
+@RestController
+@RequestMapping("/v1.0.0/datasource")
+public class DataSourceController {
+    @Autowired
+    private DataSourceService dataSourceService;
+
+    //查询
+    @PostMapping("/getDataSource")
+    public Result getDataSource(@RequestBody Map<String, String> paramsMap) throws ParseException {
+        return dataSourceService.getDataSource(paramsMap);
+    }
+
+    //新增
+    @PostMapping("/setDataSouce")
+    public Result setDataSouce(@RequestBody Map<String, String> paramsMap) throws ParseException {
+        return dataSourceService.setDataSouce(paramsMap);
+    }
+}
+

huimv-cattle/src/main/java/com/huimv/cattle/mapper/DataSourceMapper.java

@@ -0,0 +1,16 @@
+package com.huimv.cattle.mapper;
+
+import com.huimv.cattle.pojo.DataSource;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+
+/**
+ * <p>
+ *  Mapper 接口
+ * </p>
+ *
+ * @author zn
+ * @since 2022-12-15
+ */
+public interface DataSourceMapper extends BaseMapper<DataSource> {
+
+}

huimv-cattle/src/main/java/com/huimv/cattle/mapper/xml/DataSourceMapper.xml

@@ -0,0 +1,17 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.huimv.cattle.mapper.DataSourceMapper">
+
+    <!-- 通用查询映射结果 -->
+    <resultMap id="BaseResultMap" type="com.huimv.cattle.pojo.DataSource">
+        <id column="id" property="id" />
+        <result column="ds_status" property="dsStatus" />
+        <result column="farm_code" property="farmCode" />
+    </resultMap>
+
+    <!-- 通用查询结果列 -->
+    <sql id="Base_Column_List">
+        id, ds_status, farm_code
+    </sql>
+
+</mapper>

huimv-cattle/src/main/java/com/huimv/cattle/pojo/DataSource.java

@@ -0,0 +1,32 @@
+package com.huimv.cattle.pojo;
+
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.Version;
+import com.baomidou.mybatisplus.annotation.TableId;
+import java.io.Serializable;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+/**
+ * <p>
+ * 
+ * </p>
+ *
+ * @author zn
+ * @since 2022-12-15
+ */
+@Data
+@EqualsAndHashCode(callSuper = false)
+public class DataSource implements Serializable {
+
+    private static final long serialVersionUID=1L;
+
+      @TableId(value = "id", type = IdType.AUTO)
+    private Integer id;
+
+    private Integer dsStatus;
+
+    private String farmCode;
+
+
+}

huimv-cattle/src/main/java/com/huimv/cattle/service/DataSourceService.java

@@ -0,0 +1,23 @@
+package com.huimv.cattle.service;
+
+import com.huimv.cattle.pojo.DataSource;
+import com.baomidou.mybatisplus.extension.service.IService;
+import com.huimv.common.utils.Result;
+
+import java.util.Map;
+
+/**
+ * <p>
+ *  服务类
+ * </p>
+ *
+ * @author zn
+ * @since 2022-12-15
+ */
+public interface DataSourceService extends IService<DataSource> {
+    Result getDataSource(Map<String, String> paramsMap);
+
+    Result setDataSouce(Map<String, String> paramsMap);
+
+    Integer getDataSourceStatus(Map<String, String> paramsMap);
+}

huimv-cattle/src/main/java/com/huimv/cattle/service/impl/DataSourceServiceImpl.java

@@ -0,0 +1,66 @@
+package com.huimv.cattle.service.impl;
+
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.huimv.cattle.pojo.DataSource;
+import com.huimv.cattle.mapper.DataSourceMapper;
+import com.huimv.cattle.service.DataSourceService;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.huimv.common.utils.Result;
+import com.huimv.common.utils.ResultCode;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.util.Map;
+
+/**
+ * <p>
+ * 服务实现类
+ * </p>
+ *
+ * @author zn
+ * @since 2022-12-15
+ */
+@Service
+public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSource> implements DataSourceService {
+    //默认牧场
+//    private static final String DEFAULT_FARM_CODE = "330211";
+    @Autowired
+    private DataSourceMapper dataSourceMapper;
+
+    @Override
+    public Result getDataSource(Map<String, String> paramsMap) {
+        QueryWrapper<DataSource> totalQueryWrapper = new QueryWrapper();
+        return new Result(ResultCode.SUCCESS, dataSourceMapper.selectOne(totalQueryWrapper));
+    }
+
+    @Override
+    public Result setDataSouce(Map<String, String> paramsMap) {
+        //
+        String datasourceStatus = "0";
+        if (paramsMap.get("datasourceStatus") != null) {
+            datasourceStatus = paramsMap.get("datasourceStatus");
+        }
+        QueryWrapper<DataSource> totalQueryWrapper = new QueryWrapper();
+        DataSource dataSource = dataSourceMapper.selectOne(totalQueryWrapper);
+        if (dataSource == null) {
+            dataSource = new DataSource();
+            dataSource.setDsStatus(Integer.parseInt(datasourceStatus));
+            dataSourceMapper.insert(dataSource);
+        } else {
+            dataSource.setDsStatus(Integer.parseInt(datasourceStatus));
+            dataSourceMapper.updateById(dataSource);
+        }
+        return new Result(ResultCode.SUCCESS);
+    }
+
+    @Override
+    public Integer getDataSourceStatus(Map<String, String> paramsMap) {
+        QueryWrapper<DataSource> totalQueryWrapper = new QueryWrapper();
+        DataSource dataSource = dataSourceMapper.selectOne(totalQueryWrapper);
+        if (dataSource == null) {
+            return 0;
+        } else {
+            return dataSource.getDsStatus();
+        }
+    }
+}