公司禁止JOIN查询怎么办？

Posted 2022-01-06 Monster_起飞

场景

很多公司（特别是做电商的）其实都是不允许多表关联查询的，或者严格控制关联的表数量，比如最多关联2、3张表。此时，如果某个需求又确实需要进行关联查询怎么办呢？

比如前端有个页面：

很明显，这个页面字段来自两张表：

• t_product
• t_user

正常来说，直接这样写SQL即可：

SELECT p.id, p.product_name, p.price, u.user_name, u.user_age
FROM t_product p 
LEFT JOIN t_user u ON p.user_id=u.id;

但上面说了，不能关联查询。

解决方案

作为替代方案，可以先从t_product表查出10条数据到内存（t_product作为主表），然后取出10条数据的uid，再调用UserService#listUser(uids)，得到对应的userList。此时内存中有10条product，也有10条user，匹配组合即可。

public List<ProductExtendsTO> getList(Integer page, Integer pageSize) 
    // 1.查询Product
    List<Product> productList = listProduct(page, pageSize);
    // 2.取出里面的所有uid
    List<Long> uids = productList.stream()
            .map(Product::getUid)
            .collect(Collectors.toList());
    // 3.查出userList
    List<User> userList = listUser(uids);
    // 4.把List转成Map
    Map<Long, User> userMap = new HashMap<Long, User>();
    for(userList : user)
        userMap.put(user.getId(), user);
    

    // 组合并返回数据
    List<ProductExtendsTO> result = new ArrayList<>();
    productList.foreach(product->
        ProductExtendsTO productExtends = new ProductExtendsTO();
        BeanUtils.copyProperties(product, productExtends);
        // 根据product的uid从userMap获取user（此处省略user null判断）
        User user = userMap.get(product.getUid());
        productExtends.setUserAge(user.getUserAge());
        productExtends.setUserName(user.getUserName());
        result.add(productExtends);
    );

    return result;

上面的代码可以优化为（主要第4点）：

public List<ProductExtendsTO> getList(Integer page, Integer pageSize) 
    // 1.查询Product
    List<Product> productList = listProduct(page, pageSize);
    // 2.取出里面的所有uid
    List<Long> uids = productList.stream()
            .map(Product::getUid)
            .collect(Collectors.toList());
    // 3.查出userList
    List<User> userList = listUser(uids);
    // 4.把List转成Map（优化了这里）
    Map<Long, User> userMap = userList.stream()
            .collect(Collectors.toMap(User::getId(), user->user));

    // 组合并返回数据
    List<ProductExtendsTO> result = new ArrayList<>();
    productList.foreach(product->
        ProductExtendsTO productExtends = new ProductExtendsTO();
        BeanUtils.copyProperties(product, productExtends);
        // 根据product的uid从userMap获取user（此处省略user null判断）
        User user = userMap.get(product.getUid());
        productExtends.setUserAge(user.getUserAge());
        productExtends.setUserName(user.getUserName());
        result.add(productExtends);
    )

    return result;

代码优化：封装ConvertUtil

List转Map是非常普遍的需求，Stream API其实还是有点啰嗦（代码太长了），所以我们可以试着封装一下：

public final class ConvertUtil 

    private ConvertUtil() 
    

    /**
     * 将List转为Map
     *
     * @param list         原数据
     * @param keyExtractor Key的抽取规则
     * @param <K>          Key
     * @param <V>          Value
     * @return
     */
    public static <K, V> Map<K, V> listToMap(List<V> list, 
                                             Function<V, K> keyExtractor) 
        if (list == null || list.isEmpty()) 
            return new HashMap<>();
        
        
        Map<K, V> map = new HashMap<>(list.size());
        for (V element : list) 
            // 利用keyExtractor从对象中抽取Key
            K key = keyExtractor.apply(element);
            // 这里默认key不能为null
            if (key == null) 
                continue;
            
            map.put(key, element);
        
        
        return map;
    

除了List转Map，从List中抽取特定字段的需求也是非常普遍的，比如上面代码：

// 2.取出里面的所有uid（省略null判断）
List<Long> uids = productList.stream()
                             .map(Product::getUid)
                             .collect(Collectors.toList());

意思是从productList中抽取uids。为了复用，我们也封装一下：

public class ConvertUtil 

    private ConvertUtil() 
    

    /**
     * 将List映射为List，比如List<Person> personList转为List<String> nameList
     *
     * @param originList 原数据
     * @param mapper     映射规则
     * @param <T>        原数据的元素类型
     * @param <R>        新数据的元素类型
     * @return
     */
    public static <T, R> List<R> resultToList(List<T> originList, 
                                              Function<T, R> mapper) 
        if (list == null || list.isEmpty()) 
            return new ArrayList<>();
        
        
        List<R> newList = new ArrayList<>(originList.size());
        for (T originElement : originList) 
            R newElement = mapper.apply(originElement);
            if (newElement == null) 
                continue;
            
            newList.add(newElement);
        
        
        return newList;
    
    
    /**
     * 将List转为Map
     *
     * @param list         原数据
     * @param keyExtractor Key的抽取规则
     * @param <K>          Key
     * @param <V>          Value
     * @return
     */
    public static <K, V> Map<K, V> listToMap(List<V> list, 
                                             Function<V, K> keyExtractor) 
        if (list == null || list.isEmpty()) 
            return new HashMap<>();
        
        
        Map<K, V> map = new HashMap<>(list.size());
        for (V element : list) 
            K key = keyExtractor.apply(element);
            if (key == null) 
                continue;
            
            map.put(key, element);
        

        return map;
    

上面权当抛砖引玉，大家可以基于实际需求自行扩展ConvertUtil，让它更好用。

总结：

• List转Map，重点是传入Map中Key的抽取规则，也就是KeyExtractor，用了函数式接口
• List抽取FieldList，重点也是定义字段的抽取规则，也用了函数式接口

其他解决策略

有时遇到复杂的统计报表等数据，很难通过上面“内存关联”的方式完成需求，此时可以让公司的大数据部门提供接口，直接从大数据那边获取数据。但这个并不需要我们操心：小公司适当关联查询无伤大雅，大公司一般都有大数据部门。