[revise] subscription.go

src/broker/modules/subscription.go

@@ -2,41 +2,55 @@ package modules
 
 import (
 	"errors"
+	"fmt"
 )
 
 type sub_manager struct {
 	// NameList의 노드들이 각각의 list를 갖음
-	list NameList
+	list topicList
 
 	/* sub# 관리 */
-	count_sub 	 int              // Subscription #
-	emptylist  []int            // To administrate sub #
-	sub2ip     map[int]string  // sub2ip[sub#] = ip
-	ip2sub     map[string][]int // ip2sub[ip]= sub#...
-	sub2node   map[int][]*Node  // sub2node[sub#] = node_addr
-	israngesub map[int]bool     // delete시에 관리를 위해
+	count_sub 	 int               // Subscription #
+	emptylist  []int                // To administrate sub #
+	ip2sub     map[string][]int     // ip2sub[ip]= sub#...
+	sub2node   map[int][]*valueNode // sub2node[sub#] = node_addr
+	israngesub map[int]bool         // delete시에 관리를 위해
+}
+
+// 초기화 시에 하고 싶은 처리
+func (self *sub_manager) Initialize() {
+	// Some initialize
+	self.ip2sub = make(map[string][]int)
+	self.sub2node = make(map[int][]*valueNode)
+	self.israngesub = make(map[int]bool)
+}
+
+// 생성자 함수를 정의
+func newSubmng() *sub_manager {
+	sub_mng := &sub_manager{}
+	sub_mng.Initialize()
+	return sub_mng
 }
 
 //	### To Insert sub#
-func (manager *sub_manager) add_subscription(msg MsgUnit) error {
+func (manager *sub_manager) addSubscription(msg MsgUnit) error {
 
 	topic := msg.(*SubscriptionMsg).topic
 	value := msg.(*SubscriptionMsg).value
 	operator := msg.(*SubscriptionMsg).operator
+	isAlpha := msg.(*SubscriptionMsg).isAlpha
 	subnumber := 0
 
 	//fmt.Println("message = ", msg)
 
-	// * 1. 들어온 Msg -> sub#, sub# -> Msg로 매핑
+	// * 1. 들어온 ipaddr -> sub# 매핑
 	if len(manager.emptylist) == 0 {
-		manager.sub2ip[manager.count_sub] = msg.(*SubscriptionMsg).from
 		manager.ip2sub[msg.(*SubscriptionMsg).from] = append(manager.ip2sub[msg.(*SubscriptionMsg).from], manager.count_sub)
 		subnumber = manager.count_sub
 		manager.count_sub++
 	} else {
 		subidx := manager.emptylist[len(manager.emptylist)-1]
 		manager.emptylist = manager.emptylist[:len(manager.emptylist)-1]
-		manager.sub2ip[subidx] = msg.(*SubscriptionMsg).from
 		manager.ip2sub[msg.(*SubscriptionMsg).from] = append(manager.ip2sub[msg.(*SubscriptionMsg).from], subidx)
 		subnumber = subidx
 	}
@@ -48,7 +62,7 @@ func (manager *sub_manager) add_subscription(msg MsgUnit) error {
 	//namelist에서 name찾고, 없으면 추가
 	for nameptr != nil {
 		// *  compare함수 구현되기 전 임시로 끼워넣은 것 (이 부분은 secure.go 완성되면 다시 확인)
-		if compare(nameptr.topic, topic) == 0 {
+		if nameptr.topic[0] == topic[0] {
 			findOk = true
 			break
 		}
@@ -56,25 +70,23 @@ func (manager *sub_manager) add_subscription(msg MsgUnit) error {
 	}
 
 	if !findOk {
-		newNode := &NameNode{topic, nil, nil, List{}}
-		manager.list.tail.next = newNode
-		manager.list.tail = newNode
-		manager.list.size++
+		manager.list.addTopicNode(topic)
 		nameptr = manager.list.tail
 	}
 
 	// list[name]에 value추가
 	if len(operator) == 1 { // 단일 식이라면 (single)
-		valptr := nameptr.list.getPos(value)
+		valptr := nameptr.list.getValueNodePos(value, isAlpha)
+		fmt.Println("pos = " ,valptr.val)
 		if valptr == nil {
-			nameptr.list.add_ValueNode(value)
+			nameptr.list.addValueNode(value)
 			valptr = nameptr.list.tail
 		}
 
 		manager.sub2node[subnumber] = append(manager.sub2node[subnumber], valptr)
-		valptr.insert_Sub(operator[0], subnumber, true)
+		valptr.insertSub(operator[0], subnumber, true)
 
-		return nil // add_subscription ok
+		return nil // addSubscription ok
 	} else {
 
 		// '&&' , '||'로 묶여 있는 복합식에 대하여
@@ -82,15 +94,15 @@ func (manager *sub_manager) add_subscription(msg MsgUnit) error {
 
 		// find(namelist[name].list.val == value)인 노드
 		// * 이 부분 다시 봐야함..
-		valptr1 := nameptr.list.getPos(value)
-		valptr2 := nameptr.list.getPos(value)
+		valptr1 := nameptr.list.getValueNodePos(value, isAlpha)
+		valptr2 := nameptr.list.getValueNodePos(value, isAlpha)
 
 		if valptr1 == nil {
-			nameptr.list.add_ValueNode(value)
+			nameptr.list.addValueNode(value)
 			valptr1 = nameptr.list.tail
 		}
 		if valptr2 == nil {
-			nameptr.list.add_ValueNode(value)
+			nameptr.list.addValueNode(value)
 			valptr2 = nameptr.list.tail
 		}
 
@@ -100,17 +112,17 @@ func (manager *sub_manager) add_subscription(msg MsgUnit) error {
 		if logical_operator == "&&" { // '&&'로 묶여있다면
 			// rangesub check
 			manager.israngesub[subnumber] = true
-			valptr1.insert_Sub(operator[0], subnumber, false)
-			valptr2.insert_Sub(operator[2], subnumber, false)
+			valptr1.insertSub(operator[0], subnumber, false)
+			valptr2.insertSub(operator[2], subnumber, false)
 
 		} else {
 			// '||'로 묶여 있다면 두 식 모두 single로 처리가능
-			valptr1.insert_Sub(operator[0], subnumber, true)
-			valptr2.insert_Sub(operator[2], subnumber, true)
+			valptr1.insertSub(operator[0], subnumber, true)
+			valptr2.insertSub(operator[2], subnumber, true)
 		}
-		return nil // add_subscription ok
+		return nil // addSubscription ok
 	}
-	return errors.New("Can't add_subscription")
+	return errors.New("Can't addSubscription")
 }
 
 //	* To delete subscription
@@ -151,7 +163,7 @@ func (manager *sub_manager) delete(from string) error {
 				}
 
 				// node가 비어있으면 delete
-				isempty := node[j].isempty()
+				isempty := node[j].isEmpty()
 				if isempty && node[j] != nil {
 					prev_node := node[j].prev
 					next_node := node[j].next
@@ -192,7 +204,7 @@ func (manager *sub_manager) delete(from string) error {
 					manager.emptylist = append(manager.emptylist, sub)
 				}
 
-				isempty := node[j].isempty()
+				isempty := node[j].isEmpty()
 				// node가 비어있으면 delete
 				if isempty && node[j] != nil {
 					prev_node := node[j].prev
@@ -206,9 +218,4 @@ func (manager *sub_manager) delete(from string) error {
 	}
 	manager.ip2sub[ip]=nil
 	return nil
-}
-
-// * 암호화된 두 value 비교함수 (임시)
-func compare(v1 []int64, v2 []int64) int {
-	return 1
-}
+}