数据源切换功能

huimv-env-platform/huimv-env-produce/src/main/java/com/huimv/env/produce/controller/DataSourceController.java

@@ -0,0 +1,42 @@
+package com.huimv.env.produce.controller;
+
+
+import com.huimv.common.utils.Result;
+import com.huimv.env.produce.service.DataSourceService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+
+import org.springframework.web.bind.annotation.RestController;
+
+import java.text.ParseException;
+import java.util.Map;
+
+/**
+ * <p>
+ *  前端控制器
+ * </p>
+ *
+ * @author zn
+ * @since 2022-11-01
+ */
+@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-env-platform/huimv-env-produce/src/main/java/com/huimv/env/produce/entity/DataSource.java

@@ -0,0 +1,32 @@
+package com.huimv.env.produce.entity;
+
+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-11-01
+ */
+@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-env-platform/huimv-env-produce/src/main/java/com/huimv/env/produce/mapper/DataSourceMapper.java

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

huimv-env-platform/huimv-env-produce/src/main/java/com/huimv/env/produce/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.env.produce.mapper.DataSourceMapper">
+
+    <!-- 通用查询映射结果 -->
+    <resultMap id="BaseResultMap" type="com.huimv.env.produce.entity.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-env-platform/huimv-env-produce/src/main/java/com/huimv/env/produce/service/DataSourceService.java

@@ -0,0 +1,22 @@
+package com.huimv.env.produce.service;
+
+import com.huimv.common.utils.Result;
+import com.huimv.env.produce.entity.DataSource;
+import com.baomidou.mybatisplus.extension.service.IService;
+
+import java.util.Map;
+
+/**
+ * <p>
+ *  服务类
+ * </p>
+ *
+ * @author zn
+ * @since 2022-11-01
+ */
+public interface DataSourceService extends IService<DataSource> {
+
+    Result getDataSource(Map<String, String> paramsMap);
+
+    Result setDataSouce(Map<String, String> paramsMap);
+}

huimv-env-platform/huimv-env-produce/src/main/java/com/huimv/env/produce/service/impl/DataSourceServiceImpl.java

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