hd/controllers/common/wechat_oauth.go

package common

import (
	"crypto/sha1"
	"encoding/xml"
	"fmt"
	"go.uber.org/atomic"
	"hudongzhuanjia/controllers"
	"hudongzhuanjia/libs/filter"
	"hudongzhuanjia/logger"
	pay_service "hudongzhuanjia/services/pay"
	"hudongzhuanjia/utils/define"
	"os"
	"time"
)

var Now = time.Now()

type WeChatOauthCtl struct {
	controllers.BaseCtl
}

const token = "hd_oauth"

// 验证服务器资源配置
func (t *WeChatOauthCtl) Oauth() {
	signature := t.MustGet("signature")
	timestamp := t.MustGet("timestamp")
	nonce := t.MustGet("nonce")
	echoStr := t.MustGet("echostr")
	tokenStr := fmt.Sprintf("%s%s%s", token, timestamp, nonce)
	s := fmt.Sprintf("%s", sha1.New().Sum([]byte(tokenStr)))
	if s == signature {
		t.OriginResponseWriter.Write([]byte(echoStr))
	}
	t.OriginResponseWriter.Write([]byte("failed"))
}

func (t *WeChatOauthCtl) Checkin() {
	content, _ := t.Get("content")
	path, _ := os.Getwd()
	t.JSON(map[string]interface{}{
		"success": "你好, tommy黄梓健",
		"version": Now,
		"wd":      path,
		"content": filter.Replace(content),
	})
}

//
//func init() {
//	ticker := time.NewTicker(5 * time.Second)
//	go func() {
//		fmt.Println("循环查询订单数据")
//		for range ticker.C {
//			pay_service.BatchQuery()
//		}
//	}()
//
//}

type CDATA struct {
	Text string `xml:",cdata"`
}

type CallbackParam struct {
	XMLName    xml.Name `xml:"xml"`
	ReturnCode CDATA    `xml:"return_code"`
	ReturnMsg  CDATA    `xml:"return_msg"`
}

var counter = new(atomic.Int64)

func (t *WeChatOauthCtl) CallbackOrder() {
	counter.Add(1)
	//搜索支付的order表， 查找到某条记录
	_, err := pay_service.NotifyOrder(t.Request.BODY)

	param := new(CallbackParam)
	if err != nil {
		param.ReturnCode = CDATA{Text: define.CODE_FAIL}
		param.ReturnMsg = CDATA{Text: err.Error()}
	} else {
		param.ReturnCode = CDATA{Text: define.CODE_SUCCESS}
		param.ReturnMsg = CDATA{Text: "OK"}
	}
	xmlRes, _ := xml.Marshal(param)
	if t.Request.OriginRequest.Method == "GET" {
		t.JSON(map[string]interface{}{
			"counter": counter.Load(),
			"time":    time.Now(),
			"xml":     param,
		})
	}
	logger.Error(fmt.Sprintf("微信响应次数: %v", counter.Load()))
	t.XML(xmlRes)
}

func (t *WeChatOauthCtl) CallbackRefund() {
	_, err := pay_service.NotifyRefund(t.Request.BODY)
	param := new(CallbackParam)
	if err != nil {
		logger.Error("", err)
		param.ReturnCode = CDATA{Text: define.CODE_FAIL}
		param.ReturnMsg = CDATA{Text: err.Error()}
	} else {
		param.ReturnCode = CDATA{Text: define.CODE_SUCCESS}
		param.ReturnMsg = CDATA{Text: "OK"}
	}
	xmlRes, _ := xml.Marshal(param)
	if t.Request.OriginRequest.Method == "GET" {
		t.JSON(map[string]interface{}{
			"counter": counter.Load(),
			"time":    time.Now(),
			"xml":     param,
		})
	}
	logger.Error(fmt.Sprintf("微信响应次数: %v", counter.Load()))
	t.XML(xmlRes)
}

//
//func (t *WeChatOauthCtl) SendRedPack() {
//	user := new(models.User)
//	exist, err := models.GetById(user, 234)
//	t.CheckErr(err)
//	if !exist {
//		t.ERROR(fmt.Sprintf("error occur: %s", err), 120)
//	}
//	res, err := pay_service.SendRedPack("欧轩互动", user.Openid, "测试", "123.207.246.51",
//		"测试", "测试", 100, 1, 1, user.Id, 1)
//	t.CheckErr(err)
//	fmt.Printf("%+v", res)
//	t.JSON(map[string]interface{}{
//		"result": res,
//	})
//}