generate agent_secret for old users (#1021)

This commit is contained in:
UUBulb
2025-03-08 18:47:42 +08:00
committed by GitHub
parent 79884c781a
commit 9ee5595da7
4 changed files with 31 additions and 8 deletions

View File

@@ -1,9 +1,11 @@
package singleton
import (
"fmt"
"sync"
"github.com/nezhahq/nezha/model"
"github.com/nezhahq/nezha/pkg/utils"
"gorm.io/gorm"
)
@@ -29,6 +31,13 @@ func initUser() {
AgentSecretToUserId[Conf.AgentSecretKey] = 0
for _, u := range users {
if u.AgentSecret == "" {
u.AgentSecret = utils.MustGenerateRandomString(model.DefaultAgentSecretLength)
if err := DB.Save(&u).Error; err != nil {
panic(fmt.Errorf("update of user %d failed: %v", u.ID, err))
}
}
UserInfoMap[u.ID] = model.UserInfo{
Role: u.Role,
AgentSecret: u.AgentSecret,