ArticleGenerationTaskServiceImpl.java 2.1 KB
package com.aigeo.article.service.impl;

import com.aigeo.article.entity.ArticleGenerationTask;
import com.aigeo.article.repository.ArticleGenerationTaskRepository;
import com.aigeo.article.service.ArticleGenerationTaskService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Optional;

/**
 * 文章生成任务服务实现类
 */
@Service
public class ArticleGenerationTaskServiceImpl implements ArticleGenerationTaskService {
    
    @Autowired
    private ArticleGenerationTaskRepository articleGenerationTaskRepository;
    
    @Override
    public ArticleGenerationTask save(ArticleGenerationTask task) {
        return articleGenerationTaskRepository.save(task);
    }
    
    @Override
    public Optional<ArticleGenerationTask> findById(Integer id) {
        return articleGenerationTaskRepository.findById(id);
    }
    
    @Override
    public List<ArticleGenerationTask> findByCompanyId(Integer companyId) {
        return articleGenerationTaskRepository.findByCompanyId(companyId);
    }
    
    @Override
    public List<ArticleGenerationTask> findByUserId(Integer userId) {
        return articleGenerationTaskRepository.findByUserId(userId);
    }
    
    @Override
    public List<ArticleGenerationTask> findByStatus(ArticleGenerationTask.TaskStatus status) {
        return articleGenerationTaskRepository.findByStatus(status);
    }
    
    @Override
    public List<ArticleGenerationTask> findByCompanyIdAndStatus(Integer companyId, ArticleGenerationTask.TaskStatus status) {
        return articleGenerationTaskRepository.findByCompanyIdAndStatus(companyId, status);
    }
    
    @Override
    public List<ArticleGenerationTask> findByUserIdAndStatus(Integer userId, ArticleGenerationTask.TaskStatus status) {
        return articleGenerationTaskRepository.findByUserIdAndStatus(userId, status);
    }
    
    @Override
    public List<ArticleGenerationTask> findAll() {
        return articleGenerationTaskRepository.findAll();
    }
    
    @Override
    public void deleteById(Integer id) {
        articleGenerationTaskRepository.deleteById(id);
    }
}