Skip to content

Commit

Permalink
chore: mod hub daemon to score when start (bittorrent#235)
Browse files Browse the repository at this point in the history
* chore: mod hub daemon to score when start

* chore: mod hub daemon to score when start

Co-authored-by: fish <[email protected]>
  • Loading branch information
2 people authored and daniel-tron committed Sep 1, 2022
1 parent a29cdac commit 393c00c
Show file tree
Hide file tree
Showing 2 changed files with 45 additions and 0 deletions.
25 changes: 25 additions & 0 deletions cmd/btfs/daemon.go
Original file line number Diff line number Diff line change
Expand Up @@ -479,6 +479,12 @@ func daemonFunc(req *cmds.Request, re cmds.ResponseEmitter, env cmds.Environment
return err
}

err = CheckHubDomainConfig(cfg, configRoot, chainid)
if err != nil {
fmt.Println("check report status, err: ", err)
return err
}

// init ip2location db
if err := bindata.Init(); err != nil {
// log init ip2location err
Expand Down Expand Up @@ -1347,3 +1353,22 @@ func CheckExistLastOnlineReport(cfg *config.Config, configRoot string, chainId i
}
return nil
}

// CheckExistLastOnlineReport sync conf and lastOnlineInfo
func CheckHubDomainConfig(cfg *config.Config, configRoot string, chainId int64) error {
var hubServerDomain string
if chainId == 199 {
hubServerDomain = config.DefaultServicesConfig().HubDomain
} else {
hubServerDomain = config.DefaultServicesConfigTestnet().HubDomain
}

if hubServerDomain != cfg.Services.HubDomain {
err := commands.SyncHubDomainConfig(configRoot, hubServerDomain)
if err != nil {
return err
}
}

return nil
}
20 changes: 20 additions & 0 deletions core/commands/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -742,6 +742,26 @@ func SyncConfigOnlineCfg(configRoot string, onlineServerDomain string, reportOnl
return nil
}

func SyncHubDomainConfig(configRoot string, hubServerDomain string) error {
r, err := fsrepo.Open(configRoot)
if err != nil {
return err
}
defer r.Close()

cfg, err := r.Config()
if err != nil {
return err
}
cfg.Services.HubDomain = hubServerDomain

err = r.SetConfig(cfg)
if err != nil {
return err
}
return nil
}

func SetConfigStorageHostEnable(configRoot string, enable bool) error {
r, err := fsrepo.Open(configRoot)
if err != nil {
Expand Down

0 comments on commit 393c00c

Please sign in to comment.