修改用户返回信息代码

ad-platform-pojo/src/main/java/com/baiye/model/dto/UserDto.java

@@ -71,6 +71,8 @@ public class UserDto extends BaseDTO implements Serializable {
 
     private Long companyId;
 
+    private Integer companyType;
+
     private Integer companyStatus;
 
     private Long organizeId;

manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/impl/UserServiceImpl.java

@@ -488,7 +488,8 @@ public class UserServiceImpl implements UserService {
             convert.setIsManager(Boolean.FALSE);
             if (user.getCompanyId() != null) {
                 companyByUserId = companyService.findCompanyById(user.getCompanyId());
-                if (companyByUserId != null && companyByUserId.getStatus() != null) {
+                if (companyByUserId != null) {
+                    convert.setCompanyType(companyByUserId.getCompanyType());
                     convert.setCompanyStatus(companyByUserId.getStatus());
                 }
             }
@@ -505,6 +506,7 @@ public class UserServiceImpl implements UserService {
                 if (role.getId() == DefaultNumberConstants.EIGHT_NUMBER) {
                     convert.setIsManager(Boolean.TRUE);
                     if (companyByUserId != null) {
+
                         convert.setUserBalance(companyByUserId.getUserBalance());
                     }
                 }