mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-03 04:16:03 +02:00
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package sessions
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/hashicorp/go-set/v3"
|
|
|
|
"github.com/pomerium/pomerium/pkg/grpc/databroker"
|
|
"github.com/pomerium/pomerium/pkg/grpc/session"
|
|
"github.com/pomerium/pomerium/pkg/protoutil"
|
|
)
|
|
|
|
var sessionTypeURL = protoutil.GetTypeURL(new(session.Session))
|
|
|
|
// CurrentUsers returns a list of users active within the current UTC day
|
|
func CurrentUsers(
|
|
ctx context.Context,
|
|
client databroker.DataBrokerServiceClient,
|
|
) ([]string, error) {
|
|
records, _, _, err := databroker.InitialSync(ctx, client, &databroker.SyncLatestRequest{
|
|
Type: sessionTypeURL,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("fetching sessions: %w", err)
|
|
}
|
|
|
|
users := set.New[string](len(records))
|
|
utcNow := time.Now().UTC()
|
|
threshold := time.Date(utcNow.Year(), utcNow.Month(), utcNow.Day(), 0, 0, 0, 0, time.UTC)
|
|
|
|
for _, record := range records {
|
|
var s session.Session
|
|
err := record.GetData().UnmarshalTo(&s)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unmarshaling session: %w", err)
|
|
}
|
|
if s.UserId == "" { // session creation is in progress
|
|
continue
|
|
}
|
|
if s.AccessedAt == nil {
|
|
continue
|
|
}
|
|
if s.AccessedAt.AsTime().Before(threshold) {
|
|
continue
|
|
}
|
|
users.Insert(s.UserId)
|
|
}
|
|
|
|
return users.Slice(), nil
|
|
}
|