geekai/api/handler/test_handler.go

package handler

import (
	"geekai/core"
	"geekai/core/middleware"
	"geekai/service"
	"geekai/service/payment"
	"net/http"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

type TestHandler struct {
	App       *core.AppServer
	db        *gorm.DB
	snowflake *service.Snowflake
	js        *payment.EPayService
}

func NewTestHandler(app *core.AppServer, db *gorm.DB, snowflake *service.Snowflake, js *payment.EPayService) *TestHandler {
	return &TestHandler{App: app, db: db, snowflake: snowflake, js: js}
}

// RegisterRoutes 注册路由
func (h *TestHandler) RegisterRoutes() {
	group := h.App.Engine.Group("/api/test/")

	// 需要用户授权的接口
	group.Use(middleware.UserAuthMiddleware(h.App.Config.Session.SecretKey, h.App.Redis))
	{
		group.Any("sse", h.PostTest, h.SseTest)
	}
}

func (h *TestHandler) SseTest(c *gin.Context) {
	//c.Header("Body-Type", "text/event-stream")
	//c.Header("Cache-Control", "no-cache")
	//c.Header("Connection", "keep-alive")
	//
	//
	//// 模拟实时数据更新
	//for i := 0; i < 10; i++ {
	//	// 发送 SSE 数据
	//	_, err := fmt.Fprintf(c.Writer, "data: %v\n\n", data)
	//	if err != nil {
	//		return
	//	}
	//	c.Writer.Flush()            // 确保立即发送数据
	//	time.Sleep(1 * time.Second) // 每秒发送一次数据
	//}
	//c.Abort()
}

func (h *TestHandler) PostTest(c *gin.Context) {
	var data struct {
		Message string `json:"message"`
		UserId  uint   `json:"user_id"`
	}

	if err := c.ShouldBindJSON(&data); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}

	// 将参数存储在上下文中
	c.Set("data", data)
	c.Next()
}