企业初始化

This commit is contained in:
zhangchenbiao
2023-05-29 14:08:05 +08:00
parent 10ea204bfd
commit 112d874931
54 changed files with 1915 additions and 62 deletions

View File

@@ -1,7 +1,10 @@
package com.cool.store.http;
import com.alibaba.fastjson.JSONObject;
import com.cool.store.dto.enterprise.EnterpriseUserDTO;
import com.cool.store.dto.login.UserIdInfoDTO;
import com.cool.store.utils.RestTemplateUtil;
import com.coolstore.base.dto.ResultDTO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.ResponseEntity;
@@ -26,18 +29,33 @@ public class ISVHttpRequest {
private String getUserIdByCode;
public JSONObject getUserIdByCode(String paramStr){
String url = isvDomain + getUserIdByCode + paramStr;
ResponseEntity<JSONObject> responseEntity = null;
public UserIdInfoDTO getUserIdByCode(String paramStr){
String url = isvDomain + "/isv/user/getUserIdByCode" + paramStr;
ResponseEntity<ResultDTO> responseEntity = null;
try {
responseEntity = RestTemplateUtil.loadGet(url, JSONObject.class);
responseEntity = RestTemplateUtil.loadGet(url, ResultDTO.class);
log.info("url:{}, response:{}", url, JSONObject.toJSONString(responseEntity));
if(Objects.nonNull(responseEntity.getBody()) && responseEntity.getBody().isSuccess()){
return JSONObject.parseObject(JSONObject.toJSONString(responseEntity.getBody().getData()), UserIdInfoDTO.class);
}
} catch (Exception e) {
log.info("调用isv出错{}", e);
}
if(Objects.isNull(responseEntity)){
return null;
return null;
}
public EnterpriseUserDTO getUserDetailByUserId(String corpId, String userId, String appType){
String url = isvDomain + "/isv/user/getUserDetail";
ResponseEntity<ResultDTO> responseEntity = null;
try {
responseEntity = RestTemplateUtil.loadGet(url, ResultDTO.class);
log.info("url:{}, response:{}", url, JSONObject.toJSONString(responseEntity));
if(Objects.nonNull(responseEntity.getBody()) && responseEntity.getBody().isSuccess()){
return JSONObject.parseObject(JSONObject.toJSONString(responseEntity.getBody().getData()), EnterpriseUserDTO.class);
}
} catch (Exception e) {
log.info("调用isv出错{}", e);
}
return responseEntity.getBody();
return null;
}
}

View File

@@ -0,0 +1,23 @@
package com.cool.store.mq;
import com.aliyun.openservices.ons.api.SendResult;
import com.coolstore.base.enums.RocketMqTagEnum;
/**
* @author zhangchenbiao
* @FileName: OrderMessageService
* @Description: 顺序消息生产者
* @date 2021-12-22 17:37
*/
public interface OrderMessageService {
/**
* 发送顺序消息
* @param message
* @param tag
* @param shardingKey
* @return
*/
SendResult send(String message, RocketMqTagEnum tag, final String shardingKey);
}

View File

@@ -0,0 +1,45 @@
package com.cool.store.mq;
import com.aliyun.openservices.ons.api.bean.OrderProducerBean;
import com.aliyun.openservices.ons.api.bean.ProducerBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
/**
* @author zhangchenbiao
* @FileName: ProducerClient
* @Description: 普通消息client
* @date 2021-12-21 11:33
*/
@Configuration
public class ProducerClient {
@Autowired
private RocketMqConfig rocketMqConfig;
/**
* 普通消息
* @return
*/
@Primary
@Bean(initMethod = "start", destroyMethod = "shutdown")
public ProducerBean producerBean() {
ProducerBean producer = new ProducerBean();
producer.setProperties(rocketMqConfig.getMqProperties());
return producer;
}
/**
* 分区顺序消息
* @return
*/
@Bean(initMethod = "start", destroyMethod = "shutdown")
public OrderProducerBean orderProducerBean() {
OrderProducerBean producer = new OrderProducerBean();
producer.setProperties(rocketMqConfig.getMqProperties());
return producer;
}
}

View File

@@ -0,0 +1,78 @@
package com.cool.store.mq;
import com.aliyun.openservices.ons.api.PropertyKeyConst;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;
import java.util.Properties;
/**
* @author zhangchenbiao
* @FileName: RocketMqConfig
* @Description: 该服务的rocketmq的配置 以及生产者的topic
* @date 2021-12-21 11:33
*/
@Configuration
@ConfigurationProperties(prefix = "rocketmq")
public class RocketMqConfig {
private String accessKey;
private String secretKey;
private String nameSrvAdder;
/**
* 普通消息的topic
*/
private String topic;
/**
* 分区顺序消息topic
*/
private String orderTopic;
public Properties getMqProperties() {
Properties properties = new Properties();
properties.setProperty(PropertyKeyConst.AccessKey, this.accessKey);
properties.setProperty(PropertyKeyConst.SecretKey, this.secretKey);
properties.setProperty(PropertyKeyConst.NAMESRV_ADDR, this.nameSrvAdder);
return properties;
}
public String getAccessKey() {
return accessKey;
}
public void setAccessKey(String accessKey) {
this.accessKey = accessKey;
}
public String getSecretKey() {
return secretKey;
}
public void setSecretKey(String secretKey) {
this.secretKey = secretKey;
}
public String getNameSrvAdder() {
return nameSrvAdder;
}
public void setNameSrvAdder(String nameSrvAdder) {
this.nameSrvAdder = nameSrvAdder;
}
public String getTopic() {
return topic;
}
public void setTopic(String topic) {
this.topic = topic;
}
public String getOrderTopic() {
return orderTopic;
}
public void setOrderTopic(String orderTopic) {
this.orderTopic = orderTopic;
}
}

View File

@@ -0,0 +1,46 @@
package com.cool.store.mq;
import com.aliyun.openservices.ons.api.SendCallback;
import com.aliyun.openservices.ons.api.SendResult;
import com.coolstore.base.enums.RocketMqTagEnum;
/**
* @author zhangchenbiao
* @FileName: MessageProducerService
* @Description: rocketmq 消息生产者
* @date 2021-12-22 16:12
*/
public interface SimpleMessageService {
/**
* 同步发送
* @param message 消息
* @param tag tag
* @return
*/
SendResult send(String message, RocketMqTagEnum tag);
/**
* 单向发送
* @param message 消息
* @param tag tag
*/
void sendOneway(String message, RocketMqTagEnum tag);
/**
* 异步发送
* @param message 消息
* @param tag tag
* @param sendCallback 回调
*/
void sendAsync(String message, RocketMqTagEnum tag, SendCallback sendCallback);
/**
* 同步发送(延时)
* @param message
* @param tag
* @param startDeliverTime
* @return
*/
SendResult send(String message, RocketMqTagEnum tag, Long startDeliverTime);
}

View File

@@ -0,0 +1,57 @@
package com.cool.store.mq.impl;
import com.aliyun.openservices.ons.api.Message;
import com.aliyun.openservices.ons.api.SendResult;
import com.aliyun.openservices.ons.api.bean.OrderProducerBean;
import com.cool.store.constants.CommonConstants;
import com.cool.store.mq.OrderMessageService;
import com.cool.store.mq.RocketMqConfig;
import com.cool.store.utils.UUIDUtils;
import com.coolstore.base.enums.RocketMqTagEnum;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.MDC;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.Objects;
import java.util.Properties;
/**
* @author zhangchenbiao
* @FileName: OrderMessageServiceImpl
* @Description: 顺序消息生产实现
* @date 2021-12-22 17:38
*/
@Slf4j
@Service
public class OrderMessageServiceImpl implements OrderMessageService {
@Resource
private RocketMqConfig rocketMqConfig;
@Resource
private OrderProducerBean orderProducerBean;
@Override
public SendResult send(String message, RocketMqTagEnum tag, String shardingKey) {
if(StringUtils.isAnyBlank(message, shardingKey) || Objects.isNull(tag)){
return new SendResult();
}
try {
Message msg = new Message(rocketMqConfig.getOrderTopic(), tag.getTag(), message.getBytes("UTF-8"));
Properties properties = new Properties();
String requestId = MDC.get(CommonConstants.REQUEST_ID);
if(StringUtils.isBlank(requestId)){
requestId = UUIDUtils.get32UUID();
}
properties.setProperty(CommonConstants.REQUEST_ID, requestId);
msg.setUserProperties(properties);
SendResult send = orderProducerBean.send(msg, shardingKey);
return send;
} catch (Exception e) {
log.error("send@@@@@@@@@ddd", e);
}
return null;
}
}

View File

@@ -0,0 +1,129 @@
package com.cool.store.mq.impl;
import com.alibaba.fastjson.JSONObject;
import com.aliyun.openservices.ons.api.Message;
import com.aliyun.openservices.ons.api.SendCallback;
import com.aliyun.openservices.ons.api.SendResult;
import com.aliyun.openservices.ons.api.bean.ProducerBean;
import com.cool.store.constants.CommonConstants;
import com.cool.store.mq.RocketMqConfig;
import com.cool.store.mq.SimpleMessageService;
import com.cool.store.utils.UUIDUtils;
import com.coolstore.base.enums.RocketMqTagEnum;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.MDC;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.Date;
import java.util.Objects;
import java.util.Properties;
/**
* @author zhangchenbiao
* @FileName: MessageProducerServiceImpl
* @Description: 消息生产者实现
* @date 2021-12-22 16:12
*/
@Service
@Slf4j
public class SimpleMessageServiceImpl implements SimpleMessageService {
@Resource
private RocketMqConfig rocketMqConfig;
@Resource
private ProducerBean producer;
@Override
public SendResult send(String message, RocketMqTagEnum tag) {
if(StringUtils.isBlank(message) || Objects.isNull(tag)){
return new SendResult();
}
try {
Message msg = new Message(rocketMqConfig.getTopic(), tag.getTag(), message.getBytes("UTF-8"));
Properties properties = new Properties();
String requestId = MDC.get(CommonConstants.REQUEST_ID);
if(StringUtils.isBlank(requestId)){
requestId = UUIDUtils.get32UUID();
}
properties.setProperty(CommonConstants.REQUEST_ID, requestId);
msg.setUserProperties(properties);
SendResult send = producer.send(msg);
log.info("消息发送send response:{}", JSONObject.toJSONString(send));
return send;
} catch (Exception e) {
log.error("send#######", e);
}
return new SendResult();
}
@Override
public void sendOneway(String message, RocketMqTagEnum tag) {
if(StringUtils.isBlank(message) || Objects.isNull(tag)){
return;
}
try {
Message msg = new Message(rocketMqConfig.getTopic(), tag.getTag(), message.getBytes("UTF-8"));
Properties properties = new Properties();
String requestId = MDC.get(CommonConstants.REQUEST_ID);
if(StringUtils.isBlank(requestId)){
requestId = UUIDUtils.get32UUID();
}
properties.setProperty(CommonConstants.REQUEST_ID, requestId);
msg.setUserProperties(properties);
producer.sendOneway(msg);
} catch (Exception e) {
log.error("send@@@@@@", e);
}
}
@Override
public void sendAsync(String message, RocketMqTagEnum tag, SendCallback sendCallback) {
if(StringUtils.isBlank(message) || Objects.isNull(tag)){
return;
}
try {
Message msg = new Message(rocketMqConfig.getTopic(), tag.getTag(), message.getBytes("UTF-8"));
Properties properties = new Properties();
String requestId = MDC.get(CommonConstants.REQUEST_ID);
if(StringUtils.isBlank(requestId)){
requestId = UUIDUtils.get32UUID();
}
properties.setProperty(CommonConstants.REQUEST_ID, requestId);
msg.setUserProperties(properties);
producer.sendAsync(msg, sendCallback);
} catch (Exception e) {
log.error("sendAsync@@@@@@", e);
}
}
@Override
public SendResult send(String message, RocketMqTagEnum tag, Long startDeliverTime) {
if(StringUtils.isBlank(message) || Objects.isNull(tag)){
return new SendResult();
}
try {
Message msg = new Message(rocketMqConfig.getTopic(), tag.getTag(), message.getBytes("UTF-8"));
if(Objects.nonNull(startDeliverTime)) {
log.info("{} startDeliverTime:{}", tag.getTag(),new Date(startDeliverTime));
msg.setStartDeliverTime(startDeliverTime);
}
Properties properties = new Properties();
String requestId = MDC.get(CommonConstants.REQUEST_ID);
if(StringUtils.isBlank(requestId)){
requestId = UUIDUtils.get32UUID();
}
properties.setProperty(CommonConstants.REQUEST_ID, requestId);
msg.setUserProperties(properties);
SendResult send = producer.send(msg);
log.info("发送消息data:{}", JSONObject.toJSONString(send));
return send;
} catch (Exception e) {
log.error("send@@@@@@@@@", e);
}
return new SendResult();
}
}

View File

@@ -0,0 +1,197 @@
package com.cool.store.mq.util;
import com.aliyun.ons20190214.*;
import com.aliyun.ons20190214.models.*;
import com.aliyun.teaopenapi.models.Config;
import com.cool.store.constants.CommonConstants;
import com.coolstore.base.enums.RocketMqGroupEnum;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.ListUtils;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
* @author chenyupeng
* @since 2022/1/5
*/
@Slf4j
public class RocketMqUtil {
private static final String INSTANCE_ID = "MQ_INST_1947409023213164_BX3sLZnA";
/**
* 本地
*/
private static final String LOCAL_INSTANCE_ID = "MQ_INST_1255228665351616_BX3wfyPO";
private static final String INSTANCE_ID_ONLINE_HD_PRE = "MQ_INST_1947409023213164_BX5N7rwl";
private static final String ACCESS_KEY = "LTAI5t5ouXZuFgxJMbQea3b2";
private static final String ACCESS_SECRET = "yuomDstRjSdihtN5zo8viDbWu8Z0ig";
/**
* 使用AK&SK初始化账号Client
* @param accessKeyId
* @param accessKeySecret
* @return Client
* @throws Exception
*/
public static Client createClient(String accessKeyId, String accessKeySecret) throws Exception {
Config config = new Config()
// 您的AccessKey ID
.setAccessKeyId(accessKeyId)
// 您的AccessKey Secret
.setAccessKeySecret(accessKeySecret);
// 访问的域名
config.endpoint = "ons.cn-hangzhou.aliyuncs.com";
return new Client(config);
}
public static Client createLocalClient(String accessKeyId, String accessKeySecret) throws Exception {
Config config = new Config()
// 您的AccessKey ID
.setAccessKeyId(accessKeyId)
// 您的AccessKey Secret
.setAccessKeySecret(accessKeySecret);
// 访问的域名
config.endpoint = "ons.mq-internet-access.aliyuncs.com";
return new Client(config);
}
/**
* 新增测试、开发环境的group
* @author chenyupeng
* @date 2022/1/13
* @param profiles 环境后缀,逗号分隔 例如: dev,dev2
* @return void
*/
public static void initGroup(String profiles) throws Exception {
Client client = RocketMqUtil.createClient(ACCESS_KEY, ACCESS_SECRET);
List<String> profileList = Arrays.asList(profiles.split(CommonConstants.COMMA));
if(CollectionUtils.isEmpty(profileList)){
return;
}
OnsGroupListRequest onsGroupListRequest = new OnsGroupListRequest();
onsGroupListRequest.setInstanceId(INSTANCE_ID);
OnsGroupListResponse onsGroupListResponse = client.onsGroupList(onsGroupListRequest);
List<OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo> subscribeInfoDo = onsGroupListResponse.getBody().getData().getSubscribeInfoDo();
Map<String, OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo> collect = ListUtils.emptyIfNull(subscribeInfoDo).stream().
collect(Collectors.toMap(OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo::getGroupId, data -> data, (a, b) -> a));
for (String profile : profileList) {
for (RocketMqGroupEnum groupEnum : RocketMqGroupEnum.values()) {
String groupId = groupEnum.getDefaultGroup() + "_" + profile;
//已经创建就继续
if(collect.get(groupId) != null){
continue;
}
OnsGroupCreateRequest onsGroupCreateRequest = new OnsGroupCreateRequest();
onsGroupCreateRequest.setGroupId(groupId);
onsGroupCreateRequest.setInstanceId(INSTANCE_ID);
// 复制代码运行请自行打印 API 的返回值
try {
client.onsGroupCreate(onsGroupCreateRequest);
}catch (Exception e){
log.error("initGroupAndTag fail",e);
}
Thread.sleep(1000);
}
}
}
public static void initLocalGroup(String profiles) throws Exception {
Client client = RocketMqUtil.createLocalClient("LTAI5tPWCTeCyngfYLqoSGWk", "jkzIXlvNF17ne5TPPEFP1sQhcfg4Je");
List<String> profileList = Arrays.asList(profiles.split(CommonConstants.COMMA));
if(CollectionUtils.isEmpty(profileList)){
return;
}
OnsGroupListRequest onsGroupListRequest = new OnsGroupListRequest();
onsGroupListRequest.setInstanceId(LOCAL_INSTANCE_ID);
OnsGroupListResponse onsGroupListResponse = client.onsGroupList(onsGroupListRequest);
List<OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo> subscribeInfoDo = onsGroupListResponse.getBody().getData().getSubscribeInfoDo();
Map<String, OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo> collect = ListUtils.emptyIfNull(subscribeInfoDo).stream().
collect(Collectors.toMap(OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo::getGroupId, data -> data, (a, b) -> a));
for (String profile : profileList) {
for (RocketMqGroupEnum groupEnum : RocketMqGroupEnum.values()) {
String groupId = groupEnum.getDefaultGroup() + "_" + profile;
//已经创建就继续
if(collect.get(groupId) != null){
continue;
}
OnsGroupCreateRequest onsGroupCreateRequest = new OnsGroupCreateRequest();
onsGroupCreateRequest.setGroupId(groupId);
onsGroupCreateRequest.setInstanceId(LOCAL_INSTANCE_ID);
// 复制代码运行请自行打印 API 的返回值
try {
client.onsGroupCreate(onsGroupCreateRequest);
}catch (Exception e){
log.error("initGroupAndTag fail",e);
}
Thread.sleep(1000);
}
}
}
/**
* 新增线上、灰度、预发环境的group
* @author chenyupeng
* @date 2022/1/13
* @param profiles 环境后缀,逗号分隔 例如: online,hd,pre
* @return void
*/
public static void initOnlineAndHdAndPreGroup(String profiles) throws Exception {
Client client = RocketMqUtil.createClient(ACCESS_KEY, ACCESS_SECRET);
List<String> profileList = Arrays.asList(profiles.split(CommonConstants.COMMA));
if(CollectionUtils.isEmpty(profileList)){
return;
}
OnsGroupListRequest onsGroupListRequest = new OnsGroupListRequest();
onsGroupListRequest.setInstanceId(INSTANCE_ID_ONLINE_HD_PRE);
OnsGroupListResponse onsGroupListResponse = client.onsGroupList(onsGroupListRequest);
List<OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo> subscribeInfoDo = onsGroupListResponse.getBody().getData().getSubscribeInfoDo();
Map<String, OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo> collect = ListUtils.emptyIfNull(subscribeInfoDo).stream().
collect(Collectors.toMap(OnsGroupListResponseBody.OnsGroupListResponseBodyDataSubscribeInfoDo::getGroupId, data -> data, (a, b) -> a));
for (String profile : profileList) {
for (RocketMqGroupEnum groupEnum : RocketMqGroupEnum.values()) {
String groupId = groupEnum.getDefaultGroup() + "_" + profile;
//已经创建就继续
if(collect.get(groupId) != null){
continue;
}
OnsGroupCreateRequest onsGroupCreateRequest = new OnsGroupCreateRequest();
onsGroupCreateRequest.setGroupId(groupId);
onsGroupCreateRequest.setInstanceId(INSTANCE_ID_ONLINE_HD_PRE);
// 复制代码运行请自行打印 API 的返回值
try {
client.onsGroupCreate(onsGroupCreateRequest);
}catch (Exception e){
log.error("initGroupAndTag fail",e);
}
Thread.sleep(1000);
}
}
}
/**
* 删除测试、开发环境的group
* @author chenyupeng
* @date 2022/1/13
* @param groupId
* @return void
*/
public static void deleteGroupById(String groupId) throws Exception {
Client client = RocketMqUtil.createClient(ACCESS_KEY, ACCESS_SECRET);
OnsGroupDeleteRequest request = new OnsGroupDeleteRequest();
request.setGroupId(groupId);
request.setInstanceId(INSTANCE_ID);
client.onsGroupDelete(request);
}
}

View File

@@ -0,0 +1,110 @@
package com.cool.store.service.impl;
import com.alibaba.fastjson.JSONObject;
import com.cool.store.constants.CommonConstants;
import com.cool.store.dao.EnterpriseUserDAO;
import com.cool.store.dao.EnterpriseUserRoleDAO;
import com.cool.store.dao.SysRoleDAO;
import com.cool.store.dto.enterprise.EnterpriseInitDTO;
import com.cool.store.dto.enterprise.EnterpriseOpenMsg;
import com.cool.store.dto.enterprise.EnterpriseUserDTO;
import com.cool.store.dto.enterprise.EnterpriseUserRole;
import com.cool.store.entity.EnterpriseUserDO;
import com.cool.store.enums.RoleEnum;
import com.cool.store.http.ISVHttpRequest;
import com.cool.store.mq.SimpleMessageService;
import com.cool.store.utils.DataSourceHelper;
import com.cool.store.utils.ScriptUtil;
import com.cool.store.utils.UUIDUtils;
import com.coolstore.base.enums.RocketMqTagEnum;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.support.EncodedResource;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.*;
/**
* @author zhangchenbiao
* @FileName: EnterpriseInitService
* @Description: 企业初始化
* @date 2023-05-26 16:28
*/
@Slf4j
@Service
public class EnterpriseInitServiceImpl {
@Resource
protected ScriptUtil scriptUtil;
@Resource
private SysRoleDAO sysRoleDAO;
@Resource
private EnterpriseUserRoleDAO enterpriseUserRoleDAO;
@Resource
private EnterpriseUserDAO enterpriseUserDAO;
@Resource
private SimpleMessageService simpleMessageService;
@Resource
private ISVHttpRequest isvHttpRequest;
public void runEnterpriseScript(EnterpriseOpenMsg msg) {
runEnterpriseScriptAndInitAuthUser(msg);
//抛出开始数据同步消息
EnterpriseInitDTO enterpriseInitDTO = new EnterpriseInitDTO();
enterpriseInitDTO.setEid(msg.getEid());
enterpriseInitDTO.setAppType(msg.getAppType());
enterpriseInitDTO.setCorpId(msg.getCorpId());
enterpriseInitDTO.setDbName(msg.getDbName());
enterpriseInitDTO.setUserId(msg.getAuthUserId());
log.info("send msg to enterprise_open_data_sync, eid:{}, appType:{}, corpId:{}, dbName:{}", msg.getEid(), msg.getAppType(), msg.getCorpId(), msg.getDbName(), msg.getAuthUserId());
simpleMessageService.send(JSONObject.toJSONString(enterpriseInitDTO), RocketMqTagEnum.ENTERPRISE_OPEN_DATA_SYNC);
}
public void runEnterpriseScriptAndInitAuthUser(EnterpriseOpenMsg msg) {
DataSourceHelper.changeToSpecificDataSource(msg.getDbName());
//执行脚本代码
ClassPathResource rc = new ClassPathResource("script/enterpriseInit.sql");
EncodedResource er = new EncodedResource(rc, "utf-8");
HashMap<String, Object> objectObjectHashMap = new HashMap<>();
objectObjectHashMap.put("enterpriseId", msg.getEid());
String userId = msg.getAuthUserId();
if(StringUtils.isBlank(userId)){
objectObjectHashMap.put("userId", CommonConstants.SYSTEM_USER_ID);
} else {
objectObjectHashMap.put("userId", userId);
}
String groupId = UUIDUtils.get32UUID();
objectObjectHashMap.put("groupId", groupId);
scriptUtil.executeSqlScript(er, objectObjectHashMap);
log.info("初始化开通用户");
if (StringUtils.isNotBlank(msg.getAuthUserId())) {
//初始化开通用户
initAuthUser(msg.getEid(), msg.getCorpId(), msg.getAppType(), msg.getAuthUserId(), userId, msg.getDbName());
}
}
public void initAuthUser(String eid, String corpId, String appType, String authUserId, String userId, String dbName) {
List<EnterpriseUserRole> userRoles = new ArrayList<>();
Long roleIdByRoleEnum = sysRoleDAO.getRoleIdByRoleEnum(eid, RoleEnum.MASTER.getRoleEnum());
Long shopOwner = sysRoleDAO.getRoleIdByRoleEnum(eid, RoleEnum.SHOPOWNER.getRoleEnum());
Long subMaster = sysRoleDAO.getRoleIdByRoleEnum(eid, RoleEnum.SUB_MASTER.getRoleEnum());
//钉钉或者企业走用户app开通不需要
EnterpriseUserDTO userDTO = null;
try {
userDTO = isvHttpRequest.getUserDetailByUserId(corpId, authUserId, appType);
} catch (Exception e) {
log.error("getUserDetailByUserId error", e);
}
EnterpriseUserDO userDO = EnterpriseUserDTO.transUserDtoToDo(userDTO);
userDO.setMainAdmin(true);
userDO.setIsAdmin(true);
enterpriseUserDAO.batchInsertOrUpdate(eid, Collections.singletonList(userDO));
userRoles.add(new EnterpriseUserRole(roleIdByRoleEnum.toString(), userId));
userRoles.add(new EnterpriseUserRole(shopOwner.toString(), userId));
userRoles.add(new EnterpriseUserRole(subMaster.toString(), userId));
enterpriseUserRoleDAO.insertBatchUserRole(eid, userRoles);
}
}