package com.ld.vps.service;

import com.ld.vps.bean.UploadTaskInfo;
import com.ld.vps.bean.UploadTaskStatus;
import com.ld.vps.dao.UploadTaskRepository;
import com.ld.vps.manager.UploadTaskManager;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service
public class UploadTaskService {
  @Autowired
  private UploadTaskRepository taskRepo;
  @Autowired
  private UploadTaskManager taskManager;

  public Long createTask(String file, String userId) {
    UploadTaskInfo task = new UploadTaskInfo();
    task.setFilePath(file);
    task.setUserId(userId);
    task.setFileSize(0L);
    task.setUploadedSize(0L);
    task.setStatus(UploadTaskStatus.WAITING);
    taskRepo.save(task);
    taskManager.addTask(task);
    return task.getId();
  }
}