chore: add logging for session and redis (#614)

Author: Stream29

internal/core/io_tunnel/backwards_invocation/transaction/serverless_handler.go

@@ -89,6 +89,14 @@ func (h *ServerlessTransactionHandler) Handle(ctx *gin.Context, sessionId string
 					backwardsRequestId, _ = invokePayload["backwards_request_id"].(string)
 				}
 
+				log.ErrorContext(
+					ctx.Request.Context(),
+					"failed to get session info from cache",
+					"session_id", sessionId,
+					"backwards_request_id", backwardsRequestId,
+					"error", err,
+				)
+
 				respData := backwards_invocation.BackwardsInvocationResponseEvent{
 					BackwardsRequestId: backwardsRequestId,
 					Event:              backwards_invocation.REQUEST_EVENT_ERROR,
@@ -99,9 +107,15 @@ func (h *ServerlessTransactionHandler) Handle(ctx *gin.Context, sessionId string
 						"session_id": sessionId,
 					},
 				}
-				_, err = ctx.Writer.Write(parser.MarshalJsonBytes(respData))
-				if err != nil {
-					log.Error("failed to write response", "error", err)
+				_, writeErr := ctx.Writer.Write(parser.MarshalJsonBytes(respData))
+				if writeErr != nil {
+					log.ErrorContext(
+						ctx.Request.Context(),
+						"failed to write serverless transaction error response",
+						"session_id", sessionId,
+						"backwards_request_id", backwardsRequestId,
+						"error", writeErr,
+					)
 				}
 				_ = writer.Close()
 				return
@@ -133,7 +147,12 @@ func (h *ServerlessTransactionHandler) Handle(ctx *gin.Context, sessionId string
 		},
 		func() {},
 		func(err string) {
-			log.Warn("invoke dify failed, received errors", "error", err)
+			log.WarnContext(
+				ctx.Request.Context(),
+				"invoke dify failed, received errors",
+				"session_id", sessionId,
+				"error", err,
+			)
 		},
 		func(plugin_entities.PluginLogEvent) {}, //log
 	)

internal/core/session_manager/session.go

@@ -95,7 +95,15 @@ func NewSession(payload NewSessionPayload) *Session {
 
 	if !payload.IgnoreCache {
 		if err := cache.Store(sessionKey(s.ID), s, time.Minute*30); err != nil {
-			log.Error("set session info to cache failed", "error", err)
+			log.ErrorContext(
+				s.RequestContext(),
+				"set session info to cache failed",
+				"session_id", s.ID,
+				"cache_key", sessionKey(s.ID),
+				"cluster_id", s.ClusterID,
+				"plugin_unique_identifier", s.PluginUniqueIdentifier.String(),
+				"error", err,
+			)
 		}
 	}
 
@@ -139,12 +147,32 @@ type DeleteSessionPayload struct {
 
 func DeleteSession(payload DeleteSessionPayload) {
 	sessionLock.Lock()
+	session := sessions[payload.ID]
 	delete(sessions, payload.ID)
 	sessionLock.Unlock()
 
 	if !payload.IgnoreCache {
-		if _, err := cache.Del(sessionKey(payload.ID)); err != nil {
-			log.Error("delete session info from cache failed", "error", err)
+		cacheKey := sessionKey(payload.ID)
+		if _, err := cache.Del(cacheKey); err != nil {
+			if session != nil {
+				log.ErrorContext(
+					session.RequestContext(),
+					"delete session info from cache failed",
+					"session_id", payload.ID,
+					"cache_key", cacheKey,
+					"cluster_id", session.ClusterID,
+					"plugin_unique_identifier", session.PluginUniqueIdentifier.String(),
+					"error", err,
+				)
+			} else {
+				log.ErrorContext(
+					context.Background(),
+					"delete session info from cache failed",
+					"session_id", payload.ID,
+					"cache_key", cacheKey,
+					"error", err,
+				)
+			}
 		}
 	}
 }

internal/service/serverless_transaction.go

@@ -3,12 +3,21 @@ package service
 import (
 	"github.com/gin-gonic/gin"
 	"github.com/langgenius/dify-plugin-daemon/internal/core/io_tunnel/backwards_invocation/transaction"
+	"github.com/langgenius/dify-plugin-daemon/pkg/utils/log"
 )
 
 func HandleServerlessPluginTransaction(handler *transaction.ServerlessTransactionHandler) gin.HandlerFunc {
 	return func(c *gin.Context) {
 		// get session id from the context
 		sessionId := c.Request.Header.Get("Dify-Plugin-Session-ID")
+		if sessionId == "" {
+			log.WarnContext(
+				c.Request.Context(),
+				"missing Dify-Plugin-Session-ID header",
+				"method", c.Request.Method,
+				"path", c.Request.URL.Path,
+			)
+		}
 
 		handler.Handle(c, sessionId)
 	}