Esempio n. 1
0
func (da *DemoApp) AddInstanceListener(event controller.Event) {
	if event.Type() != controller.AddInstanceEventType {
		panic("")
	}

	e := event.(*controller.AddInstanceEvent)
	for _, r := range da.Controller.Routers {
		instance := e.Instance

		port, ok := ServicePortMap[instance.Service]
		if !ok {
			log.Println("Unknown port for service:", instance.Service)
			continue
		}
		// TODO: debugging, delete it
		if r.Id() == "rr" {
			port = ":8081"
		}

		err := r.AddService(instance.Service, port, rIdPolicyMap[r.Id()])
		if err != nil {
			log.Println(err)
		}

		err = r.AddServiceInstance(instance.Service, instance)
		if err != nil {
			log.Println(err)
		}
	}
}
Esempio n. 2
0
func (da *DemoApp) FailureInstanceListener(event controller.Event) {
	if event.Type() != controller.FailureInstanceEventType {
		panic("")
	}

	e := event.(*controller.FailureInstanceEvent)

	log.Printf("Failure Instance: '%v', '%v'\n", e.Reporter, e.Instance)

	list := da.failureRecord[e.Instance.Name]
	for _, reporter := range list {
		if reporter == e.Reporter {
			log.Println(reporter, "has reported failure of", e.Instance.Name, "before")
			return
		}
	}

	list = append(list, e.Reporter)
	da.failureRecord[e.Instance.Name] = list

	if len(list) == len(da.Controller.Routers) {
		for _, r := range da.Controller.Routers {
			instance := e.Instance

			err := r.RemoveServiceInstance(instance.Service, instance)
			if err != nil {
				log.Println(err)
			}
		}
	}
}
Esempio n. 3
0
// AddInstanceListener makes the Load Balancer know of the newly added service instance.
// It adds the service instances to the routers it knew.
// TODO: It assumes the new instance is newly created: which means no other router had it before.
func (lb *LoadBalancer) AddInstanceListener(event controller.Event) {
	if event.Type() != controller.AddInstanceEventType {
		panic("")
	}

	e := event.(*controller.AddInstanceEvent)
	for _, r := range lb.Controller.Routers {
		instance := e.Instance

		port, ok := ServicePortMap[instance.Service]
		if !ok {
			log.Println("Unknown port for service:", instance.Service)
			continue
		}

		err := r.AddService(instance.Service, port, router.NewRoundRobinPolicy())
		if err != nil {
			log.Println(err)
		}

		err = r.AddServiceInstance(instance.Service, instance)
		if err != nil {
			log.Println(err)
		}
	}
}
Esempio n. 4
0
// AddRouterListener makes the Load Balancer know of the newly added router.
// It adds known service instances to the router and set service policy as round robin.
// TODO: It assumes the new router is newly created: which means it has no services.
func (lb *LoadBalancer) AddRouterListener(event controller.Event) {
	if event.Type() != controller.AddRouterEventType {
		panic("")
	}

	e := event.(*controller.AddRouterEvent)

	r := lb.Controller.Routers[e.Id]

	for service, instances := range lb.Controller.ServiceInstances {
		port, ok := ServicePortMap[service]
		if !ok {
			log.Println("Unknown port for service:", service)
			continue
		}
		err := r.AddService(service, port, router.NewRoundRobinPolicy())
		if err != nil {
			log.Println(err)
		}

		for _, instance := range instances {
			err := r.AddServiceInstance(service, instance)
			if err != nil {
				log.Println(err)
			}
		}
	}
}
Esempio n. 5
0
func (lb *LoadBalancer) FailureInstanceListener(event controller.Event) {
	if event.Type() != controller.FailureInstanceEventType {
		panic("")
	}

	e := event.(*controller.FailureInstanceEvent)

	log.Printf("Failure Instance: '%v', '%v'", e.Reporter, e.Instance)
}
Esempio n. 6
0
func (da *DemoApp) RmInstanceListener(event controller.Event) {
	if event.Type() != controller.RmInstanceEventType {
		panic("")
	}

	e := event.(*controller.RmInstanceEvent)

	for _, r := range da.Controller.Routers {
		instance := e.Instance

		err := r.RemoveServiceInstance(instance.Service, instance)
		if err != nil {
			log.Println(err)
		}
	}
}