0529

admin

2025-05-29 ef4c38330371547b66bc0c5b7ebc02d13c81cb2c

 aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/service/impl/AiSecondMenuServiceImpl.java

@@ -1,10 +1,17 @@
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.List;

/**
* @author admin
* @description 针对表【ai_second_menu(ai机器人二级菜单表)】的数据库操作Service实现
@@ -13,7 +20,24 @@
@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()));
        }
        return page1;
    }
}