package com.vincent.rsf.server.system.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.vincent.rsf.server.system.entity.AiModelRoute;
import com.vincent.rsf.server.system.mapper.AiModelRouteMapper;
import com.vincent.rsf.server.system.service.AiModelRouteService;
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.List;

@Service("aiModelRouteService")
public class AiModelRouteServiceImpl extends ServiceImpl<AiModelRouteMapper, AiModelRoute> implements AiModelRouteService {

    @Override
    public AiModelRoute getTenantRoute(Long tenantId, Long id) {
        if (tenantId == null || id == null) {
            return null;
        }
        return this.getOne(new LambdaQueryWrapper<AiModelRoute>()
                .eq(AiModelRoute::getTenantId, tenantId)
                .eq(AiModelRoute::getId, id)
                .last("limit 1"));
    }

    @Override
    public List<AiModelRoute> listAvailableRoutes(Long tenantId, String routeCode) {
        Date now = new Date();
        return this.list(new LambdaQueryWrapper<AiModelRoute>()
                .eq(AiModelRoute::getTenantId, tenantId)
                .eq(AiModelRoute::getRouteCode, routeCode)
                .eq(AiModelRoute::getStatus, 1)
                .and(wrapper -> wrapper.isNull(AiModelRoute::getCooldownUntil).or().le(AiModelRoute::getCooldownUntil, now))
                .orderByAsc(AiModelRoute::getPriority, AiModelRoute::getId));
    }
}