package main

import (
	"log"
	"sync"

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

// Message 消息结构体
type Message struct {
	ID   uint   `gorm:"primaryKey"`
	Text string `gorm:"notNull"`
}

// Operation 操作记录结构体
type Operation struct {
	ID     uint   `gorm:"primaryKey"`
	Action string `gorm:"notNull"`
}

var (
	queue      []Message
	queueMutex sync.Mutex
	db         *gorm.DB
)

func main() {
	// 初始化数据库
	initDB()

	// 初始化Gin框架
	router := gin.Default()

	// 提交消息的接口
	router.POST("/enqueue", enqueueHandler)

	// 读取消息的接口
	router.GET("/dequeue", dequeueHandler)

	// 运行服务器
	err := router.Run(":8080")
	if err != nil {
		log.Fatal(err)
	}
}

// 初始化数据库
func initDB() {
	var err error
	db, err = gorm.Open(sqlite.Open("queue.db"), &gorm.Config{})
	if err != nil {
		log.Fatal(err)
	}

	// 自动迁移表结构
	err = db.AutoMigrate(&Message{}, &Operation{})
	if err != nil {
		log.Fatal(err)
	}
}

// 提交消息的处理函数
func enqueueHandler(c *gin.Context) {
	text := c.PostForm("text")

	// 加锁
	queueMutex.Lock()
	defer queueMutex.Unlock()

	// 将消息添加到队列
	message := Message{
		Text: text,
	}
	queue = append(queue, message)

	// 将操作记录添加到数据库
	operation := Operation{
		Action: "enqueue",
	}
	db.Create(&operation)

	c.Status(200)
}

// 读取消息的处理函数
func dequeueHandler(c *gin.Context) {
	// 加锁
	queueMutex.Lock()
	defer queueMutex.Unlock()

	// 从队列中取出消息
	if len(queue) > 0 {
		message := queue[0]
		queue = queue[1:]

		// 将操作记录添加到数据库
		operation := Operation{
			Action: "dequeue",
		}
		db.Create(&operation)

		c.JSON(200, gin.H{
			"text": message.Text,
		})
	} else {
		c.Status(204)
	}
}