Revise message.go/queue.go

Write manage.go

modules/manage.go

@@ -15,5 +15,41 @@ func (node *MSnode) Getipaddr() string {
 
 //모든 Microservice정보 저장
 type MStable struct {
-	IpTable map[string]MSnode
+	NodeTable map[string]MSnode
+}
+
+func (manager *MStable) getIpaddr(nodename string) (string, bool) {
+	//해당 이름의 노드이름이 존재하는지 확인
+	node, exists := manager.NodeTable[nodename]
+
+	//존재하지 않는 경우 nil리턴
+	if !exists {
+		return "", false
+	} else {
+		return node.ipAddr, true
+	}
+}
+
+func (manager *MStable) add_microservice(node MSnode) bool {
+	//삽입 전 존재여부 확인
+	_, exists := manager.NodeTable[node.Getname()]
+
+	if exists {
+		return false
+	} else {
+		manager.NodeTable[node.Getname()] = node
+		return true
+	}
+}
+
+func (manager *MStable) remove_microservice(nodename string) bool {
+	//삭제 전 존재여부 확인
+	_, exists := manager.NodeTable[nodename]
+
+	if exists {
+		delete(manager.NodeTable, nodename)
+		return true
+	} else {
+		return false
+	}
 }

modules/message.go

@@ -1,45 +1,61 @@
 package modules
 
+import "encoding/json"
+
+//import "strconv"
+//*****메세지 타입 상수화
+const (
+	KGM = 1 + iota
+	KSM
+	PM
+	SM
+	RM
+	WM
+)
+
 //*****메세지 틀*****
+
 type Message struct {
-	from    string
+	from    string //ip주소
 	version string
 	time    string
-	kine    string
-	content MsgUnit
+	kind    int //종류
 }
 
 type MsgUnit interface {
 	// ConvertToJson - send 전 json형식으로 바꾸는 함수
 	ConvertToJson() ([]byte, error)
-	// Send - rpc를 이용하여 msg전송
-	Send() (Message, error)
-	// Recieve - rpc를 이용하여 msg전달 받음(rpc call)
-	Recieve() (Message, error)
+	// CheckType - Message의 타입을 알려줌
+	CheckType() int
+	// SetType - Message 객체가 생성 되었을때 종류 정하기
+	//SetType(int)
 }
 
-//**********************
-
 //*****각 메세지 형식 및 정의**********
 
 //KeyGen 명령 메세지
 type KeyGenMsg struct {
 	Message
+	iptable []string
 }
 
 //Key공유 메세지
 type KeyShareMsg struct {
 	Message
+	base32 string
 }
 
 //전달할 내용을 담은 메세지
 type PublishMsg struct {
 	Message
+	subscription []int64 //선호도
+	content      []int64 // 내용
 }
 
 //구독 정보를 담은 메세지
 type SubscriptionMsg struct {
 	Message
+	subscription []int64 //선호도
 }
 
 //Microservice 등록 메세지
@@ -53,3 +69,68 @@ type WithdrawMsg struct {
 }
 
 //**************************
+
+func (msg *KeyGenMsg) ConvertToJson() ([]byte, error) {
+	js := msg
+	jsonBytes, err := json.Marshal(js)
+	return jsonBytes, err
+}
+
+func (msg *KeyShareMsg) ConvertToJson() ([]byte, error) {
+	js := msg
+	jsonBytes, err := json.Marshal(js)
+	return jsonBytes, err
+}
+
+func (msg *PublishMsg) ConvertToJson() ([]byte, error) {
+	js := msg
+	jsonBytes, err := json.Marshal(js)
+	return jsonBytes, err
+}
+
+func (msg *SubscriptionMsg) ConvertToJson() ([]byte, error) {
+	js := msg
+	jsonBytes, err := json.Marshal(js)
+	return jsonBytes, err
+}
+
+func (msg *RegisterMsg) ConvertToJson() ([]byte, error) {
+	js := msg
+	jsonBytes, err := json.Marshal(js)
+	return jsonBytes, err
+}
+
+func (msg *WithdrawMsg) ConvertToJson() ([]byte, error) {
+	js := msg
+	jsonBytes, err := json.Marshal(js)
+	return jsonBytes, err
+}
+
+func (msg Message) CheckType() int {
+	return msg.kind
+}
+
+/*
+func (msg Message)SetType(t int){
+   msg.kind=t
+}
+
+func NewMessage() *Message{
+	//m:=&Message{
+	//from: ip
+	//version: version
+	//time : time
+	//kind : kind(종류)
+	//}
+}
+*/
+
+//*********************************
+
+// Send - rpc를 이용하여 msg전송
+//func (msg *MsgUnit) Send(ipaddr string, message MsgUnit) (error){}
+
+// Recieve - rpc를 이용하여 msg전달 받음(rpc call)
+//func Recieve(msg *MsgUnit) (MsgUnit, error){ }
+
+//*********************************

modules/queue.go

@@ -20,7 +20,7 @@ func (mq *MsgQueue) queue_init() error {
 	if mq.queue != nil && len(mq.queue) != 0 {
 		return errors.New("Queue Hadlerer Error: Already initialized.")
 	} else if mq.queue == nil {
-		mq.queue = make(chan Message)
+		mq.queue = make(chan Message, 1000)
 		return nil
 	} else {
 		close(mq.queue)