From 5d4680e496174cef037c1120d7ba9f94bf817910 Mon Sep 17 00:00:00 2001
From: c9s <yoan@maicoin.com>
Date: Mon, 9 Nov 2020 01:10:14 +0800
Subject: [PATCH] add lock and unlock functions

---
 pkg/types/account.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/pkg/types/account.go b/pkg/types/account.go
index 0a9feca66..83fb0ccdb 100644
--- a/pkg/types/account.go
+++ b/pkg/types/account.go
@@ -52,6 +52,28 @@ func (a *Account) Balance(currency string) (balance Balance, ok bool) {
 	return balance, ok
 }
 
+func (a *Account) UnlockBalance(currency string, unlocked float64) bool {
+	a.Lock()
+	balance, ok := a.balances[currency]
+	if ok {
+		balance.Locked -= unlocked
+		balance.Available += unlocked
+	}
+	a.Unlock()
+	return ok
+}
+
+func (a *Account) LockBalance(currency string, locked float64) bool {
+	a.Lock()
+	balance, ok := a.balances[currency]
+	if ok {
+		balance.Locked += locked
+		balance.Available -= locked
+	}
+	a.Unlock()
+	return ok
+}
+
 func (a *Account) UpdateBalances(balances map[string]Balance) {
 	a.Lock()
 	defer a.Unlock()