Merge pull request #8 from Famcache/fix/2.0.1

Disconnect: Cleanup Jobs after disconnect

VERSION

@@ -1 +1 @@
-2.0.0
+2.0.1

domain/jobs/manager.go

@@ -4,6 +4,6 @@ type JobsManager interface {
 	Add(peerId string, delay uint64, isPeriodic bool) string
 	Get(id string) (Job, bool)
 	Cancel(id string)
-	Jobs() []Job
+	Jobs() *map[string]Job
 	Chan() <-chan Job
 }

pkg/actor/actor_impl.go

@@ -51,7 +51,7 @@ func (actor *Actor) DisconnectPeer(peer connection.Peer) {
 	actor.peers.Remove(peer)
 	jobs := actor.jobs.Jobs()
 
-	for _, job := range jobs {
+	for _, job := range *jobs {
 		if job.PeerId() == peer.ID() {
 			actor.jobs.Cancel(job.ID())
 		}

pkg/actor/register_job.go

@@ -13,7 +13,5 @@ func (actor *Actor) RegisterJob(peer connection.Peer, query command.JobCommand)
 
 	logger.Info(fmt.Sprintf("Job registered with id %s", id))
 
-	reply := fmt.Sprintf("%s %s", query.ID(), id)
-
-	query.ReplyOK(peer.Conn(), reply)
+	query.ReplyOK(peer.Conn(), id)
 }

pkg/jobs/manager_impl.go

@@ -43,12 +43,6 @@ func (jm *JobsManager) Chan() <-chan job.Job {
 	return jm.triggerChan
 }
 
-func (jm *JobsManager) Jobs() []job.Job {
-	jobs := make([]job.Job, len(jm.jobs))
-
-	for _, job := range jm.jobs {
-		jobs = append(jobs, job)
-	}
-
-	return jobs
+func (jm *JobsManager) Jobs() *map[string]mJob.Job {
+	return &jm.jobs
 }