mirror of https://github.com/danielmiessler/fabric
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
726 B
Go
39 lines
726 B
Go
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/danielmiessler/fabric/common"
|
|
)
|
|
|
|
type Sessions struct {
|
|
*Storage
|
|
}
|
|
|
|
func (o *Sessions) GetOrCreateSession(name string) (session *Session, err error) {
|
|
session = &Session{Name: name}
|
|
|
|
if o.Exists(name) {
|
|
err = o.LoadAsJson(name, &session.Messages)
|
|
} else {
|
|
fmt.Printf("Creating new session: %s\n", name)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (o *Sessions) SaveSession(session *Session) (err error) {
|
|
return o.SaveAsJson(session.Name, session.Messages)
|
|
}
|
|
|
|
type Session struct {
|
|
Name string
|
|
Messages []*common.Message
|
|
}
|
|
|
|
func (o *Session) IsEmpty() bool {
|
|
return len(o.Messages) == 0
|
|
}
|
|
|
|
func (o *Session) Append(messages ...*common.Message) {
|
|
o.Messages = append(o.Messages, messages...)
|
|
}
|