活码

haoban-manage3-service/src/main/java/com/gic/haoban/manage/service/service/hm/impl/HmLinkServiceImpl.java

@@ -3,6 +3,7 @@ package com.gic.haoban.manage.service.service.hm.impl;
 import java.util.Date;
 import java.util.List;
 
+import org.apache.commons.collections.CollectionUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -44,18 +45,18 @@ public class HmLinkServiceImpl implements HmLinkService {
     @Override
     public HmLinkBO getByLinkId(String wxEnterpriseId, String enterprieId, Long linkId) {
         TabHmLink link = this.linkMapper.selectById(linkId);
-        if(null == link) {
-            return null ;
+        if (null == link) {
+            return null;
         }
         HmLinkBO bo = EntityUtil.changeEntityByJSON(HmLinkBO.class, link);
         return bo;
     }
-    
+
     @Override
     public HmLinkBO getByLinkShortCode(String shortCode) {
         TabHmLink link = this.linkMapper.selectByShortCode(shortCode);
-        if(null == link) {
-            return null ;
+        if (null == link) {
+            return null;
         }
         HmLinkBO bo = EntityUtil.changeEntityByJSON(HmLinkBO.class, link);
         return bo;
@@ -99,6 +100,12 @@ public class HmLinkServiceImpl implements HmLinkService {
             TabHmLink entity = EntityUtil.changeEntityByJSON(TabHmLink.class, dto);
             this.linkMapper.update(entity);
         }
+        if (CollectionUtils.isNotEmpty(storeList)) {
+            storeList.forEach(item -> {
+                item.setWxEnterpriseId(dto.getWxEnterpriseId());
+                item.setEnterpriseId(dto.getEnterpriseId());
+            });
+        }
         this.hmLinkStoreService.saveStore(linkId, storeList);
         return dto.getLinkCode();
     }
@@ -134,12 +141,12 @@ public class HmLinkServiceImpl implements HmLinkService {
             sb.append(PADDING[x % totalLength]);
         }
         sb.append(PADDING[(int) (Math.random() * totalLength)]);
-        String code =  sb.toString();
-        TabHmLink link = this.linkMapper.selectByShortCode(code) ;
-        if(null == link) {
-            return code ;
+        String code = sb.toString();
+        TabHmLink link = this.linkMapper.selectByShortCode(code);
+        if (null == link) {
+            return code;
         }
-        return shortCode(enterpriseId) ;
+        return shortCode(enterpriseId);
     }
 
 }