From f2f1349eb1dc80024d95bc053d227486eaa21ba4 Mon Sep 17 00:00:00 2001
From: kangxiaoshan <kangxiaoshan@reyun.com>
Date: Mon, 20 Apr 2020 18:42:20 +0800
Subject: [PATCH] adtest 项目

---
 pom.xml                                                    |  3 +++
 src/main/java/common/controller/ContractController.java    | 13 +++++++++----
 src/main/java/common/service/impl/ContractServiceImpl.java | 36 +++++++++++++++++++++++++++---------
 src/main/java/util/Constant.java                           |  1 +
 4 files changed, 40 insertions(+), 13 deletions(-)

diff --git a/pom.xml b/pom.xml
index 753cd49..a197b9d 100644
--- a/pom.xml
+++ b/pom.xml
@@ -72,6 +72,7 @@
                 <mail.valid_time>3</mail.valid_time>
 
                 <report.url>http://10.3.20.41:9010</report.url>
+                <import.temp.url>http://10.3.20.41:9001/</import.temp.url>
 
                 <!--Redis setting-->
                 <redis.key.validtime>120</redis.key.validtime>
@@ -136,6 +137,7 @@
 
                 <!--<report.url>http://localhost:9010</report.url>-->
                 <report.url>http://10.3.20.41:11010</report.url>
+                <import.temp.url>http://10.3.20.41:9001/</import.temp.url>
                 <contract.uppath></contract.uppath>
                 <!--Redis setting-->
                 <redis.key.validtime>120</redis.key.validtime>
@@ -221,6 +223,7 @@
                 <mail.valid_time>3</mail.valid_time>
 
                 <report.url>http://internal-report-api-976419027.cn-north-1.elb.amazonaws.com.cn</report.url>
+                <import.temp.url>http://172.31.25.137:9001</import.temp.url>
                 <!--Redis setting-->
                 <redis.key.validtime>120</redis.key.validtime>
                 <redis.ip.requesttimes>5</redis.ip.requesttimes>
diff --git a/src/main/java/common/controller/ContractController.java b/src/main/java/common/controller/ContractController.java
index 8afeedd..7af4593 100644
--- a/src/main/java/common/controller/ContractController.java
+++ b/src/main/java/common/controller/ContractController.java
@@ -19,10 +19,7 @@ import org.springframework.web.multipart.MultipartFile;
 import org.springframework.web.multipart.commons.CommonsMultipartFile;
 import security.annotation.AuthKey;
 import security.annotation.CurrentAccount;
-import util.IPAddrUtil;
-import util.NewUserLogThread;
-import util.ResultModel;
-import util.ValidateUtil;
+import util.*;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
@@ -413,4 +410,12 @@ public class ContractController {
     }
 
 
+
+    @RequestMapping(value = "/upload/tempurl",method = RequestMethod.GET)
+    @ResponseBody
+    public ResultModel fileTempUrl() {
+        return ResultModel.OK(Constant.importTempUrl);
+    }
+
+
 }
diff --git a/src/main/java/common/service/impl/ContractServiceImpl.java b/src/main/java/common/service/impl/ContractServiceImpl.java
index f81dec6..21cdd1c 100644
--- a/src/main/java/common/service/impl/ContractServiceImpl.java
+++ b/src/main/java/common/service/impl/ContractServiceImpl.java
@@ -102,6 +102,8 @@ public class ContractServiceImpl implements ContractService {
         put("DMP", "dmp");
         put("防作弊卫士", "fake");
         put("ADI", "adi");
+        put("ADS", "ads");
+        put("ABTEST", "abtest");
     }};
 
     Logger logger = LoggerFactory.getLogger(ContractServiceImpl.class);
@@ -253,8 +255,16 @@ public class ContractServiceImpl implements ContractService {
             List<ContractBody> all = contractBodyRepository.findAllDis();
             return all;
         } else {
-            List<ContractBody> common = contractBodyRepository.findByPlatform("common");
+
+            List<ContractBody> common = null;
+            if (!"abtest".equals(platform) || !"ads".equals(platform)) {
+                common = contractBodyRepository.findByPlatform("common");
+            }
+
             List<ContractBody> byPlatform = contractBodyRepository.findByPlatform(platform);
+            if (common == null) {
+                common = new ArrayList<>();
+            }
             common.addAll(byPlatform);
 
             if (RoleEnum.FINANCE.getKey().equals(loginAccount.getRole())) {
@@ -365,6 +375,7 @@ public class ContractServiceImpl implements ContractService {
 //        }
 
         List list = new ArrayList();
+
         if ("tkio".equals(platform) || "io".equals(platform)) {
             List<PackageType> packageTypeList = packageTypeRepository.findIsNewAll();
             for (PackageType p : packageTypeList) {
@@ -374,9 +385,7 @@ public class ContractServiceImpl implements ContractService {
 
             return list;
         } else {
-
             List<PackageBase> packageBases = packageBaseRepository.findByPlatAndStatus(platform, 1);
-
             return packageBases;
         }
 
@@ -2082,7 +2091,7 @@ public class ContractServiceImpl implements ContractService {
 //            while (sheetIter.hasNext()) {
 //                Sheet sheet = sheetIter.next();
 //            }
-            return saveDataBySheet(workbook.getSheetAt(0), row_length);
+            return saveDataBySheet(workbook.getSheetAt(0), row_length, platform);
 
         } catch (Exception ex) {
             logger.error("批量上传合同错误", ex);
@@ -2091,12 +2100,16 @@ public class ContractServiceImpl implements ContractService {
     }
 
 
-    public ResultModel saveDataBySheet(Sheet sheet, int row_length) {
+    public ResultModel saveDataBySheet(Sheet sheet, int row_length, String platformexcl) {
         String sheetName = sheet.getSheetName();
         if (!SHEET_NAMES.containsKey(sheetName)) {
             return ResultModel.ERROR("模板标签页名称错误");
         }
 
+        if (sheetName.equals(platformexcl)) {
+            return ResultModel.ERROR("请上传对应项目模板处理的数据");
+        }
+
         String platform = SHEET_NAMES.get(sheetName);
         Map<String, Long> tradTypeMap = tradeTypeRepsitory.findAll()
                 .stream().collect(Collectors.toMap(TradeType::getName, TradeType::getId));
@@ -2112,9 +2125,14 @@ public class ContractServiceImpl implements ContractService {
                     .stream().collect(Collectors.toMap(PackageBase::getPackageName, PackageBase::getId));
         }
 
-        Map<String, String> cBodyMap = contractBodyRepository.findAllDis()
-                .stream().collect(Collectors.toMap(ContractBody::getName, ContractBody::getCode));
-
+        Map<String, String> cBodyMap = null;
+        if ("abtest".equals(platform) || "ads".equals(platform)) {
+            cBodyMap = contractBodyRepository.findByPlatform(platform)
+                    .stream().collect(Collectors.toMap(ContractBody::getName, ContractBody::getCode));
+        } else {
+            cBodyMap = contractBodyRepository.findAllDis()
+                    .stream().collect(Collectors.toMap(ContractBody::getName, ContractBody::getCode));
+        }
 
         Row row = sheet.getRow(0);
         List<String> titleKey = new ArrayList<>();
@@ -2189,7 +2207,7 @@ public class ContractServiceImpl implements ContractService {
                             DateTime dateTime = DateTime.parse(dataSTR, DateTimeFormat.forPattern(pattern));
                             s_data[w] = dateTime.toString("yyyy-MM-dd");
 
-                            if("start_date".equals(sheetTitle)){
+                            if ("start_date".equals(sheetTitle)) {
                                 ds = s_data[w].toString();
                             }
 
diff --git a/src/main/java/util/Constant.java b/src/main/java/util/Constant.java
index 13393c0..7ce4e7b 100644
--- a/src/main/java/util/Constant.java
+++ b/src/main/java/util/Constant.java
@@ -23,6 +23,7 @@ public class Constant {
 
 	public static ResourceBundle commonBundle = ResourceBundle.getBundle("common");
 	public static String reportUrl = commonBundle.getString("report.url");
+	public static String importTempUrl = commonBundle.getString("import.temp.url");
 
 	public static ResourceBundle redis = ResourceBundle.getBundle("redis");
 	public static String REDIS_KEY_VALID_TIME = redis.getString("redis.key.validtime");
--
libgit2 0.27.1