fix bug

liulingling.177216

2024-08-26 349f1cfc5fa77fbc636d542df0d8050fddec48c2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package com.dingzhuo.compute.engine.actor.alarm;
 
import akka.actor.ActorSelection;
import akka.actor.ActorSystem;
import akka.actor.Cancellable;
import akka.actor.UntypedAbstractActor;
import com.dingzhuo.compute.engine.message.alarm.AlarmJudgeMessage;
import com.dingzhuo.compute.engine.message.alarm.AlarmRegisterMessage;
import com.dingzhuo.compute.engine.message.alarm.AlarmType;
import com.dingzhuo.compute.engine.message.timer.RegisterType;
import com.dingzhuo.compute.engine.utils.ActorUtil;
import com.dingzhuo.compute.engine.utils.CacheService;
import com.dingzhuo.energy.dataservice.domain.TagValue;
import com.dingzhuo.energy.dataservice.service.RealtimeDatabaseService;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import org.springframework.beans.factory.config.ConfigurableBeanFactory;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;
import scala.concurrent.duration.Duration;
import scala.concurrent.duration.FiniteDuration;
 
@Component("alarmTimerActor")
@Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE)
public class AlarmTimerActor extends UntypedAbstractActor {
 
  public static final String ACTOR_NAME = "alarmTimerActor";
  private final ActorSystem actorSystem;
  private Cancellable timer;
  private Map<AlarmType, List<String>> actorIds = new HashMap<>();
  private List<String> realtimeTags = new ArrayList<>();
  private ActorSelection realtimeAlarmActor;
  private ActorSelection periodAlarmActor;
  private final RealtimeDatabaseService realtimeDatabaseService;
  private CacheService cacheService;
 
  public AlarmTimerActor(ActorSystem actorSystem, RealtimeDatabaseService realtimeDatabaseService,
      CacheService cacheService) {
    this.actorSystem = actorSystem;
    this.realtimeDatabaseService = realtimeDatabaseService;
    this.cacheService = cacheService;
  }
 
  @Override
  public void preStart() {
    realtimeAlarmActor = getContext()
        .actorSelection(ActorUtil.getActorAddress(RealtimeAlarmActor.ACTOR_NAME));
    periodAlarmActor = getContext()
        .actorSelection(ActorUtil.getActorAddress(PeriodAlarmActor.ACTOR_NAME));
    FiniteDuration interval = Duration.create(30, TimeUnit.SECONDS);
    FiniteDuration delay = Duration.Zero();
    timer = actorSystem.scheduler()
        .scheduleAtFixedRate(delay, interval, self(), Message.TIMER, actorSystem.dispatcher(),
            self());
  }
 
  @Override
  public void postStop() throws Exception {
    super.postStop();
    if (timer != null) {
      timer.cancel();
    }
  }
 
  @Override
  public void onReceive(Object message) {
    if (message == Message.TIMER) {
      timer();
    } else if (message instanceof AlarmRegisterMessage) {
      registerAlarm((AlarmRegisterMessage) message);
    }
  }
 
  private void registerAlarm(AlarmRegisterMessage registerMessage) {
    AlarmType alarmType = registerMessage.getAlarmType();
    if (!actorIds.containsKey(alarmType)) {
      actorIds.put(alarmType, new ArrayList<>());
    }
 
    if (registerMessage.getRegisterType() == RegisterType.REGISTER) {
      actorIds.get(alarmType).add(registerMessage.getActorId());
      if (alarmType == AlarmType.LIVE) {
        realtimeTags.add(registerMessage.getAlarmItem().getIndexCode());
      }
    } else if (registerMessage.getRegisterType() == RegisterType.UNREGISTER) {
      actorIds.get(alarmType).remove(registerMessage.getActorId());
      if (alarmType == AlarmType.LIVE) {
        realtimeTags.remove(registerMessage.getAlarmItem().getIndexCode());
      }
    }
  }
 
  private void timer() {
    if (!actorIds.isEmpty()) {
      List<String> realtimeAlarmIds = actorIds.get(AlarmType.LIVE);
      List<String> periodAlarmIds = actorIds.get(AlarmType.PERIOD);
      if (!realtimeTags.isEmpty()) {
        List<TagValue> tagValues = realtimeDatabaseService.retrieve(realtimeTags);
        cacheService.cacheTagValues(tagValues);
      }
 
      if (realtimeAlarmIds != null && !realtimeAlarmIds.isEmpty()) {
        realtimeAlarmIds
            .forEach(id -> realtimeAlarmActor.tell(new AlarmJudgeMessage(id), getSender()));
      }
 
      if (periodAlarmIds != null && !periodAlarmIds.isEmpty()) {
        periodAlarmIds.forEach(id -> periodAlarmActor.tell(new AlarmJudgeMessage(id), getSender()));
      }
    }
  }
 
  private enum Message {
    /**
     * 时间触发
     */
    TIMER
  }
}