修正部分API授权

api/service/license_service.go

@@ -110,6 +110,11 @@ func (s *LicenseService) SyncLicense() {
 				s.license.IsActive = false
 			} else {
 				s.license = license
+				// 保存 License 到数据库
+				err = s.db.Model(&model.Config{}).Where("name = ?", types.ConfigKeyLicense).UpdateColumn("value", utils.JsonEncode(s.license)).Error
+				if err != nil {
+					logger.Errorf("保存 License 到数据库失败: %v", err)
+				}
 			}
 
 			urls, err := s.fetchUrlWhiteList()
@@ -184,6 +189,11 @@ func (s *LicenseService) GetLicense() *types.License {
 	return s.license
 }
 
+func (s *LicenseService) SetLicense(licenseKey string) {
+	s.license.Key = licenseKey
+
+}
+
 // IsValidApiURL 判断是否合法的中转 URL
 func (s *LicenseService) IsValidApiURL(uri string) error {
 	// 获得许可授权的直接放行

api/service/migration_service.go

@@ -28,18 +28,20 @@ const (
 
 // MigrationService 配置迁移服务
 type MigrationService struct {
-	db          *gorm.DB
-	redisClient *redis.Client
-	appConfig   *types.AppConfig
-	levelDB     *store.LevelDB
+	db             *gorm.DB
+	redisClient    *redis.Client
+	appConfig      *types.AppConfig
+	levelDB        *store.LevelDB
+	licenseService *LicenseService
 }
 
-func NewMigrationService(db *gorm.DB, redisClient *redis.Client, appConfig *types.AppConfig, levelDB *store.LevelDB) *MigrationService {
+func NewMigrationService(db *gorm.DB, redisClient *redis.Client, appConfig *types.AppConfig, levelDB *store.LevelDB, licenseService *LicenseService) *MigrationService {
 	return &MigrationService{
-		db:          db,
-		redisClient: redisClient,
-		appConfig:   appConfig,
-		levelDB:     levelDB,
+		db:             db,
+		redisClient:    redisClient,
+		appConfig:      appConfig,
+		levelDB:        levelDB,
+		licenseService: licenseService,
 	}
 }
 
@@ -71,6 +73,7 @@ func (s *MigrationService) MigrateLicense() {
 		logger.Errorf("迁移 License 失败: %v", err)
 		return
 	}
+	s.licenseService.SetLicense(license.Key)
 	logger.Info("迁移 License 完成")
 	s.redisClient.Set(context.Background(), key, "1", 0)
 }