Compare commits

..

No commits in common. "821d60021ba7f6bd0b77725e5ed654d96cf5501b" and "1512dd457532225a3db48dddfc8b3f7d0bdf7d38" have entirely different histories.

2 changed files with 0 additions and 34 deletions

View File

@ -33,14 +33,6 @@ type InsertedTask struct {
Attempts int `bson:"attempts"`
}
func Count() (*int64, error) {
count, err := collection().CountDocuments(context.TODO(), bson.M{})
if err != nil {
return nil, err
}
return &count, nil
}
func Add(task InsertedTask) error {
_, err := collection().InsertOne(context.TODO(), task)
if err != nil {

26
main.go
View File

@ -9,7 +9,6 @@ import (
"os"
"queues-go/app/routers"
client "queues-go/app/storage/mongo"
tasks "queues-go/app/storage/mongo/collections"
"strconv"
"sync"
"time"
@ -59,30 +58,6 @@ func metricProxy(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(202)
}
func writeCount() {
for {
count, err := tasks.Count()
if err != nil {
log.Printf("Failed getting docs count: %s", err.Error())
} else {
loc, _ := time.LoadLocation("Europe/Moscow")
s := fmt.Sprintf(
`{"timestamp":"%s","service":"queues","environment":"%s","name":"tasks","count":%s}`,
time.Now().In(loc).Format("2006-01-02T15:04:05Z"),
os.Getenv("STAGE"),
strconv.Itoa(int(*count)),
)
data := []byte(s)
r := bytes.NewReader(data)
_, err := http.Post("http://monitoring:1237/api/v1/metrics/increment", "application/json", r)
if err != nil {
log.Printf("ERROR %s", err.Error())
}
}
time.Sleep(time.Second)
}
}
func main() {
client.Connect()
routers.MutexMap = make(map[string]*sync.Mutex)
@ -91,6 +66,5 @@ func main() {
http.HandleFunc("/api/v1/put", handlerWrapper(routers.Put))
http.HandleFunc("/api/v1/metric", metricProxy)
log.Printf("Server started")
go writeCount()
http.ListenAndServe(":1239", nil)
}