[fix] matching re-encryption 동작 수정

src/broker/modules/init.go

@@ -126,7 +126,6 @@ func (moscato *Moscato) preProcessMsg(originalMsg MsgUnit) MsgUnit {
 		for index := 0; index < len(pubMsg.Value); index++ {
 			pubMsg.Value[index] = pubMsg.Value[index] - moscato.SecureManager.GetNodeKey(pubMsg.From)
 		}
-		fmt.Println("preprocess : ", pubMsg)
 		return pubMsg
 	} else if originalMsg.CheckType() == SM {
 		subMsg := originalMsg.(SubscriptionMsg)
@@ -136,26 +135,25 @@ func (moscato *Moscato) preProcessMsg(originalMsg MsgUnit) MsgUnit {
 		for index := 0; index < len(subMsg.Value); index++ {
 			subMsg.Value[index] = subMsg.Value[index] - moscato.SecureManager.GetNodeKey(subMsg.From)
 		}
-		fmt.Println("preprocess : ", subMsg)
 		return subMsg
 	}
 	return nil
 }
 
 func (moscato *Moscato) SendWithEncrypt() MsgUnit {
-	mt := <-moscato.SendQueue
-	fmt.Println(mt)
-	if mt.err == nil {
-		for index := 0; index < len(mt.subList); index++ {
-			tmpNode := mt.subList[index]
-			tmpNodeIpAddr, _ := moscato.MicroServiceManager.GetIpaddr(tmpNode)
-			fmt.Println("before reEnc: ", mt.pubMsg)
-			moscato.SecureManager.ReEncPubMsg(mt.pubMsg.(PublishMsg), tmpNode)
-			fmt.Println("after reEnc: ", mt.pubMsg)
-			moscato.Send2MS(tmpNodeIpAddr, mt.pubMsg)
+	for {
+		mt := <-moscato.SendQueue
+		fmt.Println(mt)
+		if mt.err == nil {
+			for index := 0; index < len(mt.subList); index++ {
+				tmpNode := mt.subList[index]
+				tmpNodeIpAddr, _ := moscato.MicroServiceManager.GetIpaddr(tmpNode)
+				moscato.SecureManager.ReEncPubMsg(mt.pubMsg.(PublishMsg), tmpNode)
+				moscato.Send2MS(tmpNodeIpAddr, mt.pubMsg)
+			}
 		}
+		return nil
 	}
-	return nil
 }
 
 //Recieve - MM가 msg전달 받음
@@ -171,10 +169,14 @@ func (moscato *Moscato) Receive(msg MsgUnit) (MsgUnit, error) {
 		log.Println("PM received")
 		fmt.Println("PM: ", msg)
 		moscato.queue.push(moscato.preProcessMsg(msg))
+		fmt.Println("pushed")
 
 	case SM: //Subscription msg
-		fmt.Println("SM: ", msg)
-		moscato.SubscriptionManager.addSubscription(moscato.preProcessMsg(msg))
+		err := moscato.SubscriptionManager.addSubscription(moscato.preProcessMsg(msg))
+		if err != nil {
+			println(err)
+			//return nil, err
+		}
 
 	case RM: //Register msg
 		log.Println("RM received")
@@ -200,6 +202,7 @@ func (moscato *Moscato) Receive(msg MsgUnit) (MsgUnit, error) {
 
 	case WM: //Withdraw msg
 		moscato.MicroServiceManager.RemoveMicroservice(msg.(WithdrawMsg).From)
+		moscato.SecureManager.RemoveSecureKey(msg.(WithdrawMsg).From)
 
 	default:
 		return nil, errors.New("Message type Error: Not registered message type")
@@ -250,8 +253,14 @@ func (moscato *Moscato) Run() {
 	}
 
 	//go routine -> matching 동작
-	go moscato.Matching()
-	go moscato.SendWithEncrypt()
+	go func() {
+		for {
+			msg := moscato.queue.pop(true)
+			fmt.Println(msg)
+			go moscato.Matching(msg)
+			go moscato.SendWithEncrypt()
+		}
+	}()
 
 	//go moscato.CheckQueue()
 

src/broker/modules/matching.go

@@ -12,8 +12,8 @@ type match_manager struct {
 }
 
 // Matching -> Return (list of IP addresses of matched subs, Pub Msg, error)
-func (moscato *Moscato) Matching() {
-	msg := moscato.queue.pop(true)
+func (moscato *Moscato) Matching(msg MsgUnit) {
+	//msg := moscato.queue.pop(true)
 	topic := msg.(PublishMsg).Topic
 	value := msg.(PublishMsg).Value
 	sub_mng := moscato.SubscriptionManager

src/broker/modules/secure.go

@@ -2,6 +2,7 @@ package modules
 
 import (
 	"fmt"
+	"log"
 	"strconv"
 )
 
@@ -21,6 +22,7 @@ type SecurityManager interface {
 	GetNodeKey(nodeName string) int64
 	ReEncrypt(fromKey int64, toKey int64, target []int64) []int64
 	ReEncPubMsg(fromPubMsg PublishMsg, nodeName string) *PublishMsg
+	RemoveSecureKey(nodeName string) bool
 	//CompareTopic(topic1 []int64, topic2 []int64) int
 	//CompareDigit(topic1 int64, topic2 int64) int
 	//CompareAlpha(topic1 []int64, topic2 []int64) int
@@ -79,6 +81,19 @@ func (sc Security) ReEncPubMsg(fromPubMsg PublishMsg, nodeName string) *PublishM
 	return toPubMsg
 }
 
+func (sc *Security) RemoveSecureKey(nodeName string) bool {
+	//삭제 전 존재여부 확인
+	_, exists := sc.KeyMap[nodeName]
+
+	if exists {
+		delete(sc.KeyMap, nodeName)
+		log.Println("[" + nodeName + "] : delete Key successful")
+		return true
+	} else {
+		return false
+	}
+}
+
 /**
 Compare 함수들은 같으면 0 다르면 -1 (비교가 필요한 경우 오름차순 1 내림차순 -1)
 */