From d8e96c78f8dbd746c25e4ff52c68ee5be6c440e8 Mon Sep 17 00:00:00 2001 From: Garionion Date: Sat, 2 Apr 2022 02:11:32 +0200 Subject: [PATCH] lock region map on read access --- main.go | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/main.go b/main.go index a7e18a2..c3d07f7 100644 --- a/main.go +++ b/main.go @@ -82,12 +82,6 @@ func (r *RegionCounter) countRegion(region string) { r.RegionCount[region] = val + 1 } -func (r *RegionCounter) getRegionCount() map[string]int { - r.RLock() - defer r.RUnlock() - return r.RegionCount -} - func main() { regionCounter := &RegionCounter{ RegionCount: make(map[string]int), @@ -167,7 +161,9 @@ func main() { }) app.Get("/regions", func(c *fiber.Ctx) error { - return c.JSON(regionCounter.getRegionCount()) + regionCounter.RLock() + defer regionCounter.RUnlock() + return c.JSON(regionCounter.RegionCount) }) app.Listen(":8080")