Split out sendLogFallback

Author: ChristopherHX

protocol/logger/job_logger.go

@@ -193,6 +193,20 @@ func (logger *WebsocketLiveloggerWithFallback) Close() error {
 	return logger.replace(&errorLogger{})
 }
 
+func (logger *WebsocketLiveloggerWithFallback) sendLogFallback(err error, wrapper *protocol.TimelineRecordFeedLinesWrapper) {
+	if !logger.ForceWebsock {
+		if logger.Connection.Trace {
+			fmt.Printf("Failed to reconnect to websocket %s, fallback to vsslogger\n", err.Error())
+		}
+		currentLogger = logger.initializeVssLogger()
+		if currentLogger == nil {
+			return fmt.Errorf("failed to initialize VSS logger after websocket reconnect failure: %w", err)
+		}
+		return currentLogger.SendLog(wrapper)
+	}
+	return err
+}
+
 func (logger *WebsocketLiveloggerWithFallback) SendLog(wrapper *protocol.TimelineRecordFeedLinesWrapper) error {
 	currentLogger := getPointer(logger.currentLogger.Load())
 	if currentLogger == nil {
@@ -208,31 +222,11 @@ func (logger *WebsocketLiveloggerWithFallback) SendLog(wrapper *protocol.Timelin
 		}
 		if wslogger, ok := currentLogger.(*WebsocketLivelogger); ok {
 			if err = wslogger.Connect(); err != nil {
-				if !logger.ForceWebsock {
-					if logger.Connection.Trace {
-						fmt.Printf("Failed to reconnect to websocket %s, fallback to vsslogger\n", err.Error())
-					}
-					currentLogger = logger.initializeVssLogger()
-					if currentLogger == nil {
-						return fmt.Errorf("failed to initialize VSS logger after websocket reconnect failure: %w", err)
-					}
-					return currentLogger.SendLog(wrapper)
-				}
-				return err
+				return logger.sendLogFallback(err, wrapper)
 			}
 			err = currentLogger.SendLog(wrapper)
 			if err != nil {
-				if !logger.ForceWebsock {
-					if logger.Connection.Trace {
-						fmt.Printf("Failed to send webconsole log %s, fallback to vsslogger\n", err.Error())
-					}
-					currentLogger = logger.initializeVssLogger()
-					if currentLogger == nil {
-						return fmt.Errorf("failed to initialize VSS logger after websocket send failure: %w", err)
-					}
-					return currentLogger.SendLog(wrapper)
-				}
-				return err
+				return logger.sendLogFallback(err, wrapper)
 			}
 			return nil
 		}