chore: using proper way to use RWLock (zurvan-lab#94)

kehiy · Dec 17, 2023 · debc9ee · debc9ee
1 parent ca32799
commit debc9ee
Showing 1 changed file with 27 additions and 27 deletions.
diff --git a/core/database/database.go b/core/database/database.go
@@ -13,7 +13,7 @@ type Database struct {
 	Sets   Sets
 	Config *config.Config
 
-	lk sync.RWMutex
+	sync.RWMutex
 }
 
 func Init(cfg *config.Config) *Database {
@@ -29,8 +29,8 @@ func (db *Database) SetsMap() Sets {
 
 // ! TQL Commands.
 func (db *Database) Connect(args []string) string {
-	db.lk.RLock()
-	defer db.lk.RUnlock()
+	db.RLock()
+	defer db.RUnlock()
 
 	if len(args) != 2 {
 		return INVALID
@@ -46,8 +46,8 @@ func (db *Database) Connect(args []string) string {
 }
 
 func (db *Database) AddSet(args []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	if len(args) != 1 {
 		return INVALID
@@ -59,8 +59,8 @@ func (db *Database) AddSet(args []string) string {
 }
 
 func (db *Database) AddSubSet(args []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	if len(args) != 2 {
 		return INVALID
@@ -77,8 +77,8 @@ func (db *Database) AddSubSet(args []string) string {
 }
 
 func (db *Database) PushElement(args []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	if len(args) != 4 {
 		return INVALID
@@ -108,8 +108,8 @@ func (db *Database) PushElement(args []string) string {
 }
 
 func (db *Database) DropSet(args []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	if len(args) != 1 {
 		return INVALID
@@ -128,8 +128,8 @@ func (db *Database) DropSet(args []string) string {
 }
 
 func (db *Database) DropSubSet(args []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	if len(args) != 2 {
 		return INVALID
@@ -149,17 +149,17 @@ func (db *Database) DropSubSet(args []string) string {
 }
 
 func (db *Database) CleanSets(_ []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	db.Sets = make(Sets)
 
 	return DONE
 }
 
 func (db *Database) CleanSet(args []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	if len(args) != 1 {
 		return INVALID
@@ -178,8 +178,8 @@ func (db *Database) CleanSet(args []string) string {
 }
 
 func (db *Database) CleanSubSet(args []string) string {
-	db.lk.Lock()
-	defer db.lk.Unlock()
+	db.Lock()
+	defer db.Unlock()
 
 	if len(args) != 2 {
 		return INVALID
@@ -199,8 +199,8 @@ func (db *Database) CleanSubSet(args []string) string {
 }
 
 func (db *Database) CountSets(_ []string) string {
-	db.lk.RLock()
-	defer db.lk.RUnlock()
+	db.RLock()
+	defer db.RUnlock()
 
 	i := 0
 	for range db.Sets {
@@ -211,8 +211,8 @@ func (db *Database) CountSets(_ []string) string {
 }
 
 func (db *Database) CountSubSets(args []string) string {
-	db.lk.RLock()
-	defer db.lk.RUnlock()
+	db.RLock()
+	defer db.RUnlock()
 
 	if len(args) != 1 {
 		return INVALID
@@ -232,8 +232,8 @@ func (db *Database) CountSubSets(args []string) string {
 }
 
 func (db *Database) CountElements(args []string) string {
-	db.lk.RLock()
-	defer db.lk.RUnlock()
+	db.RLock()
+	defer db.RUnlock()
 
 	if len(args) != 2 {
 		return INVALID
@@ -253,8 +253,8 @@ func (db *Database) CountElements(args []string) string {
 }
 
 func (db *Database) GetElements(args []string) string {
-	db.lk.RLock()
-	defer db.lk.RUnlock()
+	db.RLock()
+	defer db.RUnlock()
 
 	if len(args) < 2 {
 		return INVALID