jriou/pgterminate
1
0
Fork 0
Code Issues Pull requests Activity

Add log-format option

Browse source
This commit is contained in:
Julien Riou 2019-02-16 11:47:30 +01:00
commit 24eb4fe203
No known key found for this signature in database
GPG key ID: BA3E15176E45E85D
9 changed files with 61 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

40
base/session.go
View file

@ -29,31 +29,25 @@ func NewSession(pid int64, user string, db string, client string, state string,
}
}
// String represents a Session as a string
func (s *Session) String() string {
var output []string
if s.Pid != 0 {
output = append(output, fmt.Sprintf("pid=%d", s.Pid))
// 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,
}
if s.User != "" {
output = append(output, fmt.Sprintf("user=%s", s.User))
output := format
for placeholder, value := range definitions {
output = strings.Replace(output, placeholder, value, -1)
}
if s.Db != "" {
output = append(output, fmt.Sprintf("db=%s", s.Db))
}
if s.Client != "" {
output = append(output, fmt.Sprintf("client=%s", s.Client))
}
if s.State != "" {
output = append(output, fmt.Sprintf("state=%s", s.State))
}
if s.StateDuration != 0 {
output = append(output, fmt.Sprintf("state_duration=%f", s.StateDuration))
}
if s.Query != "" && !s.IsIdle() {
output = append(output, fmt.Sprintf("query=%s", s.Query))
}
return strings.Join(output, " ")
return output
}
// IsIdle returns true when a session is doing nothing