init

internal/dao/mysql.go

@@ -0,0 +1,62 @@
+package dao
+
+import (
+	"exam_registration/internal/model"
+	"fmt"
+	"github.com/spf13/viper"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"time"
+)
+
+var DB *gorm.DB
+
+func InitMySQL(config *viper.Viper) error {
+	host := config.GetString("database.host")
+	port := config.GetString("database.port")
+	user := config.GetString("database.user")
+	password := config.GetString("database.password")
+	dbname := config.GetString("database.dbname")
+	charset := config.GetString("database.charset")
+	maxIdleConns := config.GetInt("database.max_idle_conns")
+	maxOpenConns := config.GetInt("database.max_open_conns")
+	connMaxLifetime := config.GetInt("database.conn_max_lifetime")
+
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&parseTime=True&loc=Local",
+		user, password, host, port, dbname, charset)
+
+	var err error
+	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
+		Logger: logger.Default.LogMode(logger.Info),
+	})
+
+	if err != nil {
+		return fmt.Errorf("failed to connect database: %w", err)
+	}
+
+	// 配置连接池
+	sqlDB, err := DB.DB()
+	if err != nil {
+		return fmt.Errorf("failed to get sql.DB: %w", err)
+	}
+
+	sqlDB.SetMaxIdleConns(maxIdleConns)
+	sqlDB.SetMaxOpenConns(maxOpenConns)
+	sqlDB.SetConnMaxLifetime(time.Duration(connMaxLifetime) * time.Second)
+
+	// 自动迁移表结构
+	err = DB.AutoMigrate(
+		&model.User{},
+		&model.Exam{},
+		&model.ExamRegistration{},
+		&model.ExamNotice{},
+		&model.ExamScore{},
+	)
+
+	if err != nil {
+		return fmt.Errorf("failed to auto migrate: %w", err)
+	}
+
+	return nil
+}

internal/handler/exam_handler.go

@@ -0,0 +1,100 @@
+package handler
+
+import (
+	"exam_registration/internal/model"
+	"exam_registration/internal/service"
+	"exam_registration/pkg/response"
+	"github.com/gin-gonic/gin"
+	"strconv"
+)
+
+type ExamHandler struct {
+	examService *service.ExamService
+}
+
+func NewExamHandler() *ExamHandler {
+	return &ExamHandler{
+		examService: &service.ExamService{},
+	}
+}
+
+// CreateExam 创建考试
+func (h *ExamHandler) CreateExam(c *gin.Context) {
+	var exam model.Exam
+	if err := c.ShouldBindJSON(&exam); err != nil {
+		response.Error(c, response.BAD_REQUEST, "参数错误")
+		return
+	}
+
+	userID, _ := c.Get("user_id")
+	exam.CreatorID = userID.(uint64)
+
+	if err := h.examService.CreateExam(&exam); err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, exam)
+}
+
+// GetExamByID 获取考试详情
+func (h *ExamHandler) GetExamByID(c *gin.Context) {
+	id, _ := strconv.ParseUint(c.Param("id"), 10, 64)
+
+	exam, err := h.examService.GetExamByID(id)
+	if err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, exam)
+}
+
+// GetExamList 获取考试列表
+func (h *ExamHandler) GetExamList(c *gin.Context) {
+	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
+	pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "10"))
+
+	exams, total, err := h.examService.GetExamList(page, pageSize)
+	if err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, gin.H{
+		"list":     exams,
+		"total":    total,
+		"page":     page,
+		"pageSize": pageSize,
+	})
+}
+
+// UpdateExam 更新考试
+func (h *ExamHandler) UpdateExam(c *gin.Context) {
+	id, _ := strconv.ParseUint(c.Param("id"), 10, 64)
+
+	var updates map[string]interface{}
+	if err := c.ShouldBindJSON(&updates); err != nil {
+		response.Error(c, response.BAD_REQUEST, "参数错误")
+		return
+	}
+
+	if err := h.examService.UpdateExam(id, updates); err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, nil)
+}
+
+// DeleteExam 删除考试
+func (h *ExamHandler) DeleteExam(c *gin.Context) {
+	id, _ := strconv.ParseUint(c.Param("id"), 10, 64)
+
+	if err := h.examService.DeleteExam(id); err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, nil)
+}

internal/handler/notice_handler.go

@@ -0,0 +1,106 @@
+package handler
+
+import (
+	"exam_registration/internal/model"
+	"exam_registration/internal/service"
+	"exam_registration/pkg/response"
+	"github.com/gin-gonic/gin"
+	"strconv"
+)
+
+type NoticeHandler struct {
+	noticeService *service.NoticeService
+}
+
+func NewNoticeHandler() *NoticeHandler {
+	return &NoticeHandler{
+		noticeService: &service.NoticeService{},
+	}
+}
+
+// CreateNotice 创建通知
+func (h *NoticeHandler) CreateNotice(c *gin.Context) {
+	var notice model.ExamNotice
+	if err := c.ShouldBindJSON(&notice); err != nil {
+		response.Error(c, response.BAD_REQUEST, "参数错误")
+		return
+	}
+
+	userID, _ := c.Get("user_id")
+	notice.PublisherID = userID.(uint64)
+
+	if err := h.noticeService.CreateNotice(&notice); err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, notice)
+}
+
+// GetNoticeByID 获取通知详情
+func (h *NoticeHandler) GetNoticeByID(c *gin.Context) {
+	id, _ := strconv.ParseUint(c.Param("id"), 10, 64)
+
+	notice, err := h.noticeService.GetNoticeByID(id)
+	if err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, notice)
+}
+
+// GetNoticeList 获取通知列表
+func (h *NoticeHandler) GetNoticeList(c *gin.Context) {
+	examIDStr := c.Query("exam_id")
+	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
+	pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "10"))
+
+	var examID int
+	if examIDStr != "" {
+		examID, _ = strconv.Atoi(examIDStr)
+	}
+
+	notices, total, err := h.noticeService.GetNoticeList(examID, page, pageSize)
+	if err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, gin.H{
+		"list":     notices,
+		"total":    total,
+		"page":     page,
+		"pageSize": pageSize,
+	})
+}
+
+// UpdateNotice 更新通知
+func (h *NoticeHandler) UpdateNotice(c *gin.Context) {
+	id, _ := strconv.ParseUint(c.Param("id"), 10, 64)
+
+	var updates map[string]interface{}
+	if err := c.ShouldBindJSON(&updates); err != nil {
+		response.Error(c, response.BAD_REQUEST, "参数错误")
+		return
+	}
+
+	if err := h.noticeService.UpdateNotice(id, updates); err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, nil)
+}
+
+// DeleteNotice 删除通知
+func (h *NoticeHandler) DeleteNotice(c *gin.Context) {
+	id, _ := strconv.ParseUint(c.Param("id"), 10, 64)
+
+	if err := h.noticeService.DeleteNotice(id); err != nil {
+		response.Error(c, response.ERROR, err.Error())
+		return
+	}
+
+	response.Success(c, nil)
+}