package com.dingzhuo.compute.engine.controller;

import akka.actor.ActorRef;
import akka.actor.ActorSelection;
import akka.actor.ActorSystem;
import com.dingzhuo.compute.engine.actor.indexcalc.CalculationIndexActor;
import com.dingzhuo.compute.engine.message.ExecuteType;
import com.dingzhuo.compute.engine.message.calculation.CalculateMessage;
import com.dingzhuo.compute.engine.utils.ActorUtil;
import com.dingzhuo.compute.engine.utils.SpringAkkaExtension;
import com.dingzhuo.energy.common.utils.time.TimeManager;
import com.dingzhuo.energy.data.model.domain.IndexStorage;
import com.dingzhuo.energy.data.model.service.IIndexStorageService;
import com.dingzhuo.energy.framework.web.domain.AjaxResult;
import java.util.List;
import org.joda.time.DateTime;
import org.joda.time.format.DateTimeFormat;
import org.joda.time.format.DateTimeFormatter;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/computing")
public class CalculationController {

  private final ActorSystem actorSystem;
  private final SpringAkkaExtension akkaExt;
  private final IIndexStorageService storageService;
  private ActorSelection calculateActor;

  public CalculationController(ActorSystem actorSystem, SpringAkkaExtension akkaExt,
      IIndexStorageService storageService) {
    this.actorSystem = actorSystem;
    this.akkaExt = akkaExt;
    this.storageService = storageService;
  }

  @GetMapping("/recalc")
  public AjaxResult reCalc(@RequestBody List<Recalc> recalcList) {
    calculateActor = actorSystem.actorSelection(
        ActorUtil.getActorAddress(CalculationIndexActor.ACTOR_NAME));
    DateTimeFormatter fmt = DateTimeFormat.forPattern("yyyy-MM-dd HH:mm:ss");
    recalcList.forEach(recalc -> {
      IndexStorage indexStorage = storageService.getIndexStorage(recalc.getIndexId(),
          recalc.getTimeType());
      if (indexStorage != null) {
        String actorId = ActorUtil.buildActorId(indexStorage);
        String timeCode = TimeManager.getTimeCode(
            DateTime.parse(recalc.getDataTime(), fmt).toDate(),
            recalc.getTimeType());
        calculateActor.tell(
            new CalculateMessage(actorId, timeCode, recalc.getTimeType(), ExecuteType.TIMER),
            ActorRef.noSender());
      }
    });
    return AjaxResult.success();
  }
}