Beispiel #1
0
func tearDownTests() {
	storage.CloseInfoDB()
	os.RemoveAll(config.GetConfig().GetDataDir())
	os.RemoveAll(config.GetConfig().GetInfoDir())
	os.Mkdir(config.GetConfig().GetDataDir(), 0777)
	os.Mkdir(config.GetConfig().GetInfoDir(), 0777)
}
Beispiel #2
0
func tearDownTests() {
	os.RemoveAll(config.GetConfig().GetDataDir())
	os.RemoveAll(config.GetConfig().GetInfoDir())
	os.Mkdir(config.GetConfig().GetDataDir(), 0777)
	os.Mkdir(config.GetConfig().GetInfoDir(), 0777)
	storage.CloseInfoDB()
	sketchesManager.Destroy()
}
Beispiel #3
0
func tearDownTests() {
	utils.PanicOnError(os.RemoveAll(config.GetConfig().DataDir))
	utils.PanicOnError(os.RemoveAll(config.GetConfig().InfoDir))
	utils.PanicOnError(os.Mkdir(config.GetConfig().DataDir, 0777))
	utils.PanicOnError(os.Mkdir(config.GetConfig().InfoDir, 0777))
	utils.PanicOnError(storage.CloseInfoDB())
	sketchesManager.Destroy()
}
Beispiel #4
0
/*
Run ...
*/
func (srv *Server) Run() {
	conf := config.GetConfig()
	port := int(conf.Port)
	logger.Info.Println("Server up and running on port: " + strconv.Itoa(port))
	err := gracehttp.Serve(&http.Server{Addr: ":" + strconv.Itoa(port), Handler: srv})
	utils.PanicOnError(err)
}
Beispiel #5
0
/*
save ...
*/
func (sp *SketchProxy) save(force bool) {
	if !sp.dirty {
		return
	}
	id := fmt.Sprintf("%s.%s", sp.ID, sp.Type)

	if sp.ops%config.GetConfig().SaveThresholdOps == 0 || force {
		sp.ops++
		sp.dirty = false
		manager := storage.Manager()
		serialized, err := sp.sketch.Marshal()
		if err != nil {
			logger.Error.Println(err)
		}
		err = manager.SaveData(id, serialized, 0)
		if err != nil {
			logger.Error.Println(err)
		}
		info, _ := json.Marshal(sp.Info)
		err = manager.SaveInfo(sp.Info.ID, info)
		if err != nil {
			logger.Error.Println(err)
		}
	}
}
Beispiel #6
0
func (sp *SketchProxy) autosave() {
	for {
		time.Sleep(time.Duration(config.GetConfig().SaveThresholdSeconds) * time.Second)
		if sp.dirty {
			sp.save(true)
			sp.dirty = false
		}
	}
}
Beispiel #7
0
func main() {
	var port uint
	flag.UintVar(&port, "p", 3596, "specifies the port for Counts to run on")
	flag.Parse()

	//TODO: Add arguments for dataDir and infoDir

	err := os.Setenv("COUNTS_PORT", strconv.Itoa(int(port)))
	utils.PanicOnError(err)

	logger.Info.Println("Starting counts...")
	conf := config.GetConfig()
	logger.Info.Println("Using data dir: ", conf.DataDir)
	server, err := server.New()
	utils.PanicOnError(err)
	server.Run()
}
Beispiel #8
0
func TestSaveAndDeleteData(t *testing.T) {
	setupTests()
	defer tearDownTests()
	m := newManager()
	m.Create("phoenix")
	m.SaveData("phoenix", []byte("phoenix"), 0)
	path := filepath.Join(config.GetConfig().GetDataDir(), "phoenix")
	if _, err := os.Stat(path); err != nil {
		t.Error("Expected data in,", path, "got", err)
	}

	err := m.DeleteData("phoenix")
	if err != nil {
		t.Error("Expected no error deleting data, got", err)
	}
	if _, err := os.Stat(path); err == nil {
		t.Error("Expected no data in,", path, "got", err)
	}
}
Beispiel #9
0
func newManager() *ManagerStruct {
	conf = config.GetConfig()
	dataPath = conf.DataDir
	cacheSize := int(conf.CacheSize)
	if cacheSize == 0 {
		cacheSize = 250 // default cache size
	}
	cache, err := lru.NewWithEvict(cacheSize, func(k interface{}, v interface{}) {
		f := v.(*os.File)
		err := f.Close()
		if err != nil {
			logger.Error.Println(err)
		}
	})
	utils.PanicOnError(err)
	err = os.MkdirAll(dataPath, 0777)
	utils.PanicOnError(err)
	return &ManagerStruct{cache}
}
Beispiel #10
0
func tearDownTests() {
	if err := storage.CloseInfoDB(); err != nil {
		panic("Could not close info db")
	}
	if err := os.RemoveAll(config.GetConfig().DataDir); err != nil {
		panic(fmt.Sprintf("Could not remove data dir %s", config.GetConfig().DataDir))
	}
	if err := os.RemoveAll(config.GetConfig().InfoDir); err != nil {
		panic(fmt.Sprintf("Could not remove info dir %s", config.GetConfig().InfoDir))
	}
	if err := os.Mkdir(config.GetConfig().DataDir, 0777); err != nil {
		//panic(fmt.Sprintf("Could not remove info dir %s", config.GetConfig().InfoDir))
	}
	if err := os.Mkdir(config.GetConfig().InfoDir, 0777); err != nil {
		//panic(fmt.Sprintf("Could not remove info dir %s", config.GetConfig().InfoDir))
	}
	manager.Destroy()
}
Beispiel #11
0
func tearDownTests() {
	os.RemoveAll(config.GetConfig().GetDataDir())
	os.RemoveAll(config.GetConfig().GetInfoDir())
	os.Mkdir(config.GetConfig().GetDataDir(), 0777)
	os.Mkdir(config.GetConfig().GetInfoDir(), 0777)
}
Beispiel #12
0
/*
Run ...
*/
func (srv *Server) Run() {
	conf := config.GetConfig()
	port := int(conf.GetPort())
	logger.Info.Println("Server up and running on port: " + strconv.Itoa(port))
	http.ListenAndServe(":"+strconv.Itoa(port), srv)
}
Beispiel #13
0
func tearDownTests() {
	os.RemoveAll(config.GetConfig().DataDir)
	os.RemoveAll(config.GetConfig().InfoDir)
	os.Mkdir(config.GetConfig().DataDir, 0777)
	os.Mkdir(config.GetConfig().InfoDir, 0777)
}