feat(knowledge): 知识库 API、权限与迁移

This commit is contained in:
weijuesen
2026-08-12 10:48:14 +08:00
parent 8e302349a7
commit 3afc34d0de
4 changed files with 234 additions and 0 deletions
+25
View File
@@ -27,15 +27,40 @@ func Init(cfg *config.Config) error {
&model.Threshold{}, &model.Alarm{}, &model.Camera{}, &model.VideoClip{},
&model.AuditLog{}, &model.Telemetry{},
&model.Permission{}, &model.RolePermission{},
&model.Disease{}, &model.KnowledgeArticle{},
); err != nil {
slog.Warn("自动迁移有警告(可忽略)", "err", err)
}
// 初始化权限种子数据
seedPermissions(db)
// 初始化知识库种子数据
seedKnowledge(db)
slog.Info("数据库连接成功,自动迁移完成")
return nil
}
// seedKnowledge 初始化知识库种子数据(幂等:按名称/类型+标题判重)
func seedKnowledge(db *gorm.DB) {
for _, d := range model.SeedDiseases {
var cnt int64
db.Model(&model.Disease{}).Where("name = ?", d.Name).Count(&cnt)
if cnt == 0 {
if err := db.Create(&d).Error; err != nil {
slog.Warn("写入病种种子失败", "name", d.Name, "err", err)
}
}
}
for _, a := range model.SeedKnowledgeArticles {
var cnt int64
db.Model(&model.KnowledgeArticle{}).Where("kind = ? AND title = ?", a.Kind, a.Title).Count(&cnt)
if cnt == 0 {
if err := db.Create(&a).Error; err != nil {
slog.Warn("写入知识文章种子失败", "title", a.Title, "err", err)
}
}
}
}
// seedPermissions 初始化权限和角色-权限映射种子数据(幂等)
func seedPermissions(db *gorm.DB) {
// 1. 写入权限定义(已存在则跳过)
+202
View File
@@ -0,0 +1,202 @@
package handler
import (
"net/http"
"silk-server-go/internal/middleware"
"silk-server-go/internal/model"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
)
// RegisterKnowledgeRoutes 注册知识库路由
func RegisterKnowledgeRoutes(rg *gin.RouterGroup, db *gorm.DB) {
read := middleware.RequirePermission(db, "knowledge:read")
write := middleware.RequirePermission(db, "knowledge:write")
rg.GET("/knowledge/diseases", read, listDiseases(db))
rg.GET("/knowledge/diseases/:id", read, getDisease(db))
rg.POST("/knowledge/diseases", write, createDisease(db))
rg.PATCH("/knowledge/diseases/:id", write, updateDisease(db))
rg.DELETE("/knowledge/diseases/:id", write, deleteDisease(db))
rg.GET("/knowledge/articles", read, listArticles(db))
rg.GET("/knowledge/articles/:id", read, getArticle(db))
rg.POST("/knowledge/articles", write, createArticle(db))
rg.PATCH("/knowledge/articles/:id", write, updateArticle(db))
rg.DELETE("/knowledge/articles/:id", write, deleteArticle(db))
}
// listDiseases 蚕病百科列表(支持 category/keyword/enabled 过滤)
func listDiseases(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
q := db.Model(&model.Disease{})
if cat := c.Query("category"); cat != "" {
q = q.Where("category = ?", cat)
}
if kw := c.Query("keyword"); kw != "" {
q = q.Where("name ILIKE ?", "%"+kw+"%")
}
if enabled := c.Query("enabled"); enabled != "" {
q = q.Where("enabled = ?", enabled == "true")
}
var list []model.Disease
q.Order("sort_order ASC, name ASC").Find(&list)
c.JSON(http.StatusOK, list)
}
}
// getDisease 蚕病百科详情
func getDisease(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
var d model.Disease
if db.Where("id = ?", c.Param("id")).First(&d).Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "disease not found"})
return
}
c.JSON(http.StatusOK, d)
}
}
// createDisease 新建病种
func createDisease(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
var d model.Disease
if err := c.ShouldBindJSON(&d); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
d.ID = ""
if d.Name == "" {
c.JSON(http.StatusBadRequest, gin.H{"error": "病种名称不能为空"})
return
}
if err := db.Create(&d).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "创建失败"})
return
}
c.JSON(http.StatusCreated, d)
}
}
// updateDisease 更新病种
func updateDisease(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
id := c.Param("id")
var d model.Disease
if db.Where("id = ?", id).First(&d).Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "disease not found"})
return
}
updates, err := bindUpdates(c)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
if len(updates) > 0 {
db.Model(&model.Disease{}).Where("id = ?", id).Updates(updates)
}
db.Where("id = ?", id).First(&d)
c.JSON(http.StatusOK, d)
}
}
// deleteDisease 删除病种
func deleteDisease(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
id := c.Param("id")
var d model.Disease
if db.Where("id = ?", id).First(&d).Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "disease not found"})
return
}
db.Where("id = ?", id).Delete(&model.Disease{})
c.JSON(http.StatusOK, gin.H{"id": id})
}
}
// listArticles 知识文章列表(支持 kind/enabled 过滤)
func listArticles(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
q := db.Model(&model.KnowledgeArticle{})
if kind := c.Query("kind"); kind != "" {
q = q.Where("kind = ?", kind)
}
if enabled := c.Query("enabled"); enabled != "" {
q = q.Where("enabled = ?", enabled == "true")
}
var list []model.KnowledgeArticle
q.Order("sort_order ASC, created_at DESC").Find(&list)
c.JSON(http.StatusOK, list)
}
}
// getArticle 知识文章详情
func getArticle(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
var a model.KnowledgeArticle
if db.Where("id = ?", c.Param("id")).First(&a).Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "article not found"})
return
}
c.JSON(http.StatusOK, a)
}
}
// createArticle 新建知识文章
func createArticle(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
var a model.KnowledgeArticle
if err := c.ShouldBindJSON(&a); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
a.ID = ""
if a.Kind == "" || a.Title == "" {
c.JSON(http.StatusBadRequest, gin.H{"error": "文章类型和标题不能为空"})
return
}
if err := db.Create(&a).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "创建失败"})
return
}
c.JSON(http.StatusCreated, a)
}
}
// updateArticle 更新知识文章
func updateArticle(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
id := c.Param("id")
var a model.KnowledgeArticle
if db.Where("id = ?", id).First(&a).Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "article not found"})
return
}
updates, err := bindUpdates(c)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
if len(updates) > 0 {
db.Model(&model.KnowledgeArticle{}).Where("id = ?", id).Updates(updates)
}
db.Where("id = ?", id).First(&a)
c.JSON(http.StatusOK, a)
}
}
// deleteArticle 删除知识文章
func deleteArticle(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
id := c.Param("id")
var a model.KnowledgeArticle
if db.Where("id = ?", id).First(&a).Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "article not found"})
return
}
db.Where("id = ?", id).Delete(&model.KnowledgeArticle{})
c.JSON(http.StatusOK, gin.H{"id": id})
}
}
@@ -22,6 +22,8 @@ var AllPermissions = []PermissionDef{
{"video:record", "视频录制", "启动和停止视频录制"},
{"energy:view", "能耗查看", "查看能耗数据"},
{"log:read", "日志查看", "查看控制日志"},
{"knowledge:read", "知识库查看", "查看蚕病百科和知识文章"},
{"knowledge:write", "知识库管理", "新增、编辑、删除知识库内容"},
{"user:manage", "用户管理", "管理用户、角色和权限"},
{"audit:read", "审计查看", "查看审计日志"},
}
@@ -32,19 +34,23 @@ var RolePermissionMap = map[string][]string{
"dashboard:view", "room:read", "room:write", "device:read", "device:control",
"threshold:read", "threshold:write", "alarm:read", "alarm:ack",
"video:read", "video:record", "energy:view", "log:read",
"knowledge:read", "knowledge:write",
"user:manage", "audit:read",
},
RoleOperator: {
"dashboard:view", "room:read", "room:write", "device:read", "device:control",
"threshold:read", "threshold:write", "alarm:read", "alarm:ack",
"video:read", "video:record", "energy:view", "log:read",
"knowledge:read", "knowledge:write",
},
RoleViewer: {
"dashboard:view", "room:read", "device:read",
"threshold:read", "alarm:read", "video:read", "energy:view",
"knowledge:read",
},
RoleFarmer: {
"dashboard:view", "room:read", "device:read", "device:control",
"alarm:read", "alarm:ack", "video:read", "energy:view",
"knowledge:read",
},
}