Method ReadByLine

agent/app/service/task.go:55–109

(req request.TaskLogReadReq)

}

func (u *TaskLogService) ReadByLine(req request.TaskLogReadReq) (*response.FileLineContent, error) {
	opts := []repo.DBOption{}
	if req.TaskID != "" {
		opts = append(opts, taskRepo.WithByID(req.TaskID))
	} else {
		opts = append(opts, repo.WithOrderRuleBy("created_at", "desc"), repo.WithByType(req.TaskType), taskRepo.WithOperate(req.TaskOperate), taskRepo.WithResourceID(req.ResourceID))
	}
	taskModel, err := taskRepo.GetFirst(opts...)
	if err != nil {
		return nil, err
	}

	file, err := os.Open(taskModel.LogFile)
	if err != nil {
		return nil, err
	}
	defer file.Close()
	stat, err := file.Stat()
	if err != nil {
		return nil, err
	}

	var (
		lines       []string
		isEndOfFile bool
		scope       string
		logFileRes  *dto.LogFileRes
	)
	if stat.Size() > files.MaxReadFileSize {
		lines, _ = files.TailFromEnd(taskModel.LogFile, req.PageSize)
		isEndOfFile = true
		scope = "tail"
	} else {
		logFileRes, err = files.ReadFileByLine(taskModel.LogFile, req.Page, req.PageSize, req.Latest)
		if err != nil {
			return nil, err
		}
		scope = "page"
		lines = logFileRes.Lines
	}

	res := &response.FileLineContent{
		End:        isEndOfFile,
		Path:       taskModel.LogFile,
		TaskStatus: taskModel.Status,
		Lines:      lines,
		Scope:      scope,
	}
	if logFileRes != nil {
		res.TotalLines = logFileRes.TotalLines
		res.Total = logFileRes.TotalPages
		res.End = logFileRes.IsEndOfFile
	}
	return res, nil
}

func (u *TaskLogService) SyncForRestart() error {
	return taskRepo.UpdateRunningTaskToFailed()