package discovery

import (
	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/gopherdiscovery.git"
	"fmt"
	"time"
	"webserver/extend/logger"
	"github.com/gogo/protobuf/proto"
)

const (
	Discovery_Server = "tcp://0.0.0.0:40008"
	Discovery_UrlPubSub = "tcp://0.0.0.0:50008"
)

//启动discovery的server
var discoveryServer *gopherdiscovery.DiscoveryServer
var dbChangeChan chan *protomsg.DbChangeMessage
func StartServer() {
	var clients []string
	var err error
	var (
		defaultOpts = gopherdiscovery.Options{
			SurveyTime:   3 * time.Second,
			//RecvDeadline: 3 * time.Second,
			PollTime:     5 * time.Second,
		}
	)

	discoveryServer, err = gopherdiscovery.Server(Discovery_Server, Discovery_UrlPubSub, defaultOpts)

	logger.Debug("server: ", discoveryServer)
	logger.Debug("err:",err)
	logger.Debug("clients: ",clients)

	dbChangeChan = make(chan *protomsg.DbChangeMessage)

	for {
		select {
		case dbMsg := <-dbChangeChan:
			publishMessage(dbMsg)
		}
	}
}

//广播数据库改变的msg
func publishMessage(msg *protomsg.DbChangeMessage) {
	sendBytes,err := proto.Marshal(msg)
	if err ==nil{
		discoveryServer.PublishMsg(string(sendBytes))
	}
}

func AddDbMessage(msg *protomsg.DbChangeMessage) {
	fmt.Println("MSG In")
	dbChangeChan<-msg
}