package tech.aiflowy.ai.service.impl;

import com.mybatisflex.core.paginate.Page;
import com.mybatisflex.core.query.QueryWrapper;
import org.checkerframework.checker.units.qual.A;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import tech.aiflowy.ai.entity.AiSecondMenu;
import tech.aiflowy.ai.mapper.AiSecondMenuMapper;
import tech.aiflowy.ai.service.AiSecondMenuService;
import com.mybatisflex.spring.service.impl.ServiceImpl;

import java.util.Collections;
import java.util.Comparator;
import java.util.List;

/**
* @author admin
* @description 针对表【ai_second_menu(ai机器人二级菜单表)】的数据库操作Service实现
* @createDate 2025-05-28 15:26:35
*/
@Service
public class AiSecondMenuServiceImpl extends ServiceImpl<AiSecondMenuMapper, AiSecondMenu>
    implements AiSecondMenuService {
    @Autowired
    private AiSecondMenuMapper aiSecondMenuMapper;
    @Override
    public List<AiSecondMenu> findAll(QueryWrapper query) {
        List<AiSecondMenu> list = this.list(query);
        for (AiSecondMenu aiSecondMenu : list) {
            aiSecondMenu.setFirstMenuName(aiSecondMenuMapper.getFMN(aiSecondMenu.getFirstMenuId()));
        }
        return list;
    }
    public Page<AiSecondMenu> page(Page<AiSecondMenu> page, QueryWrapper query){
        Page page1 = this.<AiSecondMenu>pageAs(page, query, (Class) null);
        List<AiSecondMenu> records = page1.getRecords();
        for (AiSecondMenu record : records) {
            record.setFirstMenuName(aiSecondMenuMapper.getFMN(record.getFirstMenuId()));
        }
        records.sort(Comparator.comparing(AiSecondMenu::getFirstMenuId));
        return page1;
    }
}