optimize code for remove timeout and failed image drawing job

This commit is contained in:
RockYang
2024-04-21 21:44:28 +08:00
parent f9da18ad52
commit 47c5a0387b
6 changed files with 45 additions and 14 deletions

View File

@@ -184,12 +184,7 @@ func (s *Service) Txt2Img(task types.SdTask) error {
for {
select {
case err := <-errChan:
if err != nil { // task failed
s.db.Model(&model.SdJob{Id: uint(task.Id)}).UpdateColumns(map[string]interface{}{
"progress": -1,
"err_msg": err.Error(),
})
s.notifyQueue.RPush(NotifyMessage{UserId: task.UserId, JobId: task.Id, Message: Failed})
if err != nil {
return err
}