|
|
@@ -18,12 +18,10 @@ import com.cyksj.common.util.StringUtil;
|
|
|
import com.cyksj.model.dto.SubmitVideoDTO;
|
|
|
import com.yhlxj.dao.mapper.GroupsRelationMapper;
|
|
|
import com.yhlxj.dao.mapper.LumaUserApiProviderMapper;
|
|
|
+import com.yhlxj.dao.mapper.UserMirrorShopMapper;
|
|
|
import com.yhlxj.dao.mapper.midjourney.LumaUserConversationMapper;
|
|
|
import com.yhlxj.dao.mapper.midjourney.LumaUserMapper;
|
|
|
-import com.yhlxj.dao.model.entity.GroupsRelation;
|
|
|
-import com.yhlxj.dao.model.entity.LumaUser;
|
|
|
-import com.yhlxj.dao.model.entity.LumaUserApiProvider;
|
|
|
-import com.yhlxj.dao.model.entity.LumaUserConversation;
|
|
|
+import com.yhlxj.dao.model.entity.*;
|
|
|
import com.yhlxj.dao.model.response.SubmitResult;
|
|
|
import com.yhlxj.dao.model.response.VideoTask;
|
|
|
import com.yhlxj.redis.RedisService;
|
|
|
@@ -85,6 +83,8 @@ public class LumaServiceImpl implements LumaService {
|
|
|
|
|
|
private final LumaUserApiProviderMapper lumaUserApiProviderMapper;
|
|
|
|
|
|
+ private final UserMirrorShopMapper userMirrorShopMapper;
|
|
|
+
|
|
|
@Override
|
|
|
public LumaUser getUser(String userToken) {
|
|
|
if(StringUtils.isBlank(userToken)){
|
|
|
@@ -102,6 +102,10 @@ public class LumaServiceImpl implements LumaService {
|
|
|
Long relationId = lumaUser.getRelationId();
|
|
|
GroupsRelation relation = groupsRelationMapper.selectById(relationId);
|
|
|
lumaUser.setAqType(relation.getAqType());
|
|
|
+ lumaUser.setCmt(relation.getCmt());
|
|
|
+ if (lumaUser.getCmt() == 2) {
|
|
|
+ Optional.ofNullable(userMirrorShopMapper.selectById(relation.getYhsId())).ifPresent(shop -> lumaUser.setCustomId(shop.getCustomId()));
|
|
|
+ }
|
|
|
return lumaUser;
|
|
|
}
|
|
|
|