commit db69c5b59a43a4a3fc6c2002e35a0b5f0cf17a67 · veryroundbird.house/core

+31
spindle/stream.go
···

       6
       6
        
       	"fmt"

     

       7
       7
        
       	"io"

     

       8
       8
        
       	"net/http"

     

       9
       9
       +
       	"os"

     

       9
       10
        
       	"strconv"

     

       10
       11
        
       	"time"

     

       11
       12
        
       

     
···

       143
       144
        
       	isFinished := models.StatusKind(status.Status).IsFinish()

     

       144
       145
        
       

     

       145
       146
        
       	filePath := models.LogFilePath(s.cfg.Server.LogDir, wid)

     

       147
       147
       +
       

     

       148
       148
       +
       	if status.Status == models.StatusKindFailed.String() && status.Error != nil {

     

       149
       149
       +
       		if _, err := os.Stat(filePath); os.IsNotExist(err) {

     

       150
       150
       +
       			msgs := []models.LogLine{

     

       151
       151
       +
       				{

     

       152
       152
       +
       					Kind:     models.LogKindControl,

     

       153
       153
       +
       					Content:  "",

     

       154
       154
       +
       					StepId:   0,

     

       155
       155
       +
       					StepKind: models.StepKindUser,

     

       156
       156
       +
       				},

     

       157
       157
       +
       				{

     

       158
       158
       +
       					Kind:    models.LogKindData,

     

       159
       159
       +
       					Content: *status.Error,

     

       160
       160
       +
       				},

     

       161
       161
       +
       			}

     

       162
       162
       +
       

     

       163
       163
       +
       			for _, msg := range msgs {

     

       164
       164
       +
       				b, err := json.Marshal(msg)

     

       165
       165
       +
       				if err != nil {

     

       166
       166
       +
       					return err

     

       167
       167
       +
       				}

     

       168
       168
       +
       

     

       169
       169
       +
       				if err := conn.WriteMessage(websocket.TextMessage, b); err != nil {

     

       170
       170
       +
       					return fmt.Errorf("failed to write to websocket: %w", err)

     

       171
       171
       +
       				}

     

       172
       172
       +
       			}

     

       173
       173
       +
       

     

       174
       174
       +
       			return nil

     

       175
       175
       +
       		}

     

       176
       176
       +
       	}

     

       146
       177
        
       

     

       147
       178
        
       	config := tail.Config{

     

       148
       179
        
       		Follow:    !isFinished,