优化启动程序时初始化数据的代码逻辑

pkg/service/num.go

@@ -0,0 +1,31 @@
+package service
+
+import (
+	"strconv"
+
+	"next-terminal/server/model"
+	"next-terminal/server/repository"
+)
+
+type NumService struct {
+	numRepository *repository.NumRepository
+}
+
+func NewNumService(numRepository *repository.NumRepository) *NumService {
+	return &NumService{numRepository: numRepository}
+}
+
+func (r NumService) InitNums() error {
+	nums, err := r.numRepository.FindAll()
+	if err != nil {
+		return err
+	}
+	if len(nums) == 0 {
+		for i := 0; i <= 30; i++ {
+			if err := r.numRepository.Create(&model.Num{I: strconv.Itoa(i)}); err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}