Show when session has expired

services/session.go

@@ -23,30 +23,35 @@ func NewSession() (*types.Session, error) {
 	sessions[s.Id] = s
 
 	// Schedule cleanup of the session
-	time.AfterFunc(1*time.Minute, func() {
+	time.AfterFunc(1*time.Hour, func() {
+		s = GetSession(s.Id)
+		log.Printf("Starting clean up of session [%s]\n", s.Id)
 		for _, i := range s.Instances {
 			if err := DeleteContainer(i.Name); err != nil {
 				log.Println(err)
 			}
 		}
-		DeleteNetwork(s.Id)
+		if err := DeleteNetwork(s.Id); err != nil {
+			log.Println(err)
+		}
+		delete(sessions, s.Id)
+		log.Printf("Cleaned up session [%s]\n", s.Id)
 	})
 
 	if err := CreateNetwork(s.Id); err != nil {
 		return nil, err
 	}
 
-	//TODO: Schedule deletion after an hour
-
 	return s, nil
 }
 
 func GetSession(sessionId string) *types.Session {
 	//TODO: Use redis
-	s := sessions[sessionId]
+	s, found := sessions[sessionId]
-	if instances[sessionId] != nil {
+	if found {
 		s.Instances = instances[sessionId]
-	}
-
 		return s
+	} else {
+		return nil
+	}
 }

www/assets/app.js

@@ -48,6 +48,9 @@
 				if ($scope.instances.length) {
 					$scope.showInstance($scope.instances[0]);
 				}
+
+				// Since session exists, we check it still exists every 10 seconds
+				$scope.checkHandler = setInterval(checkSession, 10*1000);
 			}, function(response) {
 				if (response.status == 404) {
 				  document.write('session not found');
@@ -79,5 +82,17 @@
 		}
 
 		$scope.getSession($scope.sessionId);
+
+		function checkSession() {
+			$http({
+				method: 'GET',
+				url: '/sessions/' + $scope.sessionId,
+			}).then(function(response) {}, function(response) {
+				if (response.status == 404) {
+					clearInterval($scope.checkHandler);
+					$scope.showAlert('Session timedout!', 'Your session has expire and all your instances has been deleted.')
+				}
+			});
+		}
 	}]);
 })();

www/index.html

@@ -40,6 +40,9 @@
         <p>
 	Add instances to your playground.
         </p>
+        <p>
+	<strong>Sessions and all their instances are deleted after 1 hour.</strong>
+        </p>
       </div>
 
       <div flex></div>