diff --git a/internal/database/migration.go b/internal/database/migration.go
index f8f4ec35..5c836b2e 100644
--- a/internal/database/migration.go
+++ b/internal/database/migration.go
@@ -5,6 +5,7 @@ import (
"fmt"
"os"
"strings"
+ "sync"
"github.com/Tencent/WeKnora/internal/logger"
"github.com/golang-migrate/migrate/v4"
@@ -13,6 +14,27 @@ import (
_ "github.com/golang-migrate/migrate/v4/source/file"
)
+var (
+ currentMigrationVersion uint
+ currentMigrationDirty bool
+ migrationVersionOnce sync.Once
+ migrationVersionSet bool
+)
+
+// CachedMigrationVersion returns the migration version captured at startup.
+// Returns (version, dirty, ok). ok is false if the version was never captured.
+func CachedMigrationVersion() (uint, bool, bool) {
+ return currentMigrationVersion, currentMigrationDirty, migrationVersionSet
+}
+
+func setMigrationVersion(version uint, dirty bool) {
+ migrationVersionOnce.Do(func() {
+ currentMigrationVersion = version
+ currentMigrationDirty = dirty
+ migrationVersionSet = true
+ })
+}
+
// RunMigrations executes all pending database migrations
// This should be called during application startup
func RunMigrations(dsn string) error {
@@ -142,6 +164,8 @@ func RunMigrationsWithOptions(dsn string, opts MigrationOptions) error {
return fmt.Errorf("failed to get migration version: %w", err)
}
+ setMigrationVersion(version, dirty)
+
if oldVersion != version {
logger.Infof(ctx, "Database migrated from version %d to %d", oldVersion, version)
} else {
diff --git a/internal/handler/system.go b/internal/handler/system.go
index 7725dde7..ab52d5ee 100644
--- a/internal/handler/system.go
+++ b/internal/handler/system.go
@@ -11,6 +11,7 @@ import (
"github.com/Tencent/WeKnora/internal/application/service/file"
"github.com/Tencent/WeKnora/internal/config"
+ "github.com/Tencent/WeKnora/internal/database"
"github.com/Tencent/WeKnora/internal/infrastructure/docparser"
"github.com/Tencent/WeKnora/internal/logger"
"github.com/Tencent/WeKnora/internal/types"
@@ -48,6 +49,7 @@ type GetSystemInfoResponse struct {
VectorStoreEngine string `json:"vector_store_engine,omitempty"`
GraphDatabaseEngine string `json:"graph_database_engine,omitempty"`
MinioEnabled bool `json:"minio_enabled,omitempty"`
+ DBVersion string `json:"db_version,omitempty"`
}
// 编译时注入的版本信息
@@ -82,6 +84,14 @@ func (h *SystemHandler) GetSystemInfo(c *gin.Context) {
// Get MinIO enabled status
minioEnabled := h.isMinioConfigured(c)
+ var dbVersion string
+ if ver, dirty, ok := database.CachedMigrationVersion(); ok {
+ dbVersion = fmt.Sprintf("%d", ver)
+ if dirty {
+ dbVersion += " (dirty)"
+ }
+ }
+
response := GetSystemInfoResponse{
Version: Version,
Edition: Edition,
@@ -92,6 +102,7 @@ func (h *SystemHandler) GetSystemInfo(c *gin.Context) {
VectorStoreEngine: vectorStoreEngine,
GraphDatabaseEngine: graphDatabaseEngine,
MinioEnabled: minioEnabled,
+ DBVersion: dbVersion,
}
logger.Info(ctx, "System info retrieved successfully")
diff --git a/scripts/dev.sh b/scripts/dev.sh
index ec21f62c..2b750cb1 100755
--- a/scripts/dev.sh
+++ b/scripts/dev.sh
@@ -282,7 +282,8 @@ start_app() {
log_info "未检测到 Air，使用普通模式启动"
log_warning "提示: 安装 Air 可以实现代码修改后自动重启"
log_info "安装命令: go install github.com/air-verse/air@latest"
- go run -ldflags="-X 'google.golang.org/protobuf/reflect/protoregistry.conflictPolicy=warn'" cmd/server/main.go
+ LDFLAGS="$(./scripts/get_version.sh ldflags) -X 'google.golang.org/protobuf/reflect/protoregistry.conflictPolicy=warn'"
+ go run -ldflags="$LDFLAGS" cmd/server/main.go
fi
}