[refactor] subscption.constructor

src/broker/modules/subscription_test.go

@@ -1,291 +1,288 @@
 package modules
 
-//
-//import (
-//	"fmt"
-//	_ "fmt"
-//	"github.com/stretchr/testify/assert"
-//	"math/rand"
-//	"strconv"
-//	"testing"
-//	"Time"
-//)
-//
-//func makeData(isAlpha bool) MsgUnit{
-//	rand.Seed(Time.Now().UnixNano())
-//	// Set Ipaddr
-//	msg := Message{"", "1.0", "", SM}
-//	for i := 0; i < 4; i++{
-//		itoa := strconv.Itoa(rand.Int() % 256)
-//		msg.From += itoa
-//		if i != 3{
-//			msg.From += "."
-//		}
-//	}
-//
-//	// Set Time
-//	msg.Time += strconv.Itoa(rand.Int()%24) + ":"
-//	msg.Time += strconv.Itoa(rand.Int()%60)
-//
-//	// Set Topic, Value, operator
-//	Topic := []int64{}
-//	Value := []int64{}
-//	operator := []string{}
-//	candOp := []string{">", ">=", "<=" ,"<", "=="}
-//	logicalOp := []string{"&&", "||"}
-//
-//	if isAlpha {
-//		topicLen := rand.Int() % 10 + 1
-//		for i := 0 ; i < topicLen; i++{
-//			Topic = append(Topic, rand.Int63())
-//		}
-//
-//		valueLen := rand.Int() % 10 + 1
-//		for i := 0 ; i < valueLen; i++{
-//			Value = append(Value, rand.Int63())
-//		}
-//
-//		operator = append(operator, "==")
-//
-//	} else{
-//		valueLen := rand.Int() % 2 + 1
-//
-//		Topic = append(Topic, rand.Int63())
-//		Value = append(Value, rand.Int63())
-//
-//		if valueLen == 1{
-//			operator = append(operator, candOp[rand.Int()%5])
-//		} else{
-//			Value = append(Value, rand.Int63())
-//			op := rand.Int()%2
-//			operator = append(operator, candOp[rand.Int()%2])
-//			if op == 0 {
-//				operator = append(operator, logicalOp[0])
-//			} else{
-//				operator = append(operator, logicalOp[1])
-//			}
-//			operator = append(operator, candOp[rand.Int()%2 + 2])
-//		}
-//	}
-//
-//	return &SubscriptionMsg{msg, Topic, Value, operator, isAlpha}
-//}
-//
-//func makeMsgList(dataLen int, isAlpha bool) []MsgUnit{
-//	rand.Seed(Time.Now().UnixNano())
-//	var ret []MsgUnit
-//	for i := 0 ; i < dataLen ; i++{
-//		ret = append(ret, makeData(isAlpha))
-//	}
-//	return ret
-//}
-//
-//func checkOperatorList(isSingle bool, sub int, operator string, l *valueNode) int{
-//	ret := -1
-//	if isSingle{
-//		switch operator {
-//		case "<":
-//			ret = findSub(l.single2sub_s, sub)
-//		case "<=":
-//			ret = findSub(l.single2sub_es, sub)
-//		case ">":
-//			ret = findSub(l.single2sub_b, sub)
-//		case ">=":
-//			ret = findSub(l.single2sub_eb, sub)
-//		case "==":
-//			ret = findSub(l.single2sub_e, sub)
-//		}
-//
-//	} else{
-//		switch operator {
-//		case "<":
-//			ret = findSub(l.range2sub_s, sub)
-//		case "<=":
-//			ret = findSub(l.range2sub_es, sub)
-//		case ">":
-//			ret = findSub(l.range2sub_b, sub)
-//		case ">=":
-//			ret = findSub(l.range2sub_eb, sub)
-//		}
-//	}
-//	return ret
-//}
-//
-//func watchData(msgList []MsgUnit, dataLen int){
-//	for i := 0; i < dataLen; i++{
-//		msg := msgList[i]
-//		fmt.Println(
-//			"\nFrom = ", msg.(*SubscriptionMsg).Message.From,
-//			"\nTime = ", msg.(*SubscriptionMsg).Message.Time,
-//			"\nTopic = ", msg.(*SubscriptionMsg).Topic,
-//			"\nValue = ", msg.(*SubscriptionMsg).Value,
-//			"\noperator = ", msg.(*SubscriptionMsg).operator,
-//			"\nisAlpha ?= ", msg.(*SubscriptionMsg).isAlpha,
-//		)
-//	}
-//}
-//
-////Test addSubScription(1) (dif all [Topic, Value, operator])
-//func Test_addSubscription_allDif(t *testing.T) {
-//	rand.Seed(Time.Now().UnixNano())
-//
-//	// To Init sub_mng
-//	s := sub_manager{
-//		ip2sub:     make(map[string][]int),
-//		sub2node:   make(map[int][]*valueNode),
-//		israngesub: make(map[int]bool),
-//	}
-//	mos := Moscato{sub_mng: s}
-//
-//	// Make Data set(Subscription)
-//	var msgList []MsgUnit
-//	dataLen := 100
-//	msgList = makeMsgList(dataLen, false)
-//
-//	//Watch Data set
-//	//watchData(msgList, dataLen)
-//
-//	for i := 0; i < dataLen; i++ {
-//		msg := msgList[i]
-//		ip := msg.(*SubscriptionMsg).Message.From
-//		Topic := msg.(*SubscriptionMsg).Topic
-//		Value := msg.(*SubscriptionMsg).Value
-//		operator := msg.(*SubscriptionMsg).operator
-//		subnumber := mos.sub_mng.count_sub
-//		isSingle := true
-//
-//		// 0. Check addSubscription
-//		err := mos.sub_mng.addSubscription(msg)
-//		assert.Equal(t, nil, err)
-//
-//		// 1. Check if ip mapping is correct
-//		assert.Equal(t, subnumber, mos.sub_mng.ip2sub[ip][len(mos.sub_mng.ip2sub[ip])-1], "Ip mapping is failed")
-//
-//		// 2. Check topicNode
-//		topicPtr := mos.sub_mng.list.head
-//		for topicPtr != nil {
-//			if Compare(Topic, topicPtr.Topic) == 0 {
-//				break
-//			}
-//			topicPtr = topicPtr.next
-//		}
-//
-//		assert.Equal(t, Topic, topicPtr.Topic, "topicNode Add is failed")
-//
-//		// Check isSingle ?
-//		if len(operator) == 3 && operator[1] == "&&" {
-//			isSingle = false
-//		}
-//
-//		// 3. Check Value in ValueNode & Check Operator in ValueNode
-//		if !isSingle || (len(operator) == 3 && operator[1] == "||") {
-//			valptr1 := topicPtr.list.getValueNodePos([]int64{Value[0]})
-//			valptr2 := topicPtr.list.getValueNodePos([]int64{Value[1]})
-//
-//			assert.Equal(t, []int64{Value[0]}, valptr1.val)
-//			assert.Equal(t, []int64{Value[1]}, valptr2.val)
-//
-//			assert.NotEqual(t, -1, checkOperatorList(isSingle, subnumber, operator[0], valptr1))
-//			assert.NotEqual(t, -1, checkOperatorList(isSingle, subnumber, operator[2], valptr2))
-//
-//		} else {
-//			valptr := topicPtr.list.getValueNodePos(Value)
-//			assert.Equal(t, Value, valptr.val)
-//			assert.NotEqual(t, -1, checkOperatorList(isSingle, subnumber, operator[0], valptr))
-//		}
-//	}
-//}
-//
-//// Test addSubScription(2) (same [Topic, Value] dif [operator])
-//func Test_addSubscription_same_topicNvalue(t *testing.T) {
-//	rand.Seed(Time.Now().UnixNano())
-//
-//	// To Init sub_mng
-//	s := sub_manager{
-//		ip2sub:     make(map[string][]int),
-//		sub2node:   make(map[int][]*valueNode),
-//		israngesub: make(map[int]bool),
-//	}
-//
-//	mos := Moscato{sub_mng: s}
-//
-//	// Fix Topic & Value
-//	topicLen := rand.Int()%10 + 1
-//	staticTopic := make([]int64, topicLen)
-//	staticValue := []int64{rand.Int63()}
-//
-//
-//	// Make Data
-//	var msgList []MsgUnit
-//	dataLen := 100
-//	msgList = makeMsgList(dataLen, false)
-//
-//	for i := 0; i < topicLen; i++ {
-//		staticTopic[i] = rand.Int63()
-//	}
-//
-//	// Fix Same Topic & Value
-//	for i := 0; i < dataLen; i++ {
-//		operator := msgList[i].(*SubscriptionMsg).operator
-//		msgList[i].(*SubscriptionMsg).Topic = staticTopic
-//
-//		if len(operator) == 1{
-//			msgList[i].(*SubscriptionMsg).Value = staticValue
-//		}
-//	}
-//
-//	// Watch Data set
-//	// watchData(msgList, dataLen)
-//
-//	for i := 0; i < dataLen; i++ {
-//		msg := msgList[i]
-//		err := mos.sub_mng.addSubscription(msg)
-//		assert.Equal(t, nil, err)
-//	}
-//
-//	topicPtr := mos.sub_mng.list.head
-//	for topicPtr != nil {
-//		if Compare(topicPtr.Topic, staticTopic) == 0 {
-//			break
-//		}
-//		topicPtr = topicPtr.next
-//	}
-//	watchValueNode(topicPtr.list.head)
-//
-//}
-//
-//func watchValueNode(ptr *valueNode) {
-//	valPtr := ptr
-//	for valPtr != nil{
-//		fmt.Println("Value = ", valPtr.val)
-//		if len(valPtr.single2sub_s) != 0 {
-//			fmt.Println("Single2sub (<) List = ", valPtr.single2sub_s)
-//		}
-//		if len(valPtr.single2sub_es) != 0 {
-//			fmt.Println("Single2sub (<=) List = ", valPtr.single2sub_es)
-//		}
-//		if len(valPtr.single2sub_b) != 0 {
-//			fmt.Println("Single2sub (>) List = ", valPtr.single2sub_b)
-//		}
-//		if len(valPtr.single2sub_eb) != 0 {
-//			fmt.Println("Single2sub (>=) List = ", valPtr.single2sub_eb)
-//		}
-//		if len(valPtr.single2sub_e) != 0 {
-//			fmt.Println("Single2sub (==) List = ", valPtr.single2sub_e)
-//		}
-//
-//		if len(valPtr.range2sub_s) != 0 {
-//			fmt.Println("range2sub (<) List = ", valPtr.range2sub_s)
-//		}
-//		if len(valPtr.range2sub_es) != 0 {
-//			fmt.Println("range2sub (<=) List = ", valPtr.range2sub_es)
-//		}
-//		if len(valPtr.range2sub_b) != 0 {
-//			fmt.Println("range2sub (>) List = ", valPtr.range2sub_b)
-//		}
-//		if len(valPtr.range2sub_eb) != 0 {
-//			fmt.Println("range2sub (>=) List = ", valPtr.range2sub_eb)
-//		}
-//		valPtr = valPtr.next
-//	}
-//}
+import (
+	"fmt"
+	_ "fmt"
+	"github.com/stretchr/testify/assert"
+	"math/rand"
+	"strconv"
+	"testing"
+	"time"
+)
+
+func makeData(isAlpha bool) MsgUnit{
+	rand.Seed(time.Now().UnixNano())
+	// Set Ipaddr
+	msg := Message{"", "1.0", "", SM}
+	for i := 0; i < 4; i++{
+		itoa := strconv.Itoa(rand.Int() % 256)
+		msg.from += itoa
+		if i != 3{
+			msg.from += "."
+		}
+	}
+
+	// Set Time
+	msg.time += strconv.Itoa(rand.Int()%24) + ":"
+	msg.time += strconv.Itoa(rand.Int()%60)
+
+	// Set Topic, Value, operator
+	topic := []int64{}
+	value := []int64{}
+	operator := []string{}
+	candOp := []string{">", ">=", "<=" ,"<", "=="}
+	logicalOp := []string{"&&", "||"}
+
+	if isAlpha {
+		topicLen := rand.Int() % 10 + 1
+		for i := 0 ; i < topicLen; i++{
+			topic = append(topic, rand.Int63())
+		}
+
+		valueLen := rand.Int() % 10 + 1
+		for i := 0 ; i < valueLen; i++{
+			value = append(value, rand.Int63())
+		}
+
+		operator = append(operator, "==")
+
+	} else{
+		valueLen := rand.Int() % 2 + 1
+
+		topic = append(topic, rand.Int63())
+		value = append(value, rand.Int63())
+
+		if valueLen == 1{
+			operator = append(operator, candOp[rand.Int()%5])
+		} else{
+			value = append(value, rand.Int63())
+			op := rand.Int()%2
+			operator = append(operator, candOp[rand.Int()%2])
+			if op == 0 {
+				operator = append(operator, logicalOp[0])
+			} else{
+				operator = append(operator, logicalOp[1])
+			}
+			operator = append(operator, candOp[rand.Int()%2 + 2])
+		}
+	}
+
+	return &SubscriptionMsg{msg, topic, value, operator, isAlpha}
+}
+
+func makeMsgList(dataLen int, isAlpha bool) []MsgUnit{
+	rand.Seed(time.Now().UnixNano())
+	var ret []MsgUnit
+	for i := 0 ; i < dataLen ; i++{
+		ret = append(ret, makeData(isAlpha))
+	}
+	return ret
+}
+
+func checkOperatorList(isSingle bool, sub int, operator string, l *valueNode) int{
+	ret := -1
+	if isSingle{
+		switch operator {
+		case "<":
+			ret = findSub(l.single2sub_s, sub)
+		case "<=":
+			ret = findSub(l.single2sub_es, sub)
+		case ">":
+			ret = findSub(l.single2sub_b, sub)
+		case ">=":
+			ret = findSub(l.single2sub_eb, sub)
+		case "==":
+			ret = findSub(l.single2sub_e, sub)
+		}
+
+	} else{
+		switch operator {
+		case "<":
+			ret = findSub(l.range2sub_s, sub)
+		case "<=":
+			ret = findSub(l.range2sub_es, sub)
+		case ">":
+			ret = findSub(l.range2sub_b, sub)
+		case ">=":
+			ret = findSub(l.range2sub_eb, sub)
+		}
+	}
+	return ret
+}
+
+func watchData(msgList []MsgUnit, dataLen int, isSubscription bool){
+	for i := 0; i < dataLen; i++{
+		msg := msgList[i]
+		if isSubscription {
+			fmt.Println(
+				"\nfrom = ", msg.(*SubscriptionMsg).Message.from,
+				"\ntime = ", msg.(*SubscriptionMsg).Message.time,
+				"\ntopic = ", msg.(*SubscriptionMsg).topic,
+				"\nvalue = ", msg.(*SubscriptionMsg).value,
+				"\noperator = ", msg.(*SubscriptionMsg).operator,
+				"\nisAlpha ?= ", msg.(*SubscriptionMsg).isAlpha,
+			)
+		} else{
+			fmt.Println(
+				"\nfrom = ", msg.(*PublishMsg).Message.from,
+				"\ntime = ", msg.(*PublishMsg).Message.time,
+				"\ntopic = ", msg.(*PublishMsg).topic,
+				"\nvalue = ", msg.(*PublishMsg).value,
+			)
+		}
+	}
+}
+
+//Test addSubScription(1) (dif all [topic, value, operator])
+func Test_addSubscription_allDif(t *testing.T) {
+	rand.Seed(time.Now().UnixNano())
+
+	// To Init sub_mng
+	mos := Moscato{sub_mng: *newSubmng()}
+
+	// Make Data set(Subscription)
+	var msgList []MsgUnit
+	dataLen := 100
+	msgList = makeMsgList(dataLen, false)
+
+	//Watch Data set
+	//watchData(msgList, dataLen, true)
+
+	for i := 0; i < dataLen; i++ {
+		msg := msgList[i]
+		ip := msg.(*SubscriptionMsg).Message.from
+		topic := msg.(*SubscriptionMsg).topic
+		value := msg.(*SubscriptionMsg).value
+		operator := msg.(*SubscriptionMsg).operator
+		subnumber := mos.sub_mng.count_sub
+		isSingle := true
+
+		// 0. Check addSubscription
+		err := mos.sub_mng.addSubscription(msg)
+		assert.Equal(t, nil, err)
+
+		// 1. Check if ip mapping is correct
+		assert.Equal(t, subnumber, mos.sub_mng.ip2sub[ip][len(mos.sub_mng.ip2sub[ip])-1], "Ip mapping is failed")
+
+		// 2. Check topicNode
+		topicPtr := mos.sub_mng.list.head
+		for topicPtr != nil {
+			if Compare(topic, topicPtr.topic) == 0 {
+				break
+			}
+			topicPtr = topicPtr.next
+		}
+
+		assert.Equal(t, topic, topicPtr.topic, "topicNode Add is failed")
+
+		// Check isSingle ?
+		if len(operator) == 3 && operator[1] == "&&" {
+			isSingle = false
+		}
+
+		// 3. Check Value in ValueNode & Check Operator in ValueNode
+		if !isSingle || (len(operator) == 3 && operator[1] == "||") {
+			valptr1 := topicPtr.list.getValueNodePos([]int64{value[0]})
+			valptr2 := topicPtr.list.getValueNodePos([]int64{value[1]})
+
+			assert.Equal(t, []int64{value[0]}, valptr1.val)
+			assert.Equal(t, []int64{value[1]}, valptr2.val)
+
+			assert.NotEqual(t, -1, checkOperatorList(isSingle, subnumber, operator[0], valptr1))
+			assert.NotEqual(t, -1, checkOperatorList(isSingle, subnumber, operator[2], valptr2))
+
+		} else {
+			valptr := topicPtr.list.getValueNodePos(value)
+			assert.Equal(t, value, valptr.val)
+			assert.NotEqual(t, -1, checkOperatorList(isSingle, subnumber, operator[0], valptr))
+		}
+	}
+}
+
+// Test addSubScription(2) (same [topic, value] dif [operator])
+func Test_addSubscription_same_topicNvalue(t *testing.T) {
+	rand.Seed(time.Now().UnixNano())
+
+	// To Init sub_mng
+	mos := Moscato{sub_mng: *newSubmng()}
+
+	// Fix Topic & Value
+	topicLen := rand.Int()%10 + 1
+	staticTopic := make([]int64, topicLen)
+	staticValue := []int64{rand.Int63()}
+
+
+	// Make Data
+	var msgList []MsgUnit
+	dataLen := 100
+	msgList = makeMsgList(dataLen, false)
+
+	for i := 0; i < topicLen; i++ {
+		staticTopic[i] = rand.Int63()
+	}
+
+	// Fix Same Topic & Value
+	for i := 0; i < dataLen; i++ {
+		operator := msgList[i].(*SubscriptionMsg).operator
+		msgList[i].(*SubscriptionMsg).topic = staticTopic
+
+		if len(operator) == 1{
+			msgList[i].(*SubscriptionMsg).value = staticValue
+		}
+	}
+
+	// Watch Data set
+	//watchData(msgList, dataLen, true)
+
+	for i := 0; i < dataLen; i++ {
+		msg := msgList[i]
+		err := mos.sub_mng.addSubscription(msg)
+		assert.Equal(t, nil, err)
+	}
+
+	topicPtr := mos.sub_mng.list.head
+	for topicPtr != nil {
+		if Compare(topicPtr.topic, staticTopic) == 0 {
+			break
+		}
+		topicPtr = topicPtr.next
+	}
+	watchValueNode(topicPtr.list.head)
+
+}
+
+func watchValueNode(ptr *valueNode) {
+	valPtr := ptr
+	for valPtr != nil{
+		fmt.Println("Value = ", valPtr.val)
+		if len(valPtr.single2sub_s) != 0 {
+			fmt.Println("Single2sub (<) List = ", valPtr.single2sub_s)
+		}
+		if len(valPtr.single2sub_es) != 0 {
+			fmt.Println("Single2sub (<=) List = ", valPtr.single2sub_es)
+		}
+		if len(valPtr.single2sub_b) != 0 {
+			fmt.Println("Single2sub (>) List = ", valPtr.single2sub_b)
+		}
+		if len(valPtr.single2sub_eb) != 0 {
+			fmt.Println("Single2sub (>=) List = ", valPtr.single2sub_eb)
+		}
+		if len(valPtr.single2sub_e) != 0 {
+			fmt.Println("Single2sub (==) List = ", valPtr.single2sub_e)
+		}
+
+		if len(valPtr.range2sub_s) != 0 {
+			fmt.Println("range2sub (<) List = ", valPtr.range2sub_s)
+		}
+		if len(valPtr.range2sub_es) != 0 {
+			fmt.Println("range2sub (<=) List = ", valPtr.range2sub_es)
+		}
+		if len(valPtr.range2sub_b) != 0 {
+			fmt.Println("range2sub (>) List = ", valPtr.range2sub_b)
+		}
+		if len(valPtr.range2sub_eb) != 0 {
+			fmt.Println("range2sub (>=) List = ", valPtr.range2sub_eb)
+		}
+		valPtr = valPtr.next
+	}
+}