clear cache on usr1

Author: lomik

carbon-clickhouse.go

@@ -8,6 +8,8 @@ import (
 	"net"
 	"net/http"
 	"os"
+	"os/signal"
+	"syscall"
 
 	"github.com/lomik/carbon-clickhouse/carbon"
 	"github.com/lomik/carbon-clickhouse/helper/RowBinary"
@@ -148,19 +150,16 @@ func main() {
 		logger.Info("app started")
 	}
 
-	// go func() {
-	// 	c := make(chan os.Signal, 1)
-	// 	signal.Notify(c, syscall.SIGHUP)
-	// 	for {
-	// 		<-c
-	// 		logrus.Info("HUP received. Reload config")
-	// 		if err := app.ReloadConfig(); err != nil {
-	// 			logrus.Errorf("Config reload failed: %s", err.Error())
-	// 		} else {
-	// 			logrus.Info("Config successfully reloaded")
-	// 		}
-	// 	}
-	// }()
+	go func() {
+		c := make(chan os.Signal, 1)
+		signal.Notify(c, syscall.SIGUSR1)
+
+		for {
+			<-c
+			logger.Info("USR1 received. Clear tree cache")
+			app.ClearTreeExistsCache()
+		}
+	}()
 
 	app.Loop()
 

carbon/app.go

@@ -223,6 +223,17 @@ func (app *App) Start() (err error) {
 	return
 }
 
+// ClearTreeExistsCache in Uploader
+func (app *App) ClearTreeExistsCache() {
+	app.Lock()
+	up := app.Uploader
+	app.Unlock()
+
+	if up != nil {
+		go up.ClearTreeExistsCache()
+	}
+}
+
 // Loop ...
 func (app *App) Loop() {
 	app.RLock()

uploader/cmap.go

@@ -47,6 +47,17 @@ func (m CMap) Count() int {
 	return count
 }
 
+func (m CMap) Clear() int {
+	count := 0
+	for i := 0; i < shardCount; i++ {
+		shard := m[i]
+		shard.Lock()
+		shard.items = make(map[string]bool)
+		shard.Unlock()
+	}
+	return count
+}
+
 // Returns shard under given key
 func (m CMap) GetShard(key string) *CMapShard {
 	// @TODO: remove type casts

uploader/uploader.go

@@ -152,6 +152,12 @@ func (u *Uploader) Stat(send func(metric string, value float64)) {
 	send("errors", float64(errors))
 
 	send("unhandled", float64(atomic.LoadUint32(&u.stat.unhandled)))
+
+	send("treeExistsCacheSize", float64(u.treeExists.Count()))
+}
+
+func (u *Uploader) ClearTreeExistsCache() {
+	u.treeExists.Clear()
 }
 
 func uploadData(chUrl string, table string, timeout time.Duration, data io.Reader) error {