|
| 1 | +package com.codingapi.springboot.framework.trigger; |
| 2 | + |
| 3 | +import java.lang.reflect.ParameterizedType; |
| 4 | +import java.util.ArrayList; |
| 5 | +import java.util.Iterator; |
| 6 | +import java.util.List; |
| 7 | +import java.util.Map; |
| 8 | +import java.util.concurrent.ConcurrentHashMap; |
| 9 | + |
| 10 | +@SuppressWarnings("all") |
| 11 | +public class TriggerContext{ |
| 12 | + |
| 13 | + public static TriggerContext getInstance() { |
| 14 | + return instance; |
| 15 | + } |
| 16 | + |
| 17 | + private final static TriggerContext instance = new TriggerContext(); |
| 18 | + |
| 19 | + |
| 20 | + private final Map<Class<? extends Trigger>,List<TriggerHandler>> triggers; |
| 21 | + |
| 22 | + private TriggerContext(){ |
| 23 | + this.triggers = new ConcurrentHashMap<>(); |
| 24 | + } |
| 25 | + |
| 26 | + public void addTrigger(TriggerHandler trigger){ |
| 27 | + Class<? extends Trigger> clazz = getTriggerClass(trigger); |
| 28 | + List<TriggerHandler> triggerList = this.triggers.get(clazz); |
| 29 | + if(triggerList==null){ |
| 30 | + triggerList = new ArrayList<>(); |
| 31 | + this.triggers.put(clazz,triggerList); |
| 32 | + } |
| 33 | + triggerList.add(trigger); |
| 34 | + } |
| 35 | + |
| 36 | + |
| 37 | + private Class<? extends Trigger> getTriggerClass(TriggerHandler trigger){ |
| 38 | + ParameterizedType parameterizedType = (ParameterizedType) trigger.getClass().getGenericInterfaces()[0]; |
| 39 | + return (Class<? extends Trigger>) parameterizedType.getActualTypeArguments()[0]; |
| 40 | + } |
| 41 | + |
| 42 | + |
| 43 | + public void trigger(Trigger trigger){ |
| 44 | + Class<? extends Trigger> clazz = trigger.getClass(); |
| 45 | + Iterator<TriggerHandler> iterator = triggers.get(clazz).iterator(); |
| 46 | + while (iterator.hasNext()){ |
| 47 | + TriggerHandler handler = iterator.next(); |
| 48 | + Class<? extends Trigger> triggerClass = getTriggerClass(handler); |
| 49 | + if(triggerClass.equals(clazz)) { |
| 50 | + if (handler.preTrigger(trigger)) { |
| 51 | + handler.trigger(trigger); |
| 52 | + if (handler.remove()) { |
| 53 | + iterator.remove(); |
| 54 | + } |
| 55 | + } |
| 56 | + } |
| 57 | + } |
| 58 | + } |
| 59 | + |
| 60 | + public void clear(Class<? extends Trigger> clazz){ |
| 61 | + triggers.get(clazz).clear(); |
| 62 | + } |
| 63 | + |
| 64 | + |
| 65 | + public boolean isEmpty(Class<? extends Trigger> clazz){ |
| 66 | + return triggers.get(clazz).isEmpty(); |
| 67 | + } |
| 68 | + |
| 69 | +} |
0 commit comments