package eureka

import (
	"context"
	"gitee.com/tylf2018/go-micro-framework/registry"
)

type watcher struct {
	ctx        context.Context
	cancel     context.CancelFunc
	watchChan  chan struct{}
	serverName string
	discovery  *EurekaDiscovery
}

func newWatch(ctx context.Context, discovery *EurekaDiscovery, serverName string) (*watcher, error) {
	w := &watcher{
		ctx:        ctx,
		serverName: serverName,
		discovery:  discovery,
		watchChan:  make(chan struct{}, 1),
	}
	w.ctx, w.cancel = context.WithCancel(ctx)
	e := w.discovery.Subscribe(
		serverName,
		func() {
			w.watchChan <- struct{}{}
		},
	)
	return w, e
}
func (w watcher) Next() (services []*registry.ServiceInstance, err error) {
	select {
	case <-w.ctx.Done():
		return nil, w.ctx.Err()
	case <-w.watchChan:
		instances := w.discovery.GetServiceInstances(w.ctx, w.serverName)
		services = make([]*registry.ServiceInstance, 0, len(instances))
		for _, instance := range instances {
			services = append(services, &registry.ServiceInstance{
				ID:        instance.Metadata["ID"],
				Name:      instance.Metadata["Name"],
				Version:   instance.Metadata["Version"],
				Endpoints: []string{instance.Metadata["Endpoints"]},
				Metadata:  instance.Metadata,
			})
		}
		return
	}
}

func (w *watcher) Stop() error {
	w.cancel()
	w.discovery.Unsubscribe(w.serverName)
	return nil
}