32 lines
645 B
Go
32 lines
645 B
Go
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package permissions
|
|
|
|
import (
|
|
"strconv"
|
|
"time"
|
|
|
|
"connectrpc.com/connect"
|
|
)
|
|
|
|
const defaultKeepAliveInterval = 90 * time.Second
|
|
|
|
func GetKeepAliveTicker[T any](req *connect.Request[T]) (*time.Ticker, func()) {
|
|
keepAliveIntervalHeader := req.Header().Get("Keepalive-Ping-Interval")
|
|
|
|
var interval time.Duration
|
|
|
|
keepAliveIntervalInt, err := strconv.Atoi(keepAliveIntervalHeader)
|
|
if err != nil {
|
|
interval = defaultKeepAliveInterval
|
|
} else {
|
|
interval = time.Duration(keepAliveIntervalInt) * time.Second
|
|
}
|
|
|
|
ticker := time.NewTicker(interval)
|
|
|
|
return ticker, func() {
|
|
ticker.Reset(interval)
|
|
}
|
|
}
|