4.0.0-beta.0

middleware/loggin/loggin.go

@@ -1,32 +0,0 @@
-package loggin
-
-import (
-	"context"
-	"time"
-
-	"github.com/WJQSERVER-STUDIO/logger"
-	"github.com/cloudwego/hertz/pkg/app"
-)
-
-var (
-	logw       = logger.Logw
-	logDump    = logger.LogDump
-	logDebug   = logger.LogDebug
-	logInfo    = logger.LogInfo
-	logWarning = logger.LogWarning
-	logError   = logger.LogError
-)
-
-// 日志中间件
-func Middleware() app.HandlerFunc {
-	return func(ctx context.Context, c *app.RequestContext) {
-		startTime := time.Now()
-
-		c.Next(ctx)
-
-		endTime := time.Now()
-		timingResults := endTime.Sub(startTime)
-
-		logInfo("%s %s %s %s %s %d %v ", c.ClientIP(), c.Method(), c.Request.Header.GetProtocol(), string(c.Path()), c.Request.Header.UserAgent(), c.Response.StatusCode(), timingResults)
-	}
-}

middleware/nocache/nocache.go

@@ -1,17 +1,15 @@
 package nocache
 
 import (
-	"context"
-
-	"github.com/cloudwego/hertz/pkg/app"
+	"github.com/infinite-iroha/touka"
 )
 
-func NoCacheMiddleware() app.HandlerFunc {
-	return func(ctx context.Context, c *app.RequestContext) {
+func NoCacheMiddleware() touka.HandlerFunc {
+	return func(c *touka.Context) {
 		// 设置禁止缓存的响应头
-		c.Response.Header.Set("Cache-Control", "no-store, no-cache, must-revalidate")
-		c.Response.Header.Set("Pragma", "no-cache")
-		c.Response.Header.Set("Expires", "0")
-		c.Next(ctx) // 继续处理请求
+		c.SetHeader("Cache-Control", "no-store, no-cache, must-revalidate")
+		c.SetHeader("Pragma", "no-cache")
+		c.SetHeader("Expires", "0")
+		c.Next() // 继续处理请求
 	}
 }