You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
6.4 KiB
115 lines
6.4 KiB
package database
|
|
|
|
import (
|
|
"healthapi/internal/config"
|
|
"healthapi/internal/model"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"golang.org/x/crypto/bcrypt"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
// NewDB 创建数据库连接
|
|
func NewDB(cfg config.DatabaseConfig) (*gorm.DB, error) {
|
|
var dialector gorm.Dialector
|
|
|
|
switch cfg.Driver {
|
|
case "sqlite":
|
|
dialector = sqlite.Open(cfg.DataSource)
|
|
case "mysql":
|
|
dialector = mysql.Open(cfg.DataSource)
|
|
case "postgres":
|
|
dialector = postgres.Open(cfg.DataSource)
|
|
default:
|
|
dialector = sqlite.Open(cfg.DataSource)
|
|
}
|
|
|
|
db, err := gorm.Open(dialector, &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Info),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 自动迁移
|
|
if err := model.AutoMigrate(db); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
logx.Info("Database connected successfully")
|
|
return db, nil
|
|
}
|
|
|
|
// SeedQuestionBank 初始化问卷题库
|
|
func SeedQuestionBank(db *gorm.DB) error {
|
|
var count int64
|
|
db.Model(&model.QuestionBank{}).Count(&count)
|
|
if count > 0 {
|
|
return nil // 已有数据，跳过
|
|
}
|
|
|
|
questions := []model.QuestionBank{
|
|
// 平和质问题
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您精力充沛吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您容易疲乏吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionPinghe, QuestionText: "您说话声音低弱无力吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 气虚质问题
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您容易气短，呼吸短促吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您容易心慌吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionQixu, QuestionText: "您容易头晕或站起时晕眩吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 阳虚质问题
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您手脚发凉吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您胃脘部、背部或腰膝部怕冷吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionYangxu, QuestionText: "您比一般人耐受不了寒冷吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 阴虚质问题
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您感到手脚心发热吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您感觉身体、脸上发热吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionYinxu, QuestionText: "您口唇干吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 痰湿质问题
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您感到胸闷或腹部胀满吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您感到身体沉重不轻松吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionTanshi, QuestionText: "您腹部肥满松软吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 湿热质问题
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您面部或鼻部有油腻感或者油亮发光吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您容易生痤疮或疮疖吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionShire, QuestionText: "您感到口苦或嘴里有异味吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 血瘀质问题
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您皮肤常在不知不觉中出现青紫瘀斑吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您两颧部有细微红丝吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionXueyu, QuestionText: "您身体上有哪里疼痛吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 气郁质问题
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您感到闷闷不乐、情绪低沉吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您容易精神紧张、焦虑不安吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionQiyu, QuestionText: "您多愁善感、容易感到害怕吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
// 特禀质问题
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您没有感冒时也会打喷嚏吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 1},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您没有感冒时也会鼻塞、流鼻涕吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 2},
|
|
{ConstitutionType: model.ConstitutionTebing, QuestionText: "您对某些药物、食物、气味或花粉过敏吗？", Options: `["没有","很少","有时","经常","总是"]`, OrderNum: 3},
|
|
}
|
|
|
|
return db.Create(&questions).Error
|
|
}
|
|
|
|
// SeedTestUser 创建测试用户
|
|
func SeedTestUser(db *gorm.DB) error {
|
|
var count int64
|
|
db.Model(&model.User{}).Count(&count)
|
|
if count > 0 {
|
|
return nil
|
|
}
|
|
|
|
// 测试用户，密码: 123456
|
|
passwordHash, _ := bcrypt.GenerateFromPassword([]byte("123456"), bcrypt.DefaultCost)
|
|
testUser := model.User{
|
|
Phone: "13800138000",
|
|
Email: "test@example.com",
|
|
PasswordHash: string(passwordHash),
|
|
Nickname: "测试用户",
|
|
}
|
|
|
|
return db.Create(&testUser).Error
|
|
}
|
|
|