126 lines
3.2 KiB
Go
126 lines
3.2 KiB
Go
// Vikunja is a to-do list application to facilitate your life.
|
|
// Copyright 2018-2020 Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package metrics
|
|
|
|
import (
|
|
"bytes"
|
|
"code.vikunja.io/api/pkg/log"
|
|
"code.vikunja.io/web"
|
|
"encoding/gob"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// SecondsUntilInactive defines the seconds until a user is considered inactive
|
|
const SecondsUntilInactive = 60
|
|
|
|
// ActiveUsersKey is the key used to store active users in redis
|
|
const ActiveUsersKey = `activeusers`
|
|
|
|
// ActiveUser defines an active user
|
|
type ActiveUser struct {
|
|
UserID int64
|
|
LastSeen time.Time
|
|
}
|
|
|
|
type activeUsersMap map[int64]*ActiveUser
|
|
|
|
// ActiveUsersMap is the type used to save active users
|
|
type ActiveUsers struct {
|
|
users activeUsersMap
|
|
mutex *sync.Mutex
|
|
}
|
|
|
|
// activeUsers holds a map with all active users
|
|
var activeUsers *ActiveUsers
|
|
|
|
func init() {
|
|
activeUsers = &ActiveUsers{
|
|
users: make(map[int64]*ActiveUser),
|
|
mutex: &sync.Mutex{},
|
|
}
|
|
|
|
promauto.NewGaugeFunc(prometheus.GaugeOpts{
|
|
Name: "vikunja_active_users",
|
|
Help: "The currently active users on this node",
|
|
}, func() float64 {
|
|
|
|
allActiveUsers, err := getActiveUsers()
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
}
|
|
if allActiveUsers == nil {
|
|
return 0
|
|
}
|
|
activeUsersCount := 0
|
|
for _, u := range allActiveUsers {
|
|
if time.Since(u.LastSeen) < SecondsUntilInactive*time.Second {
|
|
activeUsersCount++
|
|
}
|
|
}
|
|
return float64(activeUsersCount)
|
|
})
|
|
}
|
|
|
|
// SetUserActive sets a user as active and pushes it to redis
|
|
func SetUserActive(a web.Auth) (err error) {
|
|
activeUsers.mutex.Lock()
|
|
activeUsers.users[a.GetID()] = &ActiveUser{
|
|
UserID: a.GetID(),
|
|
LastSeen: time.Now(),
|
|
}
|
|
activeUsers.mutex.Unlock()
|
|
return PushActiveUsers()
|
|
}
|
|
|
|
// getActiveUsers returns the active users from redis
|
|
func getActiveUsers() (users activeUsersMap, err error) {
|
|
|
|
activeUsersR, err := r.Get(ActiveUsersKey).Bytes()
|
|
if err != nil {
|
|
if err.Error() == "redis: nil" {
|
|
return users, nil
|
|
}
|
|
return
|
|
}
|
|
|
|
var b bytes.Buffer
|
|
_, err = b.Write(activeUsersR)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
d := gob.NewDecoder(&b)
|
|
if err := d.Decode(&users); err != nil {
|
|
return nil, err
|
|
}
|
|
return
|
|
}
|
|
|
|
// PushActiveUsers pushed the content of the activeUsers map to redis
|
|
func PushActiveUsers() (err error) {
|
|
var b bytes.Buffer
|
|
e := gob.NewEncoder(&b)
|
|
activeUsers.mutex.Lock()
|
|
defer activeUsers.mutex.Unlock()
|
|
if err := e.Encode(activeUsers.users); err != nil {
|
|
return err
|
|
}
|
|
|
|
return r.Set(ActiveUsersKey, b.Bytes(), 0).Err()
|
|
}
|