[add] SecureManger 의존성 주입과 RM 받을시 키등록 가능하도록 구현, 로그 메세지 색입혀서 보기 편하게 기능 추가

src/broker/modules/init.go

@@ -11,7 +11,7 @@ import (
 	"net/rpc"
 )
 
-// TODO: RM 처리와 init.go 전반적으로 DI 이용하여 구현
+// TODO: RM 처리와 init.go 전반적으로 DI 이용하여 구현 RM 키등록 부분 구현 예정
 //temporary type for matching manager
 //type match_manager struct{}
 
@@ -31,7 +31,7 @@ type Moscato struct {
 	queue               MsgQueue
 	MicroServiceManager NodeManager `inject:""`
 	match_mng           match_manager
-	secure_mng          secure_manager
+	SecureManager       SecurityManager `inject:""`
 }
 
 type Reply struct {
@@ -154,13 +154,18 @@ func (moscato *Moscato) Receive(msg MsgUnit) (MsgUnit, error) {
 		newNode := MSNode{newMsg.From, newMsg.From}
 		resultAddNode := moscato.MicroServiceManager.AddMicroservice(newNode)
 		if resultAddNode {
-			log.Println("node added successful")
+			log.Println("Node added successful")
 		} else {
-			log.Println("node is already added, ignore RM")
+			log.Println("Node is already added, ignore RM")
+			return msg, nil
 		}
 
 		addr, _ := moscato.MicroServiceManager.GetIpaddr(newMsg.From)
+		moscato.SecureManager.RegKey(newMsg)
+		fmt.Println(moscato.SecureManager.GetNodeKey(newMsg.From))
 		fmt.Println(addr)
+
+		// ackRM 메세지 전송
 		go moscato.Send2MS(addr, newMsg)
 
 		log.Println("RM received")
@@ -206,7 +211,8 @@ func (moscato *Moscato) Run() {
 
 	err := graph.Provide(
 		&inject.Object{Value: NewMStable()},
-		&inject.Object{Value: moscato})
+		&inject.Object{Value: moscato},
+		&inject.Object{Value: NewSecurity()})
 	if err != nil {
 		fmt.Println(err)
 		return

src/broker/modules/message.go

@@ -65,6 +65,7 @@ type SubscriptionMsg struct {
 //Microservice 등록 메세지
 type RegisterMsg struct {
 	Message
+	PrivateKey int64
 }
 
 //Microservice 탈퇴 메세지(없앰)

src/broker/modules/queue.go

@@ -2,7 +2,7 @@ package modules
 
 import (
 	"errors"
-	"log"
+	"github.com/fatih/color"
 )
 
 type MsgQueue struct {
@@ -24,7 +24,8 @@ func (mq *MsgQueue) queue_init() error {
 		return errors.New("Queue Hadlerer Error: Already initialized.")
 	} else if mq.queue == nil {
 		mq.queue = make(chan MsgUnit, 1000)
-		log.Println("queue is initialized.")
+		//log.Println("queue is initialized.")
+		color.Blue("queue is initialized.")
 		return nil
 	} else {
 		close(mq.queue)

src/broker/modules/secure.go

@@ -16,7 +16,7 @@ func NewSecurity() *Security {
 }
 
 type SecurityManager interface {
-	RegKey(ksm KeyShareMsg)
+	RegKey(rm RegisterMsg)
 	GetNodeKey(nodeName string) int64
 	ReEncrypt(fromKey int64, toKey int64, target []int64) []int64
 	ReEncPubMsg(fromPubMsg PublishMsg, nodeName string) *PublishMsg
@@ -28,8 +28,8 @@ type SecurityManager interface {
 /**
 keyShareMsg 에서 각 노드의 private 키를 받아 keyMap 에 저장
 */
-func (sc Security) RegKey(ksm KeyShareMsg) {
-	sc.KeyMap[ksm.Message.From] = ksm.key
+func (sc Security) RegKey(rm RegisterMsg) {
+	sc.KeyMap[rm.Message.From] = strconv.FormatInt(rm.PrivateKey, 10)
 }
 
 /**