(超简单)SpringBoot中简单用工厂模式来实现

news2024/11/16 20:44:12

简单讲述业务需求

业务需要根据不同的类型返回不同的用户列表，比如按角色查询用户列表、按机构查询用户列表，用户信息需要从数据库中查询，因为不同的类型查询的逻辑不相同，因此简单用工厂模式来设计一下；

首先新建一个返回用户列表的接口

/**
 * 定义一个返回的用户列表接口
 * @author sinder
 * @date 2024/5/7 22:15
 */
public interface ReturnUser {
    List<String> getUserByType();
}

再分别新建返回机构用户列表和角色用户列表的实现类

这里简化一下，执行的sql操作是测试表的，返回的数据固定的；

/**
 * @author sinder
 * @date 2024/5/7 22:29
 */
@Component
public class OrgUserImpl implements ReturnUser {

    @Autowired
    private TbTestMapper tbTestMapper;


    /**
     * 返回按机构返回的用户
     * @return
     */
    @Override
    public List<String> getUserByType() {
        System.out.println("按机构返回用户列表");
        ArrayList<String> list = new ArrayList<>();
        list.add("org1");
        list.add("org2");
        list.add("org3");
        List<TbTest> tbTests = tbTestMapper.getList();
        System.out.println(tbTests);
        return list;
    }
}

/**
 * 用户工厂实现类
 * @author sinder
 * @date 2024/5/7 22:14
 */
@Component
public class RoleUserImpl implements ReturnUser {

    @Autowired
    private TbTestMapper tbTestMapper;

    /**
     * 按角色用户返回
     * @return
     */

    @Override
    public List<String> getUserByType() {
        System.out.println("按角色返回用户列表");
        ArrayList<String> list = new ArrayList<>();
        list.add("role1");
        list.add("role2");
        list.add("role3");
        List<TbTest> tbTests = tbTestMapper.getList();
        System.out.println(tbTests);
        return list;
    }
}

然后新建一个工厂类

/**
 * 返回用户工厂
 * @author sinder
 * @date 2024/5/7 22:31
 */
@Component
public class ReturnUserFactory {

    @Autowired
    private RoleUserImpl roleUser;

    @Autowired
    private OrgUserImpl orgUser;

    public ReturnUser getUserList(String module) {
        switch (UserType.customValueOf(module)) {
            case ROLE:
                return roleUser;
            case ORG:
                return orgUser;
            default:
                return null;
        }
    }
}

这里用到了枚举，并重写了valueOf()，以下：

/**
 * @author sinder
 * @date 2024/5/7 22:33
 */
public enum UserType {
    ORG("org"),
    ROLE("role");

    private final String name;
    UserType(String name) {
        this.name = name;
    }

    // 重写valueOf方法
    public static UserType customValueOf(String text) {
        for (UserType userType : UserType.values()) {
            if (userType.name.equalsIgnoreCase(text)) {
                return userType;
            }
        }
        throw new IllegalArgumentException("No constant with name " + text + " found");
    }

}

测试

@Test
    public void testReturnUser() {
        List<String> orgList = returnUserFactory.getUserList("org").getUserByType();
        List<String> roleList = returnUserFactory.getUserList("role").getUserByType();
        System.out.println(orgList);
        System.out.println(roleList);
    }