mirror of
https://github.com/Buriburizaem0n/nezha_domains.git
synced 2026-02-04 04:30:05 +00:00
generate agent_secret for old users (#1021)
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user