This commit is contained in:
poiuty 2023-05-14 13:55:57 +03:00
parent e5a0cb01f9
commit 9df4cb21b3
3 changed files with 39 additions and 36 deletions

View file

@ -93,7 +93,7 @@ func initClickhouse() {
func socketHandler() { func socketHandler() {
var( var (
err error err error
conn net.Conn conn net.Conn
) )

View file

@ -69,11 +69,24 @@ func saveDB() {
index := make(map[string]int64) index := make(map[string]int64)
ticker := time.NewTicker(10 * time.Second) ticker := time.NewTicker(10 * time.Second)
defer ticker.Stop()
cleanCache := time.NewTicker(12 * time.Hour)
defer cleanCache.Stop()
index = map[string]int64{"hours": int64(hours), "tokens": getSumTokens(), "last": time.Now().Unix()} index = map[string]int64{"hours": int64(hours), "tokens": getSumTokens(), "last": time.Now().Unix()}
for { for {
select { select {
case <-cleanCache.C:
l := len(data)
now := time.Now().Unix()
for k, v := range data {
if now > v.Last+60*60*48 {
delete(data, k)
}
}
fmt.Println("Clean map:", l, "=>", len(data))
case <-ticker.C: case <-ticker.C:
if len(bulk) > 0 { if len(bulk) > 0 {
tx, err := Mysql.Begin() tx, err := Mysql.Begin()
@ -167,16 +180,6 @@ func saveDB() {
} }
index["last"] = now index["last"] = now
} }
if randInt(0, 10000) == 777 { // 0.001%
l := len(data)
for k, v := range data {
if now > v.Last+60*60*48 {
delete(data, k)
}
}
fmt.Println("Clean map:", l, "=>", len(data))
}
} }
} }
} }
@ -184,6 +187,7 @@ func saveDB() {
func saveLogs() { func saveLogs() {
bulk := []saveLog{} bulk := []saveLog{}
ticker := time.NewTicker(10 * time.Second) ticker := time.NewTicker(10 * time.Second)
defer ticker.Stop()
for { for {
select { select {
case <-ticker.C: case <-ticker.C:

View file

@ -4,9 +4,9 @@ import (
"bytes" "bytes"
"fmt" "fmt"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"os/exec"
"net/http" "net/http"
"net/url" "net/url"
"os/exec"
"regexp" "regexp"
"strconv" "strconv"
"strings" "strings"
@ -130,7 +130,6 @@ func getToken(room string) string {
return "cant exec py" return "cant exec py"
} }
//fmt.Println(string(stdout)) //fmt.Println(string(stdout))
re := regexp.MustCompile(`"websocketUrl":"*(.*?)\s*"`) re := regexp.MustCompile(`"websocketUrl":"*(.*?)\s*"`)