mirror of
https://github.com/bingohuang/docker-labs.git
synced 2025-10-05 01:43:22 +08:00
Add session close button
This commit is contained in:
@@ -50,6 +50,27 @@ func CreateWSServer() *socketio.Server {
|
||||
return server
|
||||
}
|
||||
|
||||
func CloseSession(s *Session) error {
|
||||
s.Lock()
|
||||
defer s.Unlock()
|
||||
wsServer.BroadcastTo(s.Id, "session end")
|
||||
log.Printf("Starting clean up of session [%s]\n", s.Id)
|
||||
for _, i := range s.Instances {
|
||||
i.Conn.Close()
|
||||
if err := DeleteContainer(i.Name); err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
}
|
||||
if err := DeleteNetwork(s.Id); err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
delete(sessions, s.Id)
|
||||
log.Printf("Cleaned up session [%s]\n", s.Id)
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewSession() (*Session, error) {
|
||||
s := &Session{}
|
||||
s.Id = uuid.NewV4().String()
|
||||
@@ -60,22 +81,7 @@ func NewSession() (*Session, error) {
|
||||
|
||||
// Schedule cleanup of the session
|
||||
time.AfterFunc(4*time.Hour, func() {
|
||||
s = GetSession(s.Id)
|
||||
s.Lock()
|
||||
defer s.Unlock()
|
||||
wsServer.BroadcastTo(s.Id, "session end")
|
||||
log.Printf("Starting clean up of session [%s]\n", s.Id)
|
||||
for _, i := range s.Instances {
|
||||
i.Conn.Close()
|
||||
if err := DeleteContainer(i.Name); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
if err := DeleteNetwork(s.Id); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
delete(sessions, s.Id)
|
||||
log.Printf("Cleaned up session [%s]\n", s.Id)
|
||||
CloseSession(s)
|
||||
})
|
||||
|
||||
if err := CreateNetwork(s.Id); err != nil {
|
||||
|
Reference in New Issue
Block a user