message/handler/handler.go

package handler

import (
	"encoding/json"
	"fmt"
	"time"

	"app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/go-logger/logger"
	. "app.yhyue.com/moapp/message/db"
	"app.yhyue.com/moapp/message/model"
	"github.com/nsqio/go-nsq"
)

var (
	funcMap = map[string]func(msg *model.Message){
		"jyapp_wx_register":      VarPoints.Jy_user_new,
		"jyapp_phone_register":   VarPoints.Jy_user_new,
		"jypc_phone_register":    VarPoints.Jy_user_new,
		"jywx_subscribe_new":     VarPoints.Jy_user_new,
		"jywx_subscribe_invite":  VarPoints.Jywx_subscribe_invite,
		"jywx_subscribe_invited": VarPoints.Jywx_subscribe_invited,
		"jydocs_doc_open":        VarPoints.Jydocs_doc_open,
		"jyweb_article_open":     VarPoints.Jyweb_article_open,
		"jywx_activity_message":  VarPush.Jywx_activity_message,
	}
)

type Handler struct {
}

//
func (h *Handler) HandleMessage(m *nsq.Message) error {
	defer common.Catch()
	if len(m.Body) == 0 {
		// Returning nil will automatically send a FIN command to NSQ to mark the message as processed.
		return nil
	}
	var msg *model.Message
	err := json.Unmarshal(m.Body, &msg)
	if err != nil {
		logger.Error(err)
		return err
	}
	logger.Info("接收到消息", fmt.Sprintf("%+v", msg))
	if msg.E_code == "" {
		logger.Info("缺少参数e_code", fmt.Sprintf("%+v", msg))
		return nil
	} else if msg.E_time == 0 {
		logger.Info("缺少参数e_time", fmt.Sprintf("%+v", msg))
		return nil
	}
	f, f_ok := funcMap[msg.E_code]
	if f_ok {
		f(msg)
	} else {
		logger.Info("无效的code值", msg)
		return nil
	}
	Mgo_Log.Save("nsq_logs", map[string]interface{}{
		"createtime": time.Now().Unix(),
		"body":       msg,
		"type":       "consumer", //producer or consumer
	})
	// Returning a non-nil error will automatically send a REQ command to NSQ to re-queue the message.
	return nil
}