package message

import (
	"log"

	"github.com/nsqio/go-nsq"
)

type myMessageHandler struct{}

func processMessage(b []byte) error {
	log.Println(string(b))
	return nil
}

// HandleMessage implements the Handler interface.
func (h *myMessageHandler) HandleMessage(m *nsq.Message) error {
	if len(m.Body) == 0 {
		return nil
	}
	err := processMessage(m.Body)
	return err
}

func Message() {
	cf := nsq.NewConfig()

	consumer, err := nsq.NewConsumer("error", "error", cf)
	if err != nil {
		log.Fatal(err)
	}

	// Set the Handler for messages received by this Consumer. Can be called multiple times.
	// See also AddConcurrentHandlers.
	consumer.AddHandler(&myMessageHandler{})

	// Use nsqlookupd to discover nsqd instances.
	// See also ConnectToNSQD, ConnectToNSQDs, ConnectToNSQLookupds.
	err = consumer.ConnectToNSQLookupd("ekb-app01.myway.local:4161")
	if err != nil {
		log.Fatal(err)
	}

}