Create & Init Project...

app/interface/main/creative/service/geetest/BUILD

@@ -0,0 +1,52 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["service_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/creative/conf:go_default_library",
+        "//app/interface/main/creative/model/geetest:go_default_library",
+        "//app/interface/main/creative/service:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["service.go"],
+    importpath = "go-common/app/interface/main/creative/service/geetest",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/main/creative/conf:go_default_library",
+        "//app/interface/main/creative/dao/account:go_default_library",
+        "//app/interface/main/creative/dao/geetest:go_default_library",
+        "//app/interface/main/creative/model/geetest:go_default_library",
+        "//app/interface/main/creative/service:go_default_library",
+        "//library/log:go_default_library",
+        "//library/stat/prom:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/interface/main/creative/service/geetest/service.go

@@ -0,0 +1,127 @@
+package geetest
+
+import (
+	"context"
+	"crypto/md5"
+	"encoding/hex"
+	"math/rand"
+	"strconv"
+
+	"go-common/app/interface/main/creative/conf"
+	"go-common/app/interface/main/creative/dao/account"
+	d "go-common/app/interface/main/creative/dao/geetest"
+	m "go-common/app/interface/main/creative/model/geetest"
+	"go-common/app/interface/main/creative/service"
+	"go-common/library/log"
+	"go-common/library/stat/prom"
+)
+
+//Service struct.
+type Service struct {
+	c    *conf.Config
+	gt   *d.Dao
+	acc  *account.Dao
+	pErr *prom.Prom
+}
+
+//New get service.
+func New(c *conf.Config, rpcdaos *service.RPCDaos) *Service {
+	s := &Service{
+		c:    c,
+		gt:   d.New(c),
+		acc:  rpcdaos.Acc,
+		pErr: prom.BusinessErrCount,
+	}
+	return s
+}
+
+// PreProcessAdd fn
+func (s *Service) PreProcessAdd(c context.Context, mid int64, ip, clientType string, newCaptcha int) (res *m.ProcessRes, err error) {
+	var pre string
+	res = &m.ProcessRes{
+		Limit: map[string]bool{
+			"add": false,
+		},
+	}
+	if exist, _, _ := s.acc.HalfMin(c, mid); exist {
+		log.Info("halfMin exist | mid(%d)", mid)
+		res.Limit["add"] = true
+	}
+	if res.Limit["add"] {
+		res.CaptchaID = s.c.Geetest.CaptchaID
+		res.NewCaptcha = newCaptcha
+		if pre, err = s.gt.PreProcess(c, mid, ip, clientType, newCaptcha); err != nil || pre == "" {
+			randOne := md5.Sum([]byte(strconv.Itoa(rand.Intn(100))))
+			randTwo := md5.Sum([]byte(strconv.Itoa(rand.Intn(100))))
+			challenge := hex.EncodeToString(randOne[:]) + hex.EncodeToString(randTwo[:])[0:2]
+			res.Challenge = challenge
+			log.Info("s.gt.PreProcess err != nil (%s,%d,%d) ", clientType, newCaptcha, mid)
+			s.pErr.Incr("geetest_preprocess")
+			res.Success = 0
+			err = nil
+			return
+		}
+		res.Success = 1
+		slice := md5.Sum([]byte(pre + s.c.Geetest.PrivateKEY))
+		res.Challenge = hex.EncodeToString(slice[:])
+	}
+	return
+}
+
+// PreProcess getGeetestChal
+func (s *Service) PreProcess(c context.Context, mid int64, ip, clientType string, newCaptcha int) (res *m.ProcessRes, err error) {
+	var pre string
+	res = &m.ProcessRes{
+		Limit: map[string]bool{
+			"add": false,
+		},
+	}
+	if exist, _, _ := s.acc.HalfMin(c, mid); exist {
+		log.Info("halfMin exist | mid(%d)", mid)
+		res.Limit["add"] = true
+	}
+	res.CaptchaID = s.c.Geetest.CaptchaID
+	res.NewCaptcha = newCaptcha
+	if pre, err = s.gt.PreProcess(c, mid, ip, clientType, newCaptcha); err != nil || pre == "" {
+		randOne := md5.Sum([]byte(strconv.Itoa(rand.Intn(100))))
+		randTwo := md5.Sum([]byte(strconv.Itoa(rand.Intn(100))))
+		challenge := hex.EncodeToString(randOne[:]) + hex.EncodeToString(randTwo[:])[0:2]
+		res.Challenge = challenge
+		log.Info("s.gt.PreProcess err != nil (%s,%d,%d) ", clientType, newCaptcha, mid)
+		s.pErr.Incr("geetest_preprocess")
+		res.Success = 0
+		err = nil
+		return
+	}
+	res.Success = 1
+	slice := md5.Sum([]byte(pre + s.c.Geetest.PrivateKEY))
+	res.Challenge = hex.EncodeToString(slice[:])
+	return
+}
+
+// Validate recheck the seccode
+func (s *Service) Validate(c context.Context, challenge, validate, seccode, clientType, ip string, success int, mid int64) (stat bool) {
+	if len(validate) != 32 {
+		log.Error("s.Validate(%s,%s,%s,%d) err(validate not eq 32byte)", challenge, validate, seccode, mid)
+		return
+	}
+	if success != 1 {
+		slice := md5.Sum([]byte(challenge))
+		stat = hex.EncodeToString(slice[:]) == validate
+		return
+	}
+	slice := md5.Sum([]byte(s.c.Geetest.PrivateKEY + "geetest" + challenge))
+	if hex.EncodeToString(slice[:]) != validate {
+		log.Error("s.Validate(%s,%s,%s,%d) err(challenge not found)", challenge)
+		return
+	}
+	res, err := s.gt.Validate(c, challenge, seccode, clientType, ip, s.c.Geetest.CaptchaID, mid)
+	if err != nil {
+		s.pErr.Incr("geetest_validate")
+		log.Error("s.Validate(%s,%s,%s,%d) err(gtServer validate failed.)", challenge, validate, seccode, mid)
+		return
+	}
+	slice = md5.Sum([]byte(seccode))
+	stat = hex.EncodeToString(slice[:]) == res.Seccode
+	return
+}

app/interface/main/creative/service/geetest/service_test.go

@@ -0,0 +1,51 @@
+package geetest
+
+import (
+	"context"
+	"flag"
+	"go-common/app/interface/main/creative/conf"
+	m "go-common/app/interface/main/creative/model/geetest"
+	"path/filepath"
+	"testing"
+	"time"
+
+	. "github.com/smartystreets/goconvey/convey"
+	"go-common/app/interface/main/creative/service"
+)
+
+var (
+	s *Service
+)
+
+func init() {
+	dir, _ := filepath.Abs("../../cmd/creative.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	rpcdaos := service.NewRPCDaos(conf.Conf)
+	s = New(conf.Conf, rpcdaos)
+	time.Sleep(time.Second)
+}
+
+func WithService(f func(s *Service)) func() {
+	return func() {
+		Reset(func() {})
+		f(s)
+	}
+}
+
+func Test_PreProcess(t *testing.T) {
+	var (
+		c          = context.TODO()
+		err        error
+		res        *m.ProcessRes
+		mid        = int64(2089809)
+		ip         = "127.0.0.1"
+		clientType = "clientType"
+		newCaptcha = 1
+	)
+	Convey("PreProcess", t, WithService(func(s *Service) {
+		res, err = s.PreProcess(c, mid, ip, clientType, newCaptcha)
+		So(err, ShouldBeNil)
+		So(res, ShouldNotBeNil)
+	}))
+}