main.go

package main

import (
	"crm-server/configs"
	"crm-server/internal/common"
	"crm-server/internal/logic"
	"crm-server/internal/model"
	"crm-server/internal/service"
	"encoding/json"
	"fmt"
	"github.com/streadway/amqp"
	"log"
	"strconv"
	"time"
)

var(
	InsertData model.MemberAddUserData
)

type WaitAddUser struct {
	User_Id 	int `json:"user_id"`
}


//监听用户添加队列
func main(){

	//定义队列类型和错误类型
	service.ProGramErrType = "member_user_add"
	//设置所有城市
	common.SetCityName()

	conn, err := amqp.Dial("amqp://"+configs.RABBITMQDSN+"/")
	if err != nil {
		service.WriteErrDetail(err.Error() + "Failed to connect to RabbitMQ")
	}
	defer conn.Close()

	ch, err := conn.Channel()
	if err != nil {
		service.WriteErrDetail(err.Error() + "Failed to open a channel")
	}
	defer ch.Close()

	q, err := ch.QueueDeclare(
		service.ProGramErrType, // name
		true,   // durable
		false,   // delete when unused
		false,   // exclusive
		false,   // no-wait
		nil,     // arguments
	)
	if err != nil {
		service.WriteErrDetail(err.Error()+"Failed to declare a queue")
	}

	msgs, err := ch.Consume(
		q.Name, // queue
		"",     // consumer
		true,   // auto-ack
		false,  // exclusive
		false,  // no-local
		false,  // no-wait
		nil,    // args
	)
	if err != nil {
		service.WriteErrDetail(err.Error() +"Failed to register a consumer")
	}

	forever := make(chan bool)
	var user WaitAddUser

	go func() {
		for d := range msgs {
			time.Sleep(5*time.Second)
			json.Unmarshal(d.Body,&user)
			fmt.Println(user)
			handle(user.User_Id)
		}
	}()

	log.Printf(" [*] Waiting for messages. To exit press CTRL+C")
	<-forever
}



func handle(memberId int)(result bool) {
	time.Sleep(5*time.Second)
	//获取当前会员的基本数据
	InsertData = logic.GetMemberUserInfo(memberId)

	//检测用户是否存在
	if InsertData.User_Id == 0 {
		service.WriteErrDetail(strconv.Itoa(memberId) + "会员系统中用户信息不存在")
		return false
	}
	if logic.CheckMemberIdIsHave(InsertData.User_Id) != 0 {
		service.WriteErrDetail(strconv.Itoa(memberId) + "crm系统中用户已经存在")
		return false
	}

	//处理得到的用户数据
	InsertData = logic.HandleData(InsertData)

	//插入数据
	if !logic.InsertMemberUser(InsertData) {
		errSourcedata, _ := json.Marshal(InsertData)
		service.WriteErrDetail("插入用户数据错误：" + string(errSourcedata))
		return false
	}
	return true
}