[feat] Add delete when withdrawing

src/broker/modules/init.go

@@ -178,7 +178,7 @@ func (moscato *Moscato) Receive(msg MsgUnit) (MsgUnit, error) {
 	switch msg_type {
 
 	case KSM: //Key share msg
-
+		
 	case PM: //Publish msg
 		log.Println("PM received")
 		moscato.queue.push(moscato.preProcessMsg(msg))
@@ -214,14 +214,10 @@ func (moscato *Moscato) Receive(msg MsgUnit) (MsgUnit, error) {
 		go moscato.Send2MS(addr, newMsg)
 
 	case WM: //Withdraw msg
-		//ip := msg.(WithdrawMsg).From
-		//sublist := moscato.SubscriptionManager.ip2sub[ip]
-		//fmt.Println("prev list = ", sublist)
+		ip := msg.(WithdrawMsg).From
 		moscato.MicroServiceManager.RemoveMicroservice(msg.(WithdrawMsg).From)
 		moscato.SecureManager.RemoveSecureKey(msg.(WithdrawMsg).From)
-		//moscato.SubscriptionManager.delete(ip)
-		//sublist2 := moscato.SubscriptionManager.ip2sub[ip]
-		//fmt.Println("after list =", sublist2)
+		moscato.SubscriptionManager.delete(ip)
 
 	default:
 		return nil, errors.New("Message type Error: Not registered message type")
@@ -325,4 +321,4 @@ func Listen() {
 		conn, _ := l.Accept()
 		go rpc.ServeConn(conn)
 	}
-}
+}