Protect cache blocks map with mutex

cblocks/block1.go

@@ -5,15 +5,20 @@ import (
 )
 
 func (this *CacheBlocks) GetBlock1(host, url string) (template.HTML, bool) {
+	this.mutex.Lock()
 	if mapCache, ok := this.cacheBlocks[host]; ok {
 		if data, ok := mapCache.CacheBlock1[url]; ok {
+			this.mutex.Unlock()
 			return data, ok
 		}
 	}
+	this.mutex.Unlock()
 	return template.HTML(""), false
 }
 
 func (this *CacheBlocks) SetBlock1(host, url string, data template.HTML) {
+	this.mutex.Lock()
 	this.prepare(host)
 	this.cacheBlocks[host].CacheBlock1[url] = data
+	this.mutex.Unlock()
 }

cblocks/block2.go

@@ -5,15 +5,20 @@ import (
 )
 
 func (this *CacheBlocks) GetBlock2(host, url string) (template.HTML, bool) {
+	this.mutex.Lock()
 	if mapCache, ok := this.cacheBlocks[host]; ok {
 		if data, ok := mapCache.CacheBlock2[url]; ok {
+			this.mutex.Unlock()
 			return data, ok
 		}
 	}
+	this.mutex.Unlock()
 	return template.HTML(""), false
 }
 
 func (this *CacheBlocks) SetBlock2(host, url string, data template.HTML) {
+	this.mutex.Lock()
 	this.prepare(host)
 	this.cacheBlocks[host].CacheBlock2[url] = data
+	this.mutex.Unlock()
 }

cblocks/block3.go

@@ -5,15 +5,20 @@ import (
 )
 
 func (this *CacheBlocks) GetBlock3(host, url string) (template.HTML, bool) {
+	this.mutex.Lock()
 	if mapCache, ok := this.cacheBlocks[host]; ok {
 		if data, ok := mapCache.CacheBlock3[url]; ok {
+			this.mutex.Unlock()
 			return data, ok
 		}
 	}
+	this.mutex.Unlock()
 	return template.HTML(""), false
 }
 
 func (this *CacheBlocks) SetBlock3(host, url string, data template.HTML) {
+	this.mutex.Lock()
 	this.prepare(host)
 	this.cacheBlocks[host].CacheBlock3[url] = data
+	this.mutex.Unlock()
 }

cblocks/block4.go

@@ -5,15 +5,20 @@ import (
 )
 
 func (this *CacheBlocks) GetBlock4(host, url string) (template.HTML, bool) {
+	this.mutex.Lock()
 	if mapCache, ok := this.cacheBlocks[host]; ok {
 		if data, ok := mapCache.CacheBlock4[url]; ok {
+			this.mutex.Unlock()
 			return data, ok
 		}
 	}
+	this.mutex.Unlock()
 	return template.HTML(""), false
 }
 
 func (this *CacheBlocks) SetBlock4(host, url string, data template.HTML) {
+	this.mutex.Lock()
 	this.prepare(host)
 	this.cacheBlocks[host].CacheBlock4[url] = data
+	this.mutex.Unlock()
 }

cblocks/block5.go

@@ -5,15 +5,20 @@ import (
 )
 
 func (this *CacheBlocks) GetBlock5(host, url string) (template.HTML, bool) {
+	this.mutex.Lock()
 	if mapCache, ok := this.cacheBlocks[host]; ok {
 		if data, ok := mapCache.CacheBlock5[url]; ok {
+			this.mutex.Unlock()
 			return data, ok
 		}
 	}
+	this.mutex.Unlock()
 	return template.HTML(""), false
 }
 
 func (this *CacheBlocks) SetBlock5(host, url string, data template.HTML) {
+	this.mutex.Lock()
 	this.prepare(host)
 	this.cacheBlocks[host].CacheBlock5[url] = data
+	this.mutex.Unlock()
 }

cblocks/cblocks.go

@@ -2,6 +2,7 @@ package cblocks
 
 import (
 	"html/template"
+	"sync"
 )
 
 type cache struct {
@@ -13,11 +14,13 @@ type cache struct {
 }
 
 type CacheBlocks struct {
+	mutex       *sync.Mutex
 	cacheBlocks map[string]cache
 }
 
 func New() *CacheBlocks {
 	return &CacheBlocks{
+		mutex:       &sync.Mutex{},
 		cacheBlocks: map[string]cache{},
 	}
 }
@@ -35,7 +38,9 @@ func (this *CacheBlocks) prepare(host string) {
 }
 
 func (this *CacheBlocks) Reset(host string) {
+	this.mutex.Lock()
 	if _, ok := this.cacheBlocks[host]; ok {
 		delete(this.cacheBlocks, host)
 	}
+	this.mutex.Unlock()
 }