87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
package base
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// Session represents a PostgreSQL backend
|
|
type Session struct {
|
|
Pid int64 `json:"pid"`
|
|
User string `json:"user"`
|
|
Db string `json:"database"`
|
|
Client string `json:"client"`
|
|
State string `json:"state"`
|
|
Query string `json:"query"`
|
|
StateDuration float64 `json:"state_duration"`
|
|
ApplicationName string `json:"application_name"`
|
|
}
|
|
|
|
// NewSession instanciates a Session
|
|
func NewSession(pid int64, user string, db string, client string, state string, query string, stateDuration float64, applicationName string) *Session {
|
|
return &Session{
|
|
Pid: pid,
|
|
User: user,
|
|
Db: db,
|
|
Client: client,
|
|
State: state,
|
|
Query: query,
|
|
StateDuration: stateDuration,
|
|
ApplicationName: applicationName,
|
|
}
|
|
}
|
|
|
|
// Format returns a Session as a string by replacing placeholders with their respective value
|
|
func (s *Session) Format(format string) string {
|
|
definitions := map[string]string{
|
|
"%p": fmt.Sprintf("%d", s.Pid),
|
|
"%u": s.User,
|
|
"%d": s.Db,
|
|
"%r": s.Client,
|
|
"%s": s.State,
|
|
"%m": fmt.Sprintf("%f", s.StateDuration),
|
|
"%q": s.Query,
|
|
"%a": s.ApplicationName,
|
|
}
|
|
|
|
output := format
|
|
|
|
for placeholder, value := range definitions {
|
|
output = strings.Replace(output, placeholder, value, -1)
|
|
}
|
|
|
|
return output
|
|
}
|
|
|
|
// JSON returns a Session marshaled as a JSON string
|
|
func (s *Session) JSON() string {
|
|
b, _ := json.Marshal(s)
|
|
return string(b)
|
|
}
|
|
|
|
// IsIdle returns true when a session is doing nothing
|
|
func (s *Session) IsIdle() bool {
|
|
if s.State == "idle" || s.State == "idle in transaction" || s.State == "idle in transaction (aborted)" {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Equal returns true when two sessions share the same process id
|
|
func (s *Session) Equal(session *Session) bool {
|
|
if s.Pid == 0 {
|
|
return s.User == session.User && s.Db == session.Db && s.Client == session.Client
|
|
}
|
|
return s.Pid == session.Pid
|
|
}
|
|
|
|
// InSlice returns true when this sessions in present in the slice
|
|
func (s *Session) InSlice(sessions []*Session) bool {
|
|
for _, session := range sessions {
|
|
if s.Equal(session) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|