增加队列推送

go.mod

@@ -10,6 +10,7 @@ require (
 	github.com/google/wire v0.4.0
 	github.com/ichunt2019/go-msgserver v1.0.5
 	github.com/pkg/errors v0.8.1
+	github.com/streadway/amqp v0.0.0-20190827072141-edfb9018d271
 	google.golang.org/genproto v0.0.0-20200402124713-8ff61da6d932
 	google.golang.org/grpc v1.28.1
 )

http/internal/server/http/erp/company.go

@@ -4,8 +4,13 @@ import (
 	"encoding/json"
 	"fmt"
 	bm "github.com/go-kratos/kratos/pkg/net/http/blademaster"
+	"github.com/go-kratos/kratos/pkg/log"
+	"github.com/streadway/amqp"
 	"kaopu-server/http/internal/dao"
+	"kaopu-server/http/internal/server/http/member"
 	"strings"
+	"time"
+	"github.com/go-kratos/kratos/pkg/conf/paladin"
 )
 
 //erp需要获取的公司信息
@@ -38,6 +43,10 @@ type ErpResultStruct struct {
 	Result ErpTianYanChaCompanyInfo `json:"result"`
 }
 
+type Rabitmq struct {
+	Dsn          string          //rabitmq 配置1
+}
+
 
 func GetCompanyInfo(c *bm.Context)  {
 
@@ -63,4 +72,72 @@ func GetCompanyInfo(c *bm.Context)  {
 	erpCompanyInfo.TianyanchaJson = ""
 	c.JSON(erpCompanyInfo,nil)
 	closedb()
+
+	if erpCompanyInfo.ComName == "" {
+		PushCompanyMq(c,companyName)
+	}
+}
+
+//增加公司，并且推送mq
+func PushCompanyMq(c *bm.Context,companyName string)  {
+
+	var(
+		cfg Rabitmq
+		ct paladin.TOML
+	)
+
+	if err := paladin.Get("rabitmq.toml").Unmarshal(&ct); err != nil {
+		log.Error(err.Error())
+	}
+	if err := ct.Get("dsn1").UnmarshalTOML(&cfg); err != nil {
+		log.Error(err.Error())
+	}
+
+
+	db,closedb,err := dao.NewDB()
+
+	currentTime := time.Now().Unix()
+	sqlResult,err := db.Exec(c,member.INSERTCOMPANY,companyName,currentTime,currentTime)
+	lastid,_ := sqlResult.LastInsertId()
+	//如果添加没问题，就继续添加
+	if err != nil {
+		fmt.Println(err)
+	}else{
+		_,err  = db.Exec(c,member.INSERTCOMPANYBASICINFO,lastid,companyName,currentTime)
+		if err != nil {
+			fmt.Println(err)
+		}
+	}
+
+	closedb()
+
+	//链接mq
+	conn, err := amqp.Dial("amqp://"+cfg.Dsn+"/")
+	defer conn.Close()
+
+	//通道
+	ch, err := conn.Channel()
+	defer ch.Close()
+
+	//设置数据
+	q, err := ch.QueueDeclare(
+		"fengkong_static_count", // name
+		true,   // durable
+		false,   // delete when unused
+		false,   // exclusive
+		false,   // no-wait
+		nil,     // arguments
+	)
+
+	body := fmt.Sprintf(`{"com_credits_id":"%d","erp_company_code":""}`,lastid)
+	fmt.Println(body)
+	err = ch.Publish(
+		"fengkong_exchange",     // exchange
+		q.Name, // routing key
+		false,  // mandatory
+		false,  // immediate
+		amqp.Publishing {
+			ContentType: "text/plain",
+			Body:        []byte(body),
+		})
 }
\ No newline at end of file