|
@@ -24,10 +24,12 @@ import com.ym.mec.common.page.PageInfo;
|
|
import com.ym.mec.common.page.PageUtil;
|
|
import com.ym.mec.common.page.PageUtil;
|
|
import com.ym.mec.common.page.WrapperUtil;
|
|
import com.ym.mec.common.page.WrapperUtil;
|
|
import com.ym.mec.common.service.IdGeneratorService;
|
|
import com.ym.mec.common.service.IdGeneratorService;
|
|
|
|
+import com.ym.mec.common.tenant.TenantContextHolder;
|
|
import com.ym.mec.thirdparty.yqpay.DateUtils;
|
|
import com.ym.mec.thirdparty.yqpay.DateUtils;
|
|
import com.ym.mec.thirdparty.yqpay.Msg;
|
|
import com.ym.mec.thirdparty.yqpay.Msg;
|
|
import com.ym.mec.util.date.DateUtil;
|
|
import com.ym.mec.util.date.DateUtil;
|
|
import org.apache.commons.collections.CollectionUtils;
|
|
import org.apache.commons.collections.CollectionUtils;
|
|
|
|
+import org.apache.commons.lang3.StringUtils;
|
|
import org.joda.time.LocalDate;
|
|
import org.joda.time.LocalDate;
|
|
import org.joda.time.LocalDateTime;
|
|
import org.joda.time.LocalDateTime;
|
|
import org.redisson.api.RBucket;
|
|
import org.redisson.api.RBucket;
|
|
@@ -40,6 +42,7 @@ import org.springframework.dao.DuplicateKeyException;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
|
|
|
+import javax.servlet.http.HttpServletRequest;
|
|
import java.math.BigDecimal;
|
|
import java.math.BigDecimal;
|
|
import java.util.*;
|
|
import java.util.*;
|
|
import java.util.concurrent.TimeUnit;
|
|
import java.util.concurrent.TimeUnit;
|
|
@@ -664,5 +667,23 @@ public class TenantInfoServiceImpl extends ServiceImpl<TenantInfoDao, TenantInfo
|
|
return tps;
|
|
return tps;
|
|
}
|
|
}
|
|
|
|
|
|
-
|
|
|
|
|
|
+ @Override
|
|
|
|
+ public void saveTenantContextHolder(HttpServletRequest request) {
|
|
|
|
+ String tenantId = request.getHeader("tenantId");
|
|
|
|
+ if(StringUtils.isEmpty(tenantId)){
|
|
|
|
+ SysUser sysUser = sysUserFeignService.queryUserInfo();
|
|
|
|
+ if(sysUser != null && sysUser.getTenantId() != null){
|
|
|
|
+ tenantId = sysUser.getTenantId().toString();
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ if(StringUtils.isNotEmpty(tenantId)){
|
|
|
|
+ TenantInfo tenantInfo = this.baseMapper.getOpenTenant(Integer.parseInt(tenantId));
|
|
|
|
+ if(tenantInfo != null){
|
|
|
|
+ TenantContextHolder.setTenantId(Integer.parseInt(tenantId));
|
|
|
|
+ request.setAttribute("datasourceId", tenantInfo.getDataSource());
|
|
|
|
+ }else {
|
|
|
|
+ throw new BizException("机构信息异常,请联系管理员");
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
}
|
|
}
|