|
|
@@ -0,0 +1,617 @@
|
|
|
1
|
+package main
|
|
|
2
|
+
|
|
|
3
|
+// TODO: 등록확인 되기 전에는 PM,SM 전송 불가하게 만들 것, 실행파일로 만든 후, AWS 이용해서 여러개 마이크로서비스 상태에서 테스트 해볼것
|
|
|
4
|
+
|
|
|
5
|
+import (
|
|
|
6
|
+ "bufio"
|
|
|
7
|
+ "encoding/json"
|
|
|
8
|
+ "errors"
|
|
|
9
|
+ "fmt"
|
|
|
10
|
+ "log"
|
|
|
11
|
+ "math/rand"
|
|
|
12
|
+ "net"
|
|
|
13
|
+ "net/rpc"
|
|
|
14
|
+ "os"
|
|
|
15
|
+ "os/signal"
|
|
|
16
|
+ "strconv"
|
|
|
17
|
+ "strings"
|
|
|
18
|
+ "syscall"
|
|
|
19
|
+ "time"
|
|
|
20
|
+ "unicode"
|
|
|
21
|
+)
|
|
|
22
|
+
|
|
|
23
|
+const (
|
|
|
24
|
+ KGM = 1 + iota
|
|
|
25
|
+ KSM
|
|
|
26
|
+ PM
|
|
|
27
|
+ SM
|
|
|
28
|
+ RM
|
|
|
29
|
+ WM
|
|
|
30
|
+)
|
|
|
31
|
+
|
|
|
32
|
+type Message struct {
|
|
|
33
|
+ From string //ip 주소
|
|
|
34
|
+ Version string
|
|
|
35
|
+ Time string
|
|
|
36
|
+ Kind int //종류
|
|
|
37
|
+}
|
|
|
38
|
+
|
|
|
39
|
+type MsgUnit interface {
|
|
|
40
|
+ // ConvertToJson - send 전 json 형식으로 바꾸는 함수
|
|
|
41
|
+ ConvertToJson() ([]byte, error)
|
|
|
42
|
+ // CheckType - Message 의 타입을 알려줌
|
|
|
43
|
+ CheckType() int
|
|
|
44
|
+}
|
|
|
45
|
+
|
|
|
46
|
+type RegisterMsg struct {
|
|
|
47
|
+ Message
|
|
|
48
|
+ PrivateKey int64
|
|
|
49
|
+}
|
|
|
50
|
+
|
|
|
51
|
+type SubscriptionMsg struct {
|
|
|
52
|
+ Message
|
|
|
53
|
+ Topic []int64 //대주제
|
|
|
54
|
+ Value []int64 //피연산자
|
|
|
55
|
+ Operator []string //연산자
|
|
|
56
|
+ IsAlpha bool //value가 숫자인지 문자열인지
|
|
|
57
|
+}
|
|
|
58
|
+
|
|
|
59
|
+type PublishMsg struct {
|
|
|
60
|
+ Message
|
|
|
61
|
+ Topic []int64 //대주제
|
|
|
62
|
+ Value []int64 //Topic 의 세부적인 내용
|
|
|
63
|
+ Content []int64 // 내용
|
|
|
64
|
+}
|
|
|
65
|
+
|
|
|
66
|
+type WithdrawMsg struct {
|
|
|
67
|
+ Message
|
|
|
68
|
+}
|
|
|
69
|
+
|
|
|
70
|
+type Receiver struct {
|
|
|
71
|
+ thisNodeAddr string
|
|
|
72
|
+ microService MicroService
|
|
|
73
|
+}
|
|
|
74
|
+
|
|
|
75
|
+type MicroService struct {
|
|
|
76
|
+ ClientAddr string
|
|
|
77
|
+ PrivateKey int64
|
|
|
78
|
+ ShareKey int64
|
|
|
79
|
+ // 메세지 미들웨어에 연결되어 있는가
|
|
|
80
|
+ IsConnected chan bool
|
|
|
81
|
+ MMAddress string
|
|
|
82
|
+}
|
|
|
83
|
+
|
|
|
84
|
+func (receiver Receiver) Receive(args Args, reply *Reply) error {
|
|
|
85
|
+ switch args.Kind {
|
|
|
86
|
+
|
|
|
87
|
+ case KSM:
|
|
|
88
|
+ //var msg KeyShareMsg
|
|
|
89
|
+ //err := json.Unmarshal(args.JsonMsg, &msg)
|
|
|
90
|
+ //if err != nil {
|
|
|
91
|
+ // return err
|
|
|
92
|
+ //}
|
|
|
93
|
+ //go func() {
|
|
|
94
|
+ // _, err := receiver.moscato.Receive(msg)
|
|
|
95
|
+ // if err != nil {
|
|
|
96
|
+ //
|
|
|
97
|
+ // }
|
|
|
98
|
+ //}()
|
|
|
99
|
+ //reply.CompleteLog = "received completely"
|
|
|
100
|
+ case PM:
|
|
|
101
|
+ var msg PublishMsg
|
|
|
102
|
+ err := json.Unmarshal(args.JsonMsg, &msg)
|
|
|
103
|
+ if err != nil {
|
|
|
104
|
+ return err
|
|
|
105
|
+ }
|
|
|
106
|
+ go func() {
|
|
|
107
|
+ receiver.microService.Receive(msg)
|
|
|
108
|
+ }()
|
|
|
109
|
+ reply.CompleteLog = "[" + receiver.thisNodeAddr + "] : pubMsg received"
|
|
|
110
|
+ //var msg PublishMsg
|
|
|
111
|
+ //err := json.Unmarshal(args.JsonMsg, &msg)
|
|
|
112
|
+ //if err != nil {
|
|
|
113
|
+ // return err
|
|
|
114
|
+ //}
|
|
|
115
|
+ //
|
|
|
116
|
+ //go func() {
|
|
|
117
|
+ // _, err := receiver.moscato.Receive(msg)
|
|
|
118
|
+ // if err != nil {
|
|
|
119
|
+ //
|
|
|
120
|
+ // }
|
|
|
121
|
+ //}()
|
|
|
122
|
+ //reply.CompleteLog = "PM received completely"
|
|
|
123
|
+ case SM:
|
|
|
124
|
+ //var msg SubscriptionMsg
|
|
|
125
|
+ //err := json.Unmarshal(args.JsonMsg, &msg)
|
|
|
126
|
+ //if err != nil {
|
|
|
127
|
+ // return err
|
|
|
128
|
+ //}
|
|
|
129
|
+ //go func() {
|
|
|
130
|
+ // _, err := receiver.moscato.Receive(msg)
|
|
|
131
|
+ // if err != nil {
|
|
|
132
|
+ //
|
|
|
133
|
+ // }
|
|
|
134
|
+ //}()
|
|
|
135
|
+ //reply.CompleteLog = "received completely"
|
|
|
136
|
+ case RM:
|
|
|
137
|
+ var msg RegisterMsg
|
|
|
138
|
+ err := json.Unmarshal(args.JsonMsg, &msg)
|
|
|
139
|
+ if err != nil {
|
|
|
140
|
+ return err
|
|
|
141
|
+ }
|
|
|
142
|
+ go func() {
|
|
|
143
|
+ receiver.microService.Receive(msg)
|
|
|
144
|
+ }()
|
|
|
145
|
+ reply.CompleteLog = "[" + receiver.thisNodeAddr + "] : ackRM received"
|
|
|
146
|
+ case WM:
|
|
|
147
|
+ //var msg WithdrawMsg
|
|
|
148
|
+ //err := json.Unmarshal(args.JsonMsg, &msg)
|
|
|
149
|
+ //if err != nil {
|
|
|
150
|
+ // return err
|
|
|
151
|
+ //}
|
|
|
152
|
+ //go func() {
|
|
|
153
|
+ // _, err := receiver.moscato.Receive(msg)
|
|
|
154
|
+ // if err != nil {
|
|
|
155
|
+ //
|
|
|
156
|
+ // }
|
|
|
157
|
+ //}()
|
|
|
158
|
+ //reply.CompleteLog = "received completely"
|
|
|
159
|
+ default:
|
|
|
160
|
+ return errors.New("message type Error: Not registered message type")
|
|
|
161
|
+ }
|
|
|
162
|
+ //reply.CompleteLog = "received completely"
|
|
|
163
|
+ return nil
|
|
|
164
|
+}
|
|
|
165
|
+
|
|
|
166
|
+func (microService MicroService) Receive(msg MsgUnit) {
|
|
|
167
|
+ logger := NewMyLogger()
|
|
|
168
|
+ logger.Sync()
|
|
|
169
|
+
|
|
|
170
|
+ var msg_type = msg.CheckType()
|
|
|
171
|
+ //메세지 타입에 따라 다르게 처리
|
|
|
172
|
+ switch msg_type {
|
|
|
173
|
+
|
|
|
174
|
+ case KSM: //Key share msg
|
|
|
175
|
+
|
|
|
176
|
+ case PM: //Publish msg
|
|
|
177
|
+ //PM print
|
|
|
178
|
+ fmt.Println("----- received PM ------")
|
|
|
179
|
+ DecryptionMsg(msg.(PublishMsg), microService.ShareKey, microService.PrivateKey)
|
|
|
180
|
+ fmt.Println("------------------------")
|
|
|
181
|
+
|
|
|
182
|
+ case SM: //Subscription msg
|
|
|
183
|
+
|
|
|
184
|
+ case RM: //Register msg
|
|
|
185
|
+
|
|
|
186
|
+ microService.IsConnected <- true
|
|
|
187
|
+ logger.Info("MM Registered this node Complete!")
|
|
|
188
|
+
|
|
|
189
|
+ case WM: //Withdraw msg
|
|
|
190
|
+ //moscato.MicroServiceManager.RemoveMicroservice(msg.(*WithdrawMsg).From)
|
|
|
191
|
+
|
|
|
192
|
+ default:
|
|
|
193
|
+ errors.New("message type Error: Not registered message type")
|
|
|
194
|
+ }
|
|
|
195
|
+
|
|
|
196
|
+ return
|
|
|
197
|
+}
|
|
|
198
|
+
|
|
|
199
|
+func (msg SubscriptionMsg) ConvertToJson() ([]byte, error) {
|
|
|
200
|
+ js := msg
|
|
|
201
|
+ jsonBytes, err := json.Marshal(js)
|
|
|
202
|
+ return jsonBytes, err
|
|
|
203
|
+}
|
|
|
204
|
+
|
|
|
205
|
+func (msg PublishMsg) ConvertToJson() ([]byte, error) {
|
|
|
206
|
+ js := msg
|
|
|
207
|
+ jsonBytes, err := json.Marshal(js)
|
|
|
208
|
+ return jsonBytes, err
|
|
|
209
|
+}
|
|
|
210
|
+
|
|
|
211
|
+func (msg RegisterMsg) ConvertToJson() ([]byte, error) {
|
|
|
212
|
+ js := msg
|
|
|
213
|
+ jsonBytes, err := json.Marshal(js)
|
|
|
214
|
+ return jsonBytes, err
|
|
|
215
|
+}
|
|
|
216
|
+
|
|
|
217
|
+func (msg WithdrawMsg) ConvertToJson() ([]byte, error) {
|
|
|
218
|
+ js := msg
|
|
|
219
|
+ jsonBytes, err := json.Marshal(js)
|
|
|
220
|
+ return jsonBytes, err
|
|
|
221
|
+}
|
|
|
222
|
+
|
|
|
223
|
+func (msg Message) CheckType() int {
|
|
|
224
|
+ return msg.Kind
|
|
|
225
|
+}
|
|
|
226
|
+
|
|
|
227
|
+func CreatePubMsg(msg Message, topic string, value string, content string) PublishMsg {
|
|
|
228
|
+ //toPubMsg := new(PublishMsg)
|
|
|
229
|
+ //toPubMsg.Message = msg
|
|
|
230
|
+
|
|
|
231
|
+ intArr := []rune(topic)
|
|
|
232
|
+ //fmt.Print("Topic length ")
|
|
|
233
|
+ //fmt.Println(len(intArr))
|
|
|
234
|
+ //fmt.Println(len(toPubMsg.Topic))
|
|
|
235
|
+ var topicInt []int64
|
|
|
236
|
+ for index := 0; index < len(intArr); index++ {
|
|
|
237
|
+ //toPubMsg.Topic = append(toPubMsg.Topic, int64(intArr[index]))
|
|
|
238
|
+ topicInt = append(topicInt, int64(intArr[index]))
|
|
|
239
|
+ }
|
|
|
240
|
+ //fmt.Println(len(toPubMsg.Topic))
|
|
|
241
|
+
|
|
|
242
|
+ intArr = []rune(value)
|
|
|
243
|
+ var valueInt []int64
|
|
|
244
|
+ strInt64, _ := strconv.ParseInt(value, 10, 64)
|
|
|
245
|
+ if unicode.IsDigit(intArr[0]) {
|
|
|
246
|
+ valueInt = append(valueInt, strInt64)
|
|
|
247
|
+ } else {
|
|
|
248
|
+ for index := 0; index < len(intArr); index++ {
|
|
|
249
|
+ //toPubMsg.Value = append(toPubMsg.Value, int64(intArr[index]))
|
|
|
250
|
+ valueInt = append(valueInt, int64(intArr[index]))
|
|
|
251
|
+ }
|
|
|
252
|
+ }
|
|
|
253
|
+
|
|
|
254
|
+ intArr = []rune(content)
|
|
|
255
|
+ var contentInt []int64
|
|
|
256
|
+ for index := 0; index < len(intArr); index++ {
|
|
|
257
|
+ //toPubMsg.content = append(toPubMsg.content, int64(intArr[index]))
|
|
|
258
|
+ contentInt = append(contentInt, int64(intArr[index]))
|
|
|
259
|
+ }
|
|
|
260
|
+
|
|
|
261
|
+ return PublishMsg{msg, topicInt, valueInt, contentInt}
|
|
|
262
|
+}
|
|
|
263
|
+
|
|
|
264
|
+func CreateSubMsg(msg Message, topic string, value string, operator string, isAlpha bool) SubscriptionMsg {
|
|
|
265
|
+ //toPubMsg := new(PublishMsg)
|
|
|
266
|
+ //toPubMsg.Message = msg
|
|
|
267
|
+
|
|
|
268
|
+ intArr := []rune(topic)
|
|
|
269
|
+ //fmt.Print("Topic length ")
|
|
|
270
|
+ //fmt.Println(len(intArr))
|
|
|
271
|
+ //fmt.Println(len(toPubMsg.Topic))
|
|
|
272
|
+ var topicInt []int64
|
|
|
273
|
+ for index := 0; index < len(intArr); index++ {
|
|
|
274
|
+ //toPubMsg.Topic = append(toPubMsg.Topic, int64(intArr[index]))
|
|
|
275
|
+ topicInt = append(topicInt, int64(intArr[index]))
|
|
|
276
|
+ }
|
|
|
277
|
+ //fmt.Println(len(toPubMsg.Topic))
|
|
|
278
|
+ var valueInt []int64
|
|
|
279
|
+ if isAlpha {
|
|
|
280
|
+ intArr = []rune(value)
|
|
|
281
|
+
|
|
|
282
|
+ for index := 0; index < len(intArr); index++ {
|
|
|
283
|
+ //toPubMsg.Value = append(toPubMsg.Value, int64(intArr[index]))
|
|
|
284
|
+ valueInt = append(valueInt, int64(intArr[index]))
|
|
|
285
|
+ }
|
|
|
286
|
+ } else {
|
|
|
287
|
+ stringSlice := strings.Split(value, " ")
|
|
|
288
|
+ for index := 0; index < len(stringSlice); index++ {
|
|
|
289
|
+ strInt64, _ := strconv.ParseInt(stringSlice[index], 10, 64)
|
|
|
290
|
+ valueInt = append(valueInt, strInt64)
|
|
|
291
|
+ }
|
|
|
292
|
+ }
|
|
|
293
|
+ var operatorSlice []string
|
|
|
294
|
+ operatorSlice = strings.Split(operator, " ")
|
|
|
295
|
+
|
|
|
296
|
+ //intArr = []rune(content)
|
|
|
297
|
+ //var contentInt [] int64
|
|
|
298
|
+ //for index := 0; index < len(intArr); index++ {
|
|
|
299
|
+ // //toPubMsg.content = append(toPubMsg.content, int64(intArr[index]))
|
|
|
300
|
+ // contentInt = append(contentInt, int64(intArr[index]))
|
|
|
301
|
+ //}
|
|
|
302
|
+
|
|
|
303
|
+ return SubscriptionMsg{msg, topicInt, valueInt, operatorSlice, isAlpha}
|
|
|
304
|
+}
|
|
|
305
|
+
|
|
|
306
|
+type Args struct { // 매개변수
|
|
|
307
|
+ JsonMsg []byte
|
|
|
308
|
+ Kind int
|
|
|
309
|
+}
|
|
|
310
|
+
|
|
|
311
|
+type Reply struct { // 리턴값
|
|
|
312
|
+ CompleteLog string
|
|
|
313
|
+}
|
|
|
314
|
+
|
|
|
315
|
+func generateRanUint64() uint64 {
|
|
|
316
|
+ rand.Seed(time.Now().UnixNano())
|
|
|
317
|
+ //return uint64(rand.Uint32())<<32 + uint64(rand.Uint32())
|
|
|
318
|
+ return uint64(0 + uint64(rand.Uint32()))
|
|
|
319
|
+}
|
|
|
320
|
+
|
|
|
321
|
+func main() {
|
|
|
322
|
+ logger := NewMyLogger()
|
|
|
323
|
+ logger.Sync()
|
|
|
324
|
+
|
|
|
325
|
+ var argu string
|
|
|
326
|
+ for _, v := range os.Args {
|
|
|
327
|
+ if strings.Index(v, "-mma") == 0 {
|
|
|
328
|
+ argu = strings.Replace(v, "-mma=", "", -1)
|
|
|
329
|
+ break
|
|
|
330
|
+ }
|
|
|
331
|
+ }
|
|
|
332
|
+ MMAddress := argu
|
|
|
333
|
+ if argu == "" {
|
|
|
334
|
+ logger.Fatal("there is no Message Middleware address")
|
|
|
335
|
+ return
|
|
|
336
|
+ }
|
|
|
337
|
+
|
|
|
338
|
+ args := new(Args)
|
|
|
339
|
+ reply := new(Reply)
|
|
|
340
|
+
|
|
|
341
|
+ currentIP := getCurrentIPAddr()
|
|
|
342
|
+ const debugPK = 12345
|
|
|
343
|
+ privateKey := int64(generateRanUint64())
|
|
|
344
|
+ shareKey := int64(9999)
|
|
|
345
|
+
|
|
|
346
|
+ microService := MicroService{currentIP, privateKey, shareKey, make(chan bool), MMAddress}
|
|
|
347
|
+ receiver := Receiver{thisNodeAddr: currentIP, microService: microService}
|
|
|
348
|
+
|
|
|
349
|
+ logger.Info("current machine address : " + currentIP + " / MM address : " + MMAddress)
|
|
|
350
|
+ logger.Debug("private key : " + strconv.FormatUint(uint64(privateKey), 10))
|
|
|
351
|
+
|
|
|
352
|
+ errReg := rpc.Register(receiver)
|
|
|
353
|
+ if errReg != nil {
|
|
|
354
|
+ log.Println(errReg)
|
|
|
355
|
+ return
|
|
|
356
|
+ }
|
|
|
357
|
+
|
|
|
358
|
+ client, err := rpc.Dial("tcp", microService.MMAddress+":8160") // RPC 서버에 연결
|
|
|
359
|
+ if err != nil {
|
|
|
360
|
+ logger.Fatal(err.Error())
|
|
|
361
|
+ return
|
|
|
362
|
+ }
|
|
|
363
|
+ defer client.Close() // main 함수가 끝나기 직전에 RPC 연결을 닫음
|
|
|
364
|
+
|
|
|
365
|
+ go Listen()
|
|
|
366
|
+
|
|
|
367
|
+ // Register 메세지 생성
|
|
|
368
|
+ message := Message{From: microService.ClientAddr, Version: "1", Time: "1", Kind: RM}
|
|
|
369
|
+ regMsg := RegisterMsg{message, microService.PrivateKey}
|
|
|
370
|
+ RJsonMsg, _ := regMsg.ConvertToJson()
|
|
|
371
|
+ args.JsonMsg = RJsonMsg
|
|
|
372
|
+ args.Kind = RM
|
|
|
373
|
+ err = client.Call("Receiver.MmReceive", args, reply)
|
|
|
374
|
+ logger.Info("RM sent")
|
|
|
375
|
+ if err != nil {
|
|
|
376
|
+ fmt.Println(err)
|
|
|
377
|
+ return
|
|
|
378
|
+ }
|
|
|
379
|
+ logger.Debug("[MM] " + reply.CompleteLog)
|
|
|
380
|
+
|
|
|
381
|
+ sigs := make(chan os.Signal, 1)
|
|
|
382
|
+ done := make(chan bool, 1)
|
|
|
383
|
+
|
|
|
384
|
+ signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
|
|
385
|
+
|
|
|
386
|
+ go func() {
|
|
|
387
|
+ sig := <-sigs
|
|
|
388
|
+ withDraw(message, client)
|
|
|
389
|
+ //fmt.Println(sig)
|
|
|
390
|
+ _ = sig
|
|
|
391
|
+ done <- true
|
|
|
392
|
+ fmt.Println("\nquit Moscato Successful and terminate microservice")
|
|
|
393
|
+ os.Exit(0)
|
|
|
394
|
+ }()
|
|
|
395
|
+
|
|
|
396
|
+ // 연결 되면 채널로 연결 확인이 되어야 다음 단계로 넘어감
|
|
|
397
|
+ checkConnect := <-microService.IsConnected
|
|
|
398
|
+ _ = checkConnect
|
|
|
399
|
+ fmt.Scanln()
|
|
|
400
|
+ logger.Info("sending Subscription messages")
|
|
|
401
|
+
|
|
|
402
|
+ /*
|
|
|
403
|
+ 파일에서 subscription 읽어서 subscription 보내기
|
|
|
404
|
+ **/
|
|
|
405
|
+ subFile, err := os.Open("subscription.txt")
|
|
|
406
|
+ if err != nil {
|
|
|
407
|
+ fmt.Println("sub")
|
|
|
408
|
+ log.Fatalf("Error when opening file: %s", err)
|
|
|
409
|
+ }
|
|
|
410
|
+ defer subFile.Close()
|
|
|
411
|
+ fileScanner := bufio.NewScanner(subFile)
|
|
|
412
|
+ var subscriptionText string
|
|
|
413
|
+ numSub := 0
|
|
|
414
|
+ file1, _ := os.Create("./subTimeLog.log")
|
|
|
415
|
+ defer file1.Close()
|
|
|
416
|
+ for fileScanner.Scan() {
|
|
|
417
|
+ subscriptionText = fileScanner.Text()
|
|
|
418
|
+ subscriptionTextSlice := strings.Split(subscriptionText, "/")
|
|
|
419
|
+ subTopic := subscriptionTextSlice[0]
|
|
|
420
|
+ subValue := subscriptionTextSlice[1]
|
|
|
421
|
+ subOperator := subscriptionTextSlice[2]
|
|
|
422
|
+ var subIsAlpha bool
|
|
|
423
|
+ if subscriptionTextSlice[3] == "true" {
|
|
|
424
|
+ subIsAlpha = true
|
|
|
425
|
+ } else if subscriptionTextSlice[3] == "false" {
|
|
|
426
|
+ subIsAlpha = false
|
|
|
427
|
+ } else {
|
|
|
428
|
+ println(numSub, "subscription isAlpha type error.")
|
|
|
429
|
+ }
|
|
|
430
|
+
|
|
|
431
|
+ numSub++
|
|
|
432
|
+ message = Message{From: microService.ClientAddr, Version: "1", Time: "1", Kind: SM}
|
|
|
433
|
+
|
|
|
434
|
+ // 메세지 생성시간 측정
|
|
|
435
|
+ subMakingStartTime := time.Now()
|
|
|
436
|
+
|
|
|
437
|
+ subMsg := CreateSubMsg(message, subTopic, subValue, subOperator, subIsAlpha)
|
|
|
438
|
+ logger.Debug("subMsg #" + strconv.Itoa(numSub) + " topic: " + subTopic + " / value: " + subValue + " / operator: " + subOperator)
|
|
|
439
|
+ logger.Debug("before enc subMsg #" + strconv.Itoa(numSub) + " topic: " + IntSlice2String(subMsg.Topic))
|
|
|
440
|
+
|
|
|
441
|
+ //subMsg := CreateSubMsg(message, "soccer", "player", "==", true)
|
|
|
442
|
+ //fmt.Println(pubMsg)
|
|
|
443
|
+ subMsg = EncryptionSubMsg(subMsg, microService.ShareKey, microService.PrivateKey)
|
|
|
444
|
+ logger.Debug("after enc subMsg #" + strconv.Itoa(numSub) + " topic: " + IntSlice2String(subMsg.Topic))
|
|
|
445
|
+ jsonMsg, _ := subMsg.ConvertToJson()
|
|
|
446
|
+ args.JsonMsg = jsonMsg
|
|
|
447
|
+ args.Kind = subMsg.Kind
|
|
|
448
|
+ //fmt.Println(subMsg)
|
|
|
449
|
+
|
|
|
450
|
+ //fmt.Println(string(args.JsonMsg))
|
|
|
451
|
+ subMakingElapsedTime := time.Since(subMakingStartTime)
|
|
|
452
|
+
|
|
|
453
|
+ // 메세지 생성 완료시간 측정
|
|
|
454
|
+ fmt.Printf("sub 생성시간: %d\n", subMakingElapsedTime.Nanoseconds())
|
|
|
455
|
+
|
|
|
456
|
+ fmt.Fprintln(file1, subMakingElapsedTime.Nanoseconds())
|
|
|
457
|
+
|
|
|
458
|
+ err = client.Call("Receiver.MmReceive", args, reply)
|
|
|
459
|
+ logger.Info("SM sent! #" + strconv.Itoa(numSub))
|
|
|
460
|
+
|
|
|
461
|
+ if err != nil {
|
|
|
462
|
+ fmt.Println(err)
|
|
|
463
|
+ return
|
|
|
464
|
+ }
|
|
|
465
|
+ //log.Println(reply.CompleteLog)
|
|
|
466
|
+ logger.Debug("[MM] " + reply.CompleteLog)
|
|
|
467
|
+
|
|
|
468
|
+ }
|
|
|
469
|
+
|
|
|
470
|
+ fmt.Scanln()
|
|
|
471
|
+ logger.Info("sending Publish messages")
|
|
|
472
|
+
|
|
|
473
|
+ pubFile, err := os.Open("publish.txt")
|
|
|
474
|
+ if err != nil {
|
|
|
475
|
+ log.Fatalf("Error when opening file: %s", err)
|
|
|
476
|
+ }
|
|
|
477
|
+ defer pubFile.Close()
|
|
|
478
|
+ fileScanner2 := bufio.NewScanner(pubFile)
|
|
|
479
|
+ var publishText string
|
|
|
480
|
+ numPub := 0
|
|
|
481
|
+ file2, _ := os.Create("./pubTimeLog.log")
|
|
|
482
|
+ defer file2.Close()
|
|
|
483
|
+
|
|
|
484
|
+ for fileScanner2.Scan() {
|
|
|
485
|
+ publishText = fileScanner2.Text()
|
|
|
486
|
+ publishTextSlice := strings.Split(publishText, "/")
|
|
|
487
|
+ //fmt.Println(publishTextSlice)
|
|
|
488
|
+ pubTopic := publishTextSlice[0]
|
|
|
489
|
+ pubValue := publishTextSlice[1]
|
|
|
490
|
+ pubContent := publishTextSlice[2]
|
|
|
491
|
+
|
|
|
492
|
+ numPub++
|
|
|
493
|
+ message = Message{From: microService.ClientAddr, Version: "1", Time: "1", Kind: PM}
|
|
|
494
|
+
|
|
|
495
|
+ pubMsg := CreatePubMsg(message, pubTopic, pubValue, pubContent)
|
|
|
496
|
+ logger.Debug("pubMsg #" + strconv.Itoa(numPub) + " topic: " + pubTopic + " / value: " + pubValue + " / content: " + pubContent)
|
|
|
497
|
+ logger.Debug("before enc pubMsg #" + strconv.Itoa(numPub) + " topic: " + IntSlice2String(pubMsg.Topic))
|
|
|
498
|
+
|
|
|
499
|
+ pubMakingStartTime := time.Now()
|
|
|
500
|
+ pubMsg = EncryptionPubMsg(pubMsg, microService.ShareKey, microService.PrivateKey)
|
|
|
501
|
+ pubMakingElapsedTime := time.Since(pubMakingStartTime)
|
|
|
502
|
+ fmt.Printf("pubMsg enc시간: %d\n\n", pubMakingElapsedTime.Nanoseconds())
|
|
|
503
|
+ fmt.Fprintln(file2,pubMakingElapsedTime.Nanoseconds())
|
|
|
504
|
+ logger.Debug("after enc pubMsg #" + strconv.Itoa(numPub) + " topic: " + IntSlice2String(pubMsg.Topic))
|
|
|
505
|
+ jsonMsg, _ := pubMsg.ConvertToJson()
|
|
|
506
|
+ args.JsonMsg = jsonMsg
|
|
|
507
|
+ args.Kind = pubMsg.Kind
|
|
|
508
|
+
|
|
|
509
|
+
|
|
|
510
|
+ err = client.Call("Receiver.MmReceive", args, reply)
|
|
|
511
|
+ logger.Info("PM sent! #" + strconv.Itoa(numPub))
|
|
|
512
|
+ if err != nil {
|
|
|
513
|
+ fmt.Println(err)
|
|
|
514
|
+ return
|
|
|
515
|
+ }
|
|
|
516
|
+
|
|
|
517
|
+ logger.Debug("[MM]: " + reply.CompleteLog)
|
|
|
518
|
+
|
|
|
519
|
+ }
|
|
|
520
|
+
|
|
|
521
|
+ <-done
|
|
|
522
|
+ return
|
|
|
523
|
+}
|
|
|
524
|
+
|
|
|
525
|
+func withDraw(msg Message, client *rpc.Client) {
|
|
|
526
|
+ msg.Kind = WM
|
|
|
527
|
+ wdMsg := WithdrawMsg{msg}
|
|
|
528
|
+ var args Args
|
|
|
529
|
+ args.JsonMsg, _ = wdMsg.ConvertToJson()
|
|
|
530
|
+ args.Kind = WM
|
|
|
531
|
+ var reply Reply
|
|
|
532
|
+ client.Call("Receiver.MmReceive", args, reply)
|
|
|
533
|
+}
|
|
|
534
|
+
|
|
|
535
|
+func EncryptionPubMsg(msg PublishMsg, gyKey int64, privateKey int64) PublishMsg {
|
|
|
536
|
+ for index := range msg.Topic {
|
|
|
537
|
+ msg.Topic[index] = msg.Topic[index] + gyKey + privateKey
|
|
|
538
|
+ }
|
|
|
539
|
+ for index := range msg.Value {
|
|
|
540
|
+ msg.Value[index] = msg.Value[index] + gyKey + privateKey
|
|
|
541
|
+ }
|
|
|
542
|
+ for index := range msg.Content {
|
|
|
543
|
+ msg.Content[index] = msg.Content[index] + gyKey + privateKey
|
|
|
544
|
+ }
|
|
|
545
|
+
|
|
|
546
|
+ return msg
|
|
|
547
|
+}
|
|
|
548
|
+
|
|
|
549
|
+func EncryptionSubMsg(msg SubscriptionMsg, gyKey int64, privateKey int64) SubscriptionMsg {
|
|
|
550
|
+ for index := range msg.Topic {
|
|
|
551
|
+ msg.Topic[index] = msg.Topic[index] + gyKey + privateKey
|
|
|
552
|
+ }
|
|
|
553
|
+ for index := range msg.Value {
|
|
|
554
|
+ msg.Value[index] = msg.Value[index] + gyKey + privateKey
|
|
|
555
|
+ }
|
|
|
556
|
+ return msg
|
|
|
557
|
+}
|
|
|
558
|
+
|
|
|
559
|
+func DecryptionMsg(msg PublishMsg, gyKey int64, privateKey int64) {
|
|
|
560
|
+ for index := range msg.Topic {
|
|
|
561
|
+ msg.Topic[index] = msg.Topic[index] - gyKey - privateKey
|
|
|
562
|
+ }
|
|
|
563
|
+ for index := range msg.Value {
|
|
|
564
|
+ msg.Value[index] = msg.Value[index] - gyKey - privateKey
|
|
|
565
|
+ }
|
|
|
566
|
+ for index := range msg.Content {
|
|
|
567
|
+ msg.Content[index] = msg.Content[index] - gyKey - privateKey
|
|
|
568
|
+ }
|
|
|
569
|
+
|
|
|
570
|
+ var runeArr []rune
|
|
|
571
|
+ for index := range msg.Topic {
|
|
|
572
|
+ runeArr = append(runeArr, rune(int(msg.Topic[index])))
|
|
|
573
|
+ }
|
|
|
574
|
+ fmt.Println("Topic is: " + string(runeArr))
|
|
|
575
|
+ runeArr = nil
|
|
|
576
|
+
|
|
|
577
|
+ for index := range msg.Value {
|
|
|
578
|
+ runeArr = append(runeArr, rune(int(msg.Value[index])))
|
|
|
579
|
+ }
|
|
|
580
|
+
|
|
|
581
|
+ if len(runeArr) == 1 {
|
|
|
582
|
+ fmt.Println("Value is: ", runeArr)
|
|
|
583
|
+ } else {
|
|
|
584
|
+ fmt.Println("Value is:", string(runeArr))
|
|
|
585
|
+ }
|
|
|
586
|
+
|
|
|
587
|
+ runeArr = nil
|
|
|
588
|
+
|
|
|
589
|
+ for index := range msg.Content {
|
|
|
590
|
+ runeArr = append(runeArr, rune(int(msg.Content[index])))
|
|
|
591
|
+ }
|
|
|
592
|
+ fmt.Println("Content is: " + string(runeArr))
|
|
|
593
|
+ runeArr = nil
|
|
|
594
|
+
|
|
|
595
|
+}
|
|
|
596
|
+
|
|
|
597
|
+func Listen() {
|
|
|
598
|
+ l, err1 := net.Listen("tcp", ":8150") //MM로 부터 받는거
|
|
|
599
|
+ if err1 != nil {
|
|
|
600
|
+ log.Fatal(fmt.Sprintf("Unable to listen on given port: %s", err1))
|
|
|
601
|
+ }
|
|
|
602
|
+ defer l.Close()
|
|
|
603
|
+
|
|
|
604
|
+ for {
|
|
|
605
|
+ conn, _ := l.Accept()
|
|
|
606
|
+ go rpc.ServeConn(conn)
|
|
|
607
|
+ }
|
|
|
608
|
+}
|
|
|
609
|
+
|
|
|
610
|
+func IntSlice2String(target []int64) string {
|
|
|
611
|
+ var targetString string
|
|
|
612
|
+ targetString = ""
|
|
|
613
|
+ for _, value := range target {
|
|
|
614
|
+ targetString += strconv.FormatInt(int64(value), 10) + " "
|
|
|
615
|
+ }
|
|
|
616
|
+ return targetString
|
|
|
617
|
+}
|