Skip to content

Commit 0e94332

Browse files
authored
Minor cleanup for GetLimits RPC (#62)
1 parent 19e260b commit 0e94332

File tree

1 file changed

+36
-28
lines changed

1 file changed

+36
-28
lines changed

pkg/code/server/grpc/transaction/v2/limits.go

Lines changed: 36 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -44,13 +44,21 @@ func (s *transactionServer) GetLimits(ctx context.Context, req *transactionpb.Ge
4444
MaxPerTransaction: 0,
4545
MaxPerDay: 0,
4646
}
47+
zeroBuyModuleLimits[string(currency)] = &transactionpb.BuyModuleLimit{
48+
MaxPerTransaction: 0,
49+
MinPerTransaction: 0,
50+
}
51+
}
52+
for currency := range limit.MicroPaymentLimits {
4753
zeroMicroPaymentLimits[string(currency)] = &transactionpb.MicroPaymentLimit{
4854
MaxPerTransaction: 0,
4955
MinPerTransaction: 0,
5056
}
5157
}
5258
zeroSendLimits[string(currency_lib.KIN)] = &transactionpb.SendLimit{
53-
NextTransaction: 0,
59+
NextTransaction: 0,
60+
MaxPerTransaction: 0,
61+
MaxPerDay: 0,
5462
}
5563
zeroMicroPaymentLimits[string(currency_lib.KIN)] = &transactionpb.MicroPaymentLimit{
5664
MaxPerTransaction: 0,
@@ -154,7 +162,31 @@ func (s *transactionServer) GetLimits(ctx context.Context, req *transactionpb.Ge
154162
}
155163

156164
//
157-
// Part 2: Calculate deposit limits
165+
// Part 2: Calculate micro payment limits
166+
//
167+
168+
microPaymentLimits := make(map[string]*transactionpb.MicroPaymentLimit)
169+
for currency, limits := range limit.MicroPaymentLimits {
170+
microPaymentLimits[string(currency)] = &transactionpb.MicroPaymentLimit{
171+
MaxPerTransaction: float32(limits.Max),
172+
MinPerTransaction: float32(limits.Min),
173+
}
174+
}
175+
176+
//
177+
// Part 3: Calculate buy module limits
178+
//
179+
180+
buyModuleLimits := make(map[string]*transactionpb.BuyModuleLimit)
181+
for currency, limits := range limit.SendLimits {
182+
buyModuleLimits[string(currency)] = &transactionpb.BuyModuleLimit{
183+
MaxPerTransaction: float32(limits.PerTransaction),
184+
MinPerTransaction: float32(limits.PerTransaction / 10),
185+
}
186+
}
187+
188+
//
189+
// Part 4: Calculate deposit limits
158190
//
159191

160192
usdForNextDeposit := limit.MaxPerDepositUsdAmount
@@ -179,35 +211,11 @@ func (s *transactionServer) GetLimits(ctx context.Context, req *transactionpb.Ge
179211
usdForNextDeposit = 0
180212
}
181213

182-
//
183-
// Part 3: Calculate micro payment limits
184-
//
185-
186-
convertedMicroPaymentLimits := make(map[string]*transactionpb.MicroPaymentLimit)
187-
for currency, limits := range limit.MicroPaymentLimits {
188-
convertedMicroPaymentLimits[string(currency)] = &transactionpb.MicroPaymentLimit{
189-
MaxPerTransaction: float32(limits.Max),
190-
MinPerTransaction: float32(limits.Min),
191-
}
192-
}
193-
194-
//
195-
// Part 4: Calculate buy module limits
196-
//
197-
198-
convertedBuyModuleLimits := make(map[string]*transactionpb.BuyModuleLimit)
199-
for currency, limits := range limit.SendLimits {
200-
convertedBuyModuleLimits[string(currency)] = &transactionpb.BuyModuleLimit{
201-
MaxPerTransaction: float32(limits.PerTransaction),
202-
MinPerTransaction: float32(limits.PerTransaction / 10),
203-
}
204-
}
205-
206214
return &transactionpb.GetLimitsResponse{
207215
Result: transactionpb.GetLimitsResponse_OK,
208216
SendLimitsByCurrency: sendLimits,
209-
MicroPaymentLimitsByCurrency: convertedMicroPaymentLimits,
210-
BuyModuleLimitsByCurrency: convertedBuyModuleLimits,
217+
MicroPaymentLimitsByCurrency: microPaymentLimits,
218+
BuyModuleLimitsByCurrency: buyModuleLimits,
211219
DepositLimit: &transactionpb.DepositLimit{
212220
MaxQuarks: kin.ToQuarks(uint64(usdForNextDeposit / usdRate)),
213221
},

0 commit comments

Comments
 (0)