Ejemplo n.º 1
0
//LoadZone 读取ip地址段,并保存到rbtree中
func (ipdisp *IPDisp) LoadZone(conf string) (err error) {
	ipdisp.rbtree = rbtree.NewWith(Comparator)
	var flines []string
	flines, err = file2string(conf)
	if err != nil {
		return
	}
	zoneids := ipdisp.zoneID
	ipdisp.zoneMax = 1
	for _, fline := range flines {
		zone := Zone{}
		ipinfo := strings.Split(fline, ";")
		if len(ipinfo) != 2 {
			continue
		}
		zone.name = ipinfo[1]
		ips := strings.Split(ipinfo[0], "/")
		if len(ips) != 2 {
			continue
		}
		m, err := strconv.Atoi(ips[1])
		if err != nil {
			continue
		}
		zone.ipmin = InetNetwork(ips[0])
		mask := (1 << (32 - uint32(m))) - 1
		zone.ipmax = zone.ipmin | uint32(mask)
		if v, ok := zoneids[zone.name]; ok == true {
			zone.id = v
		} else {
			zoneids[zone.name] = ipdisp.zoneMax
			zone.id = ipdisp.zoneMax
			ipdisp.zoneMax++
		}
		ipdisp.rbtree.Put(zone)
		//fmt.Printf("%v  %v  %v\n",ipzone.ipmin,ipzone.ipmax,ipzone.Zone)
	}
	return
}
Ejemplo n.º 2
0
//LoadNode 加载每个host的node配置
func (ipdisp *IPDisp) LoadNode(conf string, vhostname string) (err error) {
	var flines []string
	flines, err = file2string(conf)
	if err != nil {
		return
	}

	ipdisp.vhosts[vhostname] = &Vhost{}
	vhost := ipdisp.vhosts[vhostname]
	vhost.nodeID = make(map[string]int)
	vhost.defaultNode = 0
	vhost.reqcount = 0
	nodeid := -1
	var cnode *Node
	for _, fline := range flines {
		flen := len(fline)
		if flen < 3 || fline[0] == '#' {
			continue
		}
		if fline[0] == '[' && fline[flen-1] == ']' {
			//获取节点名称,初始化节点配置
			//if nodeID > -1 {
			//	cnode.servers[cnode.servercount-1].next = cnode.servers[0]
			//}
			nodename := string(fline[1 : flen-1])
			cnode = &Node{}
			cnode.name = nodename
			cnode.status = 0
			cnode.servercount = 0
			cnode.balance = 'r'
			cnode.id = nodeid + 1
			cnode.reqmin = 0
			cnode.reqlastmin = 0
			cnode.reqcount = 0
			cnode.freebw = 20
			cnode.sw = make([]int, swMAX)
			cnode.serverID = make(map[string]int)
			cnode.swtree = rbtree.NewWith(Comparator)
			vhost.nodes = append(vhost.nodes, cnode)
			nodeid++
			vhost.nodeID[nodename] = nodeid
		} else {
			if nodeid == -1 {
				continue
			}
			cfline := string(fline)
			cf := strings.Split(cfline, "=")
			if len(cf) != 2 {
				continue
			}
			switch cf[0] {
			case "server":
				server := &Server{}
				server.status = 0
				server.weight = 0
				server.id = 0
				sinfo := strings.Split(cf[1], " ")
				server.ip = sinfo[0]
				sinfolen := len(sinfo)
				switch sinfolen {
				case 2:
					server.id, err = strconv.Atoi(sinfo[1])
				case 3:
					server.id, err = strconv.Atoi(sinfo[1])
					server.weightstr = sinfo[2]
				case 4:
					server.id, err = strconv.Atoi(sinfo[1])
					server.weightstr = sinfo[2]
					server.status = serverstat[sinfo[3]]
				}
				/*
					swcount = swcount + server.weight
				*/
				cnode.servers = append(cnode.servers, server)
				if cnode.servercount == 0 {
					cnode.curserver = server
				} else {
					cnode.servers[cnode.servercount-1].next = server
				}
				cnode.serverID[server.ip] = server.id
				cnode.servercount++
			case "bw":
				cnode.bw, err = strconv.Atoi(cf[1])
			case "maxbw":
				cnode.maxbw, err = strconv.Atoi(cf[1])
			case "freebw":
				cnode.freebw, err = strconv.Atoi(cf[1])
			case "overflow2node":
				cnode.overflow2node = cf[1]
			case "status":
				cnode.status = serverstat[cf[1]]
			case "default":
				vhost.defaultNode = nodeid
			case "balance":
				switch cf[1][0] {
				case 'h':
					cnode.balance = cf[1][0]
				case 'r':
					cnode.balance = cf[1][0]
				case 'w':
					cnode.balance = cf[1][0]
				case 'A':
					cnode.balance = cf[1][0]
				case 'a':
					cnode.balance = cf[1][0]
				default:
					err = errors.New(cnode.name + ": balance config is invalid")
					return
				}
			}
		}
	}
	for _, node := range vhost.nodes {
		var ok bool
		node.overflow2nodeid, ok = vhost.nodeID[node.overflow2node]
		if ok == false {
			node.overflow2nodeid = -1
		}
		//使server数组中,server.next首尾相接
		node.servers[node.servercount-1].next = node.servers[0]
		//添加特定均衡方式:o(only one),代表只有一个server
		if len(node.servers) == 1 {
			node.balance = 'o'
		}
		err = node.initbalance()
	}

	return
}