-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* liveclient rename * total token strategy * typo --------- Co-authored-by: Deniz Mert Edincik <[email protected]>
- Loading branch information
1 parent
71ced1c
commit a27d741
Showing
6 changed files
with
182 additions
and
13 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
164 changes: 164 additions & 0 deletions
164
backend/main/strategies/total_token_weighted_default.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,164 @@ | ||
| package strategies | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "math" | ||
|
|
||
| "github.com/DapperCollectives/CAST/backend/main/models" | ||
| "github.com/DapperCollectives/CAST/backend/main/shared" | ||
| s "github.com/DapperCollectives/CAST/backend/main/shared" | ||
| "github.com/rs/zerolog/log" | ||
| ) | ||
|
|
||
| type TotalTokenWeightedDefault struct { | ||
| s.StrategyStruct | ||
| DB *s.Database | ||
| } | ||
|
|
||
| func (s *TotalTokenWeightedDefault) FetchBalance( | ||
| b *models.Balance, | ||
| p *models.Proposal, | ||
| ) (*models.Balance, error) { | ||
|
|
||
| var c models.Community | ||
| if err := c.GetCommunityByProposalId(s.DB, b.Proposal_id); err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| strategy, err := models.MatchStrategyByProposal(*c.Strategies, *p.Strategy) | ||
| if err != nil { | ||
| log.Error().Err(err).Msg("Unable to find strategy for contract") | ||
| return nil, err | ||
| } | ||
|
|
||
| if err := s.FetchBalanceFromSnapshot(&strategy, b); err != nil { | ||
| log.Error().Err(err).Msg("Error calling snapshot client") | ||
| return nil, err | ||
| } | ||
| if err := b.CreateBalance(s.DB); err != nil { | ||
| log.Error().Err(err).Msg("Error creating balance in the database.") | ||
| return nil, err | ||
| } | ||
|
|
||
| return b, nil | ||
| } | ||
|
|
||
| func (s *TotalTokenWeightedDefault) FetchBalanceFromSnapshot( | ||
| strategy *models.Strategy, | ||
| b *models.Balance, | ||
| ) error { | ||
|
|
||
| var ftBalance = &shared.FTBalanceResponse{} | ||
| ftBalance.NewFTBalance() | ||
|
|
||
| if *strategy.Contract.Name == "FlowToken" { | ||
| if err := s.FlowAdapter.GetAddressBalanceAtBlockHeight( | ||
| b.Addr, | ||
| b.BlockHeight, | ||
| ftBalance, | ||
| &strategy.Contract, | ||
| ); err != nil { | ||
| log.Error().Err(err).Msg("Error fetching balance from snapshot client") | ||
| return err | ||
| } | ||
| b.PrimaryAccountBalance = ftBalance.PrimaryAccountBalance | ||
| b.SecondaryAccountBalance = ftBalance.SecondaryAccountBalance | ||
| b.StakingBalance = ftBalance.StakingBalance | ||
|
|
||
| } else { | ||
| if err := s.FlowAdapter.GetAddressBalanceAtBlockHeight( | ||
| b.Addr, | ||
| b.BlockHeight, | ||
| ftBalance, | ||
| &strategy.Contract, | ||
| ); err != nil { | ||
| log.Error().Err(err).Msg("Error fetching balance.") | ||
| return err | ||
| } | ||
| b.PrimaryAccountBalance = ftBalance.Balance | ||
| b.SecondaryAccountBalance = 0 | ||
| b.StakingBalance = 0 | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
| func (s *TotalTokenWeightedDefault) TallyVotes( | ||
| votes []*models.VoteWithBalance, | ||
| r *models.ProposalResults, | ||
| p *models.Proposal, | ||
| ) (models.ProposalResults, error) { | ||
| var zero uint64 = 0 | ||
|
|
||
| for _, vote := range votes { | ||
| if *vote.StakingBalance != zero { | ||
| var allowedBalance float64 | ||
|
|
||
| totalBalance := float64(*vote.StakingBalance) + float64(*vote.PrimaryAccountBalance) + float64(*vote.SecondaryAccountBalance) | ||
|
|
||
| if p.Max_weight != nil { | ||
| allowedBalance = p.EnforceMaxWeight(totalBalance) | ||
| } else { | ||
| allowedBalance = totalBalance | ||
| } | ||
|
|
||
| r.Results[vote.Choice] += int(allowedBalance) | ||
| r.Results_float[vote.Choice] += allowedBalance * math.Pow(10, -8) | ||
| } | ||
| } | ||
|
|
||
| return *r, nil | ||
| } | ||
|
|
||
| func (s *TotalTokenWeightedDefault) GetVoteWeightForBalance( | ||
| vote *models.VoteWithBalance, | ||
| proposal *models.Proposal, | ||
| ) (float64, error) { | ||
| var weight float64 | ||
| var ERROR error = fmt.Errorf("no weight found, address: %s, strategy: %s", vote.Addr, *proposal.Strategy) | ||
|
|
||
| totalBalance := float64(*vote.StakingBalance) + float64(*vote.PrimaryAccountBalance) + float64(*vote.SecondaryAccountBalance) | ||
|
|
||
| weight = totalBalance * math.Pow(10, -8) | ||
|
|
||
| switch { | ||
| case proposal.Max_weight != nil && weight > *proposal.Max_weight: | ||
| weight = *proposal.Max_weight | ||
| return weight, nil | ||
| case proposal.Max_weight != nil && weight < *proposal.Max_weight: | ||
| return weight, nil | ||
| case weight > 0.00: | ||
| return weight, nil | ||
| case weight == 0.00: | ||
| return 0.00, nil | ||
| default: | ||
| return weight, ERROR | ||
| } | ||
| } | ||
|
|
||
| func (s *TotalTokenWeightedDefault) GetVotes( | ||
| votes []*models.VoteWithBalance, | ||
| proposal *models.Proposal, | ||
| ) ([]*models.VoteWithBalance, error) { | ||
|
|
||
| for _, vote := range votes { | ||
| weight, err := s.GetVoteWeightForBalance(vote, proposal) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| vote.Weight = &weight | ||
| } | ||
|
|
||
| return votes, nil | ||
| } | ||
|
|
||
| func (s *TotalTokenWeightedDefault) RequiresSnapshot() bool { | ||
| return true | ||
| } | ||
|
|
||
| func (s *TotalTokenWeightedDefault) InitStrategy( | ||
| f *shared.FlowAdapter, | ||
| db *shared.Database, | ||
| ) { | ||
| s.FlowAdapter = f | ||
| s.DB = db | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| ALTER TYPE strategies ADD VALUE 'total-token-weighted-default'; | ||
| COMMIT; | ||
| INSERT INTO voting_strategies (key, name, description) | ||
| VALUES ('total-token-weighted-default', 'Total Token Weighted', 'Vote weight is proportional to the number tokens total ( primary / staked / locked ).'); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| DELETE FROM voting_strategies WHERE key = 'total-token-weighted-default' |