go-common/app/interface/bbq/wechat/internal/service/service.go
2019-04-22 18:49:16 +08:00

69 lines
1.4 KiB
Go

package service
import (
"context"
"crypto/sha1"
"fmt"
"go-common/app/interface/bbq/wechat/dao"
"go-common/app/interface/bbq/wechat/internal/conf"
"go-common/app/interface/bbq/wechat/internal/model"
"go-common/library/log"
)
// Service struct
type Service struct {
c *conf.Config
dao *dao.Dao
}
// New init
func New(c *conf.Config) (s *Service) {
s = &Service{
c: c,
dao: dao.New(c),
}
return s
}
// Ping Service
func (s *Service) Ping(ctx context.Context) (err error) {
return s.dao.Ping(ctx)
}
// Close Service
func (s *Service) Close() {
s.dao.Close()
}
// TokenGet .
func (s *Service) TokenGet(ctx context.Context, arg *model.TokenReq, url string) (sig string, err error) {
token := ""
if token, err = s.dao.TokenGetLast(ctx); err != nil {
log.Warnv(ctx, log.KV("log", "weixin token get last fail"))
return
}
if token == "" {
if token, err = s.dao.TokenGet(ctx); err != nil {
log.Warnv(ctx, log.KV("log", "weixin token get fail"))
return
}
}
sig = s.TokenCalc(ctx, token, arg.Noncestr, arg.Timestamp, url)
return
}
// TokenCalc 计算签名
func (s *Service) TokenCalc(ctx context.Context, ticket string, noncestr string, timestamp string, url string) (sig string) {
str := fmt.Sprintf("jsapi_ticket=%s&noncestr=%s&timestamp=%s&url=%s", ticket, noncestr, timestamp, url)
hash := sha1.New()
hash.Write([]byte(str))
bs := hash.Sum(nil)
sig = fmt.Sprintf("%x", bs)
return
}