package com.dsh.guns.modular.system.controller.util;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.dsh.guns.core.base.controller.BaseController;
import com.dsh.guns.modular.system.model.User;
import org.springframework.stereotype.Controller;

import java.util.List;

/**
 * 条件查询封装工具类
 */
@Controller
public class WrapperUtil extends BaseController {
    /**
     * 封装查询用户ID组
     */
    public static Object getWrapperAndInUserId(List<User> list, QueryWrapper wrapper) {
        Integer[] ids = new Integer[]{0};
        if (ids != null && list.size() > 0) {
            ids = new Integer[list.size()];
            for (int i = 0; i < list.size(); i++) {
                ids[i] = list.get(i).getId();
            }
        }

        return wrapper.in("userId", ids);
    }

}