当前位置：首页 > news >正文

组织用户数统计实现

news 来源：原创 2025/6/27 4:27:16

# 完整的组织用户数统计实现

完整的组织用户数统计实现，包括模拟SQL查询、完整的Java代码实现以及详细解释。

## 1. 模拟SQL查询

假设我们有一个组织表(organization)和用户表(user)，以下是模拟查询SQL：

```sql
-- 获取各组织及其用户数量（包括子组织用户数）
SELECT
o.id AS org_id,
o.parent_ids AS parents, -- 存储所有父组织ID，如 "1,3,5"
COUNT(u.id) AS user_sum
FROM
organization o
LEFT JOIN
user u ON u.org_id = o.id
GROUP BY
o.id, o.parent_ids;
```

## 2. 完整Java实现

```java
import java.util.*;
import java.util.stream.Collectors;

/**
* 组织用户数统计服务实现
*/
@Service
public class OrganizationServiceImpl implements OrganizationService {

@Autowired
private OrganizationMapper baseMapper;

@Override
public OrganizationalUserCountResponse getOrganizationalUserCount(OrganizationalUserCountRequest request) {
// 1. 从数据库获取各组织的用户数信息
List<OrganizationalUserCountInfo> orgUserCountList = baseMapper.getOrganizationalUserCount();

// 2. 找出最大的父组织ID（用于初始化数组大小）
int maxParentId = findMaxParentId(orgUserCountList);

// 3. 初始化用户数统计数组
int[] userCountArray = new int[maxParentId + 1];

// 4. 统计各父组织的用户数
Set<String> parentIdSet = new HashSet<>();
for (OrganizationalUserCountInfo orgInfo : orgUserCountList) {
String[] parentIds = orgInfo.getParents().split(",");
for (String parentId : parentIds) {
if (!parentId.isEmpty()) {
parentIdSet.add(parentId);
int orgId = Integer.parseInt(parentId);
userCountArray[orgId] += orgInfo.getUserSum();
}
}
}

// 5. 转换为Map结构
Map<Integer, Integer> orgUserCountMap = parentIdSet.stream()
.collect(Collectors.toMap(
Integer::parseInt,
id -> userCountArray[Integer.parseInt(id)]
));

// 6. 构建响应对象
OrganizationalUserCountResponse response = new OrganizationalUserCountResponse();
response.setOrganizationalUserCount(orgUserCountMap);

return response;
}

/**
* 找出最大的父组织ID
*/
private int findMaxParentId(List<OrganizationalUserCountInfo> orgUserCountList) {
int max = 0;
for (OrganizationalUserCountInfo orgInfo : orgUserCountList) {
String[] parentIds = orgInfo.getParents().split(",");
for (String parentId : parentIds) {
if (!parentId.isEmpty()) {
int id = Integer.parseInt(parentId);
if (id > max) {
max = id;
}
}
}
}
return max;
}
}
```

## 3. 相关实体类定义

```java
/**
* 请求对象
*/
@Data
public class OrganizationalUserCountRequest {
// 可根据需要添加查询参数
}

/**
* 响应对象
*/
@Data
public class OrganizationalUserCountResponse {
private Map<Integer, Integer> organizationalUserCount;
}

/**
* 组织用户数信息
*/
@Data
public class OrganizationalUserCountInfo {
private Integer orgId; // 组织ID
private String parents; // 父组织ID列表，如 "1,3,5"
private Integer userSum; // 该组织下的用户数
}
```

## 4. Mapper接口定义

```java
public interface OrganizationMapper {
/**
* 查询各组织及其用户数
*/
@Select("SELECT o.id AS orgId, o.parent_ids AS parents, COUNT(u.id) AS userSum " +
"FROM organization o LEFT JOIN user u ON u.org_id = o.id " +
"GROUP BY o.id, o.parent_ids")
List<OrganizationalUserCountInfo> getOrganizationalUserCount();
}
```

## 5. 代码优化建议

1. **异常处理**：
```java
try {
int id = Integer.parseInt(parentId);
// ...
} catch (NumberFormatException e) {
log.error("无效的组织ID格式: {}", parentId);
continue;
}
```

2. **空值检查**：
```java
if (orgUserCountList == null || orgUserCountList.isEmpty()) {
return new OrganizationalUserCountResponse();
}
```

3. **性能优化**：
- 对于大规模数据，考虑分批处理
- 使用StringUtils.split()替代String.split()提高性能

4. **数据结构优化**：
```java
// 使用IntStream提高原始类型处理效率
int maxParentId = orgUserCountList.stream()
.flatMapToInt(info ->
Arrays.stream(info.getParents().split(","))
.filter(StringUtils::isNotEmpty)
.mapToInt(Integer::parseInt)
)
.max()
.orElse(0);
```

## 6. 业务逻辑说明

1. 该方法统计每个组织及其所有父组织的用户数总和
2. 输入：组织用户数查询请求（可包含过滤条件）
3. 处理过程：
- 查询数据库获取各组织的用户数
- 解析每个组织的父组织链
- 累加用户数到所有相关父组织
4. 输出：包含各组织ID及其对应用户数的Map

这个实现完整展示了如何处理组织层级关系数据并统计用户数，适用于组织架构分析等场景。

相关文章：