Просмотр исходного кода

[add] RM 처리 testClient 프로그램과 함께 기초적인 구현

master
kidjung 4 лет назад
Родитель
Сommit
928b175198
1 измененных файлов: 12 добавлений и 3 удалений
  1. 12
    3
      src/broker/modules/init.go

+ 12
- 3
src/broker/modules/init.go Просмотреть файл

5
 	"errors"
5
 	"errors"
6
 	"fmt"
6
 	"fmt"
7
 	"github.com/facebookgo/inject"
7
 	"github.com/facebookgo/inject"
8
+	"github.com/fatih/color"
8
 	"log"
9
 	"log"
9
 	"net"
10
 	"net"
10
 	"net/rpc"
11
 	"net/rpc"
151
 		newMsg = msg.(RegisterMsg)
152
 		newMsg = msg.(RegisterMsg)
152
 
153
 
153
 		newNode := MSNode{newMsg.From, newMsg.From}
154
 		newNode := MSNode{newMsg.From, newMsg.From}
154
-		moscato.MicroServiceManager.AddMicroservice(newNode)
155
+		resultAddNode := moscato.MicroServiceManager.AddMicroservice(newNode)
156
+		if resultAddNode {
157
+			log.Println("node added successful")
158
+		} else {
159
+			log.Println("node is already added, ignore RM")
160
+		}
155
 
161
 
156
 		addr, _ := moscato.MicroServiceManager.GetIpaddr(newMsg.From)
162
 		addr, _ := moscato.MicroServiceManager.GetIpaddr(newMsg.From)
157
 		fmt.Println(addr)
163
 		fmt.Println(addr)
164
+		go moscato.Send2MS(addr, newMsg)
158
 
165
 
159
 		log.Println("RM received")
166
 		log.Println("RM received")
160
 
167
 
183
 		JsonMsg: jmsg,
190
 		JsonMsg: jmsg,
184
 		Kind:    msg.CheckType(),
191
 		Kind:    msg.CheckType(),
185
 	}
192
 	}
186
-	err = client.Call("receiver.MsReceive", args, reply)
193
+	err = client.Call("Receiver.Receive", args, reply)
187
 	if err != nil {
194
 	if err != nil {
188
 		fmt.Println(err)
195
 		fmt.Println(err)
189
 		return
196
 		return
190
 	}
197
 	}
191
-	fmt.Println(reply.CompleteLog) //잘 받았는지 확인 해줌
198
+	//log.Println(reply.CompleteLog) //잘 받았는지 확인 해줌
199
+	// 마이크로 서비스에게 받은 메시지는 노란색으로 출력
200
+	color.Yellow(reply.CompleteLog)
192
 }
201
 }
193
 
202
 
194
 func (moscato *Moscato) Run() {
203
 func (moscato *Moscato) Run() {

Загрузка…
Отмена
Сохранить