Function Recover

pkg/agent/proxy/util/util.go:934–967

Recover recovers from a panic in any parser and logs the stack trace to Sentry. It also closes the client and destination connection.

(logger *zap.Logger, client, dest net.Conn)

// Recover recovers from a panic in any parser and logs the stack trace to Sentry.
// It also closes the client and destination connection.
func Recover(logger *zap.Logger, client, dest net.Conn) {
	if logger == nil {
		fmt.Println(Emoji + "Failed to recover from panic. Logger is nil.")
		return
	}

	sentry.Flush(2 * time.Second)
	if r := recover(); r != nil {
		logger.Error("Recovered from panic in parser, closing active connections")
		if client != nil {
			err := client.Close()
			if err != nil {
				// Use string matching as a last resort to check for the specific error
				if !strings.Contains(err.Error(), "use of closed network connection") {
					// Log other errors
					utils.LogError(logger, err, "failed to close the client connection")
				}
			}
		}

		if dest != nil {
			err := dest.Close()
			if err != nil {
				// Use string matching as a last resort to check for the specific error
				if !strings.Contains(err.Error(), "use of closed network connection") {
					// Log other errors
					utils.LogError(logger, err, "failed to close the destination connection")
				}
			}
		}
		utils.HandleRecovery(logger, r, "Recovered from panic")
		sentry.Flush(time.Second * 2)
	}
}