【7.4.0】【org】获取当前登录用户的公司id

kernel-s-system/system-api/src/main/java/cn/stylefeng/roses/kernel/system/api/UserOrgServiceApi.java

@@ -24,6 +24,7 @@
  */
 package cn.stylefeng.roses.kernel.system.api;
 
+import cn.stylefeng.roses.kernel.system.api.pojo.organization.HrOrganizationDTO;
 import cn.stylefeng.roses.kernel.system.api.pojo.user.SysUserOrgDTO;
 
 import java.util.Set;
@@ -60,4 +61,13 @@ public interface UserOrgServiceApi {
      * @since 2020/12/19 22:33
      */
     Set<Long> getUserIdsByOrgIds(Set<Long> organizationIds);
+
+    /**
+     * 获取当前登录用户的公司id
+     *
+     * @author fengshuonan
+     * @since 2023/4/18 10:36
+     */
+    HrOrganizationDTO getUserCompany(Long userId);
+
 }

kernel-s-system/system-business-user/src/main/java/cn/stylefeng/roses/kernel/system/modular/user/service/impl/SysUserOrgServiceServiceImpl.java

@@ -96,6 +96,28 @@ public class SysUserOrgServiceServiceImpl extends ServiceImpl<SysUserOrgMapper,
         return userOrgs.stream().map(SysUserOrg::getUserId).collect(Collectors.toSet());
     }
 
+    @Override
+    public HrOrganizationDTO getUserCompany(Long userId) {
+
+        LoginUser loginUserNullable = LoginContext.me().getLoginUserNullable();
+        if (loginUserNullable == null) {
+            return null;
+        }
+
+        // 获取当前用户的组织机构id
+        Long currentOrgId = loginUserNullable.getOrganizationId();
+        if (currentOrgId == null) {
+            return null;
+        }
+
+        // 获取当前用户orgId的公司信息
+        HrOrganizationDTO currentCompanyInfo = organizationServiceApi.getOrgCompanyInfo(currentOrgId);
+        if (currentCompanyInfo == null) {
+            return null;
+        }
+        return currentCompanyInfo;
+    }
+
 
     @Override
     public void add(UserOrgRequest userOrgResponse) {
@@ -202,7 +224,9 @@ public class SysUserOrgServiceServiceImpl extends ServiceImpl<SysUserOrgMapper,
 
         // 获取当前用户orgId的公司信息
         HrOrganizationDTO currentCompanyInfo = organizationServiceApi.getOrgCompanyInfo(currentOrgId);
-        results.add(currentCompanyInfo);
+        if (currentCompanyInfo != null) {
+            results.add(currentCompanyInfo);
+        }
 
         // 获取当前用户绑定的组织机构列表
         UserOrgRequest userOrgResponse = new UserOrgRequest();