package com.zy.acs.manager.manager.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zy.acs.manager.manager.entity.Agv;
import com.zy.acs.manager.manager.entity.AgvModel;
import com.zy.acs.manager.manager.mapper.AgvModelMapper;
import com.zy.acs.manager.manager.service.AgvModelService;
import com.zy.acs.manager.manager.service.AgvService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

@Service("agvModelService")
public class AgvModelServiceImpl extends ServiceImpl<AgvModelMapper, AgvModel> implements AgvModelService {

    public static final Map<Long, AgvModel> AGV_ID_MODEL_CACHE = new HashMap<>();
    public static final Map<String, AgvModel> AGV_NO_MODEL_CACHE = new HashMap<>();

    @Autowired
    private AgvService agvService;

    @PostConstruct
    public void init() {
        for (Agv agv : agvService.list()) {
            AgvModel agvModel = this.getById(agv.getAgvModel());
            AGV_ID_MODEL_CACHE.put(agv.getId(), agvModel);
            AGV_NO_MODEL_CACHE.put(agv.getUuid(), agvModel);
        }
    }

    @Override
    public AgvModel selectByType(String type) {
        return this.getOne(new LambdaQueryWrapper<AgvModel>().eq(AgvModel::getType, type));
    }

    @Override
    public AgvModel getByAgvId(Long agvId) {
        AgvModel agvModel = AGV_ID_MODEL_CACHE.get(agvId);
        if (agvModel == null) {
            Agv agv = agvService.getById(agvId);
            agvModel = this.getById(agv.getAgvModel());
            if(agvModel != null) {
                AGV_ID_MODEL_CACHE.put(agv.getId(), agvModel);
            }
        }
        return agvModel;
    }

    @Override
    public AgvModel getByAgvNo(String agvNo) {
        AgvModel agvModel = AGV_NO_MODEL_CACHE.get(agvNo);
        if (agvModel == null) {
            Agv agv = agvService.selectByUuid(agvNo);
            agvModel = this.getById(agv.getAgvModel());
            if(agvModel != null) {
                AGV_NO_MODEL_CACHE.put(agvNo, agvModel);
            }
        }
        return agvModel;
    }

}