如何自动装配多个接口实现并使用 Map 和 Enum 作为密钥

Question

例如，有一项任务，在不违反Open/Closed 原则的情况下，安全地添加用于以不同方式发送消息的新实现。 输入带有一个参数，该参数包含用于发送消息的“传输”类型或消息将到达的设备。

作为输入参数，需要使用Enum，但是系统设计成当参数以字符串形式出现时不使用switch，而是按照规定的条件立即调用所需的服务。

这是我的实现。

• 测试

@TestInstance(TestInstance.Lifecycle.PER_CLASS)
class NotificationMasterTest extends MultiImplementationInterfacesAndEnumApplicationTests {

    @Autowired
    private NotificationMaster notification;

    @Test
    void send() {

        String phone = "phone";

        TypeCommunication typeCommunication =
                TypeCommunication.valueOf("phone");//Runtime Exception

        notification.send(TypeCommunication.PHONE);
    }
}

• 枚举

public enum TypeCommunication {

    PHONE("phone"),
    EMAIL("email"),
    KAFKA("kafka");

    private String value;

    TypeCommunication(String value) {
        this.value = value;
    }


    public String getType() {
        return this.value;
    }
}

• 界面

public interface Notification {

    void sendNotice();

    TypeCommunication getType();
}

• 电话实施

@Service
public class NotificationPhoneImpl implements Notification {

    private final TypeCommunication typeCommunication = TypeCommunication.PHONE;

    public NotificationPhoneImpl() {
    }

    @Override
    public void sendNotice() {
        System.out.println("Send through --- phone---");
    }

    @Override
    public TypeCommunication getType() {
        return this.typeCommunication;
    }
}

• 电子邮件实施

@Service
public class NotificationEmailImpl implements Notification {

    private final TypeCommunication typeCommunication = TypeCommunication.EMAIL;

    @Override
    public void sendNotice() {
        System.out.println("Send through --- email ---");
    }

    @Override
    public TypeCommunication getType() {
        return this.typeCommunication;
    }
}

• 掌握

package com.example.multi.implementation.interfaces.services;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

@Component
public class NotificationMaster {

    private  Map<TypeCommunication, Notification> notificationService = new HashMap<>();

    private Set<Notification> notification;

    public NotificationMaster(Set<Notification> notification) {
        this.notification = notification;
    }

    @PostConstruct
    private void init(){
        notification.stream()
                .forEach(service -> {
                    TypeCommunication type = service.getType();
                    notificationService.put(type, service);
                });
    }

    public void send(TypeCommunication typeCommunication) {
        Notification notification = notificationService.get(typeCommunication);
        notification.sendNotice();
    }
}

我不知道如何传递字符串并将其动态转换为 Enum（不使用 switch）并立即获得所需的实现。

也许有一个更灵活的解决方案，其中 Spring 已经准备好没有我的组件，所以我不使用 Postconstruct 并手动创建具有不同服务实现的 map？

Answer 1

我会说它不能“即时”完成。 因此，我将在TypeCommunication中创建一个 static Map和 static 方法getByType以通过其值检索枚举：

public enum TypeCommunication {

    PHONE("phone"),
    EMAIL("email"),
    KAFKA("kafka");

    private static final Map<String, TypeCommunication> TYPE_COMMUNICATION_MAP;

    private final String value;

    static {
        TYPE_COMMUNICATION_MAP = Arrays.stream(TypeCommunication.values())
                .collect(Collectors.toMap(TypeCommunication::getType, Function.identity(),
                        (existing, replacement) -> existing));
    }

    TypeCommunication(String value) {
        this.value = value;
    }

    public String getType() {
        return this.value;
    }

    public static TypeCommunication getByType(String type) {
        return TYPE_COMMUNICATION_MAP.get(type);
    }
}

如果您在请求正文中的 controller 中收到此值，那么您可以在getType()方法上使用@JsonValue注释立即按该值在枚举中反序列化它。

关于您的NotificationMaster ， spring 无法自动收集这样的 map ，但您可以在构造函数中收集您的Enum ，因为Collectors.toMap()不使用@PostConstruct ，并且使用 EnumMap 方法，也可以使用EnumMap方法。

@Component
public class NotificationMaster {

    private final Map<TypeCommunication, Notification> map;

    public NotificationMaster(List<Notification> notifications) { // or Set
        map = notifications.stream()
                .collect(Collectors.toMap(Notification::getType, Function.identity(),
                        (existing, replacement) -> existing,
                        () -> new EnumMap<>(TypeCommunication.class)));
    }


    public void send(TypeCommunication typeCommunication) {
        map.getOrDefault(typeCommunication, new DefaultNotificationImpl()).sendNotice();
    }

    private static class DefaultNotificationImpl implements Notification {
        @Override
        public void sendNotice() {
            // some logic, e.g. throw new UnsupportedOperationException
        }

        @Override
        public TypeCommunication getType() {
            return null;
        }
    }
}

在您的send方法中定义DefaultNotificationImpl以避免NullPointerException并且现在使用Map#getOrDefault方法而不是Map#get也很好。

您还可以在单独的配置 class 中取出创建 map 的逻辑，然后在NotificationMaster中简单地自动装配此 map：

@Configuration
public class Config {

    @Bean
    public Map<TypeCommunication, Notification> createMap(List<Notification> notifications) {
        return notifications.stream()
                .collect(Collectors.toMap(Notification::getType, Function.identity(),
                        (existing, replacement) -> existing,
                        () -> new EnumMap<>(TypeCommunication.class)));
    }
}

@Component
public class NotificationMaster {

    @Autowire
    private Map<TypeCommunication, Notification> map;

    ...
}