Create & Init Project...

app/interface/live/lottery-interface/internal/conf/BUILD

@@ -0,0 +1,42 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["conf.go"],
+    importpath = "go-common/app/interface/live/lottery-interface/internal/conf",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//library/cache/memcache:go_default_library",
+        "//library/cache/redis:go_default_library",
+        "//library/conf:go_default_library",
+        "//library/database/sql:go_default_library",
+        "//library/ecode/tip:go_default_library",
+        "//library/log:go_default_library",
+        "//library/log/infoc:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+        "//library/net/http/blademaster/middleware/verify:go_default_library",
+        "//library/net/rpc/warden:go_default_library",
+        "//library/net/trace:go_default_library",
+        "//vendor/github.com/BurntSushi/toml: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/live/lottery-interface/internal/conf/conf.go

@@ -0,0 +1,93 @@
+package conf
+
+import (
+	"errors"
+	"flag"
+
+	"go-common/library/cache/memcache"
+	"go-common/library/cache/redis"
+	"go-common/library/conf"
+	"go-common/library/database/sql"
+	ecode "go-common/library/ecode/tip"
+	"go-common/library/log"
+	"go-common/library/log/infoc"
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/http/blademaster/middleware/verify"
+	"go-common/library/net/rpc/warden"
+	"go-common/library/net/trace"
+
+	"github.com/BurntSushi/toml"
+)
+
+var (
+	confPath string
+	client   *conf.Client
+	// Conf config
+	Conf = &Config{}
+)
+
+// Config .
+type Config struct {
+	Log         *log.Config
+	BM          *bm.ServerConfig
+	Verify      *verify.Config
+	Tracer      *trace.Config
+	Redis       *redis.Config
+	Memcache    *memcache.Config
+	MySQL       *sql.Config
+	Ecode       *ecode.Config
+	Infoc       *infoc.Config
+	ShortClient *warden.ClientConfig
+	LongClient  *warden.ClientConfig
+}
+
+func init() {
+	flag.StringVar(&confPath, "conf", "", "default config path")
+}
+
+// Init init conf
+func Init() error {
+	if confPath != "" {
+		return local()
+	}
+	return remote()
+}
+
+func local() (err error) {
+	_, err = toml.DecodeFile(confPath, &Conf)
+	return
+}
+
+func remote() (err error) {
+	if client, err = conf.New(); err != nil {
+		return
+	}
+	if err = load(); err != nil {
+		return
+	}
+	go func() {
+		for range client.Event() {
+			log.Info("config reload")
+			if load() != nil {
+				log.Error("config reload error (%v)", err)
+			}
+		}
+	}()
+	return
+}
+
+func load() (err error) {
+	var (
+		s       string
+		ok      bool
+		tmpConf *Config
+	)
+	if s, ok = client.Toml2(); !ok {
+		return errors.New("load config center error")
+	}
+	if _, err = toml.Decode(s, &tmpConf); err != nil {
+		return errors.New("could not decode config")
+	}
+	*Conf = *tmpConf
+	return
+}

app/interface/live/lottery-interface/internal/dao/BUILD

@@ -0,0 +1,34 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["dao.go"],
+    importpath = "go-common/app/interface/live/lottery-interface/internal/dao",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/live/lottery-interface/internal/conf:go_default_library",
+        "//library/cache/memcache:go_default_library",
+        "//library/cache/redis:go_default_library",
+        "//library/database/sql: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/live/lottery-interface/internal/dao/dao.go

@@ -0,0 +1,42 @@
+package dao
+
+import (
+	"context"
+
+	"go-common/app/interface/live/lottery-interface/internal/conf"
+	"go-common/library/cache/memcache"
+	"go-common/library/cache/redis"
+	xsql "go-common/library/database/sql"
+)
+
+// Dao dao
+type Dao struct {
+	c     *conf.Config
+	mc    *memcache.Pool
+	redis *redis.Pool
+	db    *xsql.DB
+}
+
+// New init mysql db
+func New(c *conf.Config) (dao *Dao) {
+	dao = &Dao{
+		c:     c,
+		mc:    memcache.NewPool(c.Memcache),
+		redis: redis.NewPool(c.Redis),
+		db:    xsql.NewMySQL(c.MySQL),
+	}
+	return
+}
+
+// Close close the resource.
+func (d *Dao) Close() {
+	d.mc.Close()
+	d.redis.Close()
+	d.db.Close()
+}
+
+// Ping dao ping
+func (d *Dao) Ping(ctx context.Context) error {
+	// TODO: add mc,redis... if you use
+	return d.db.Ping(ctx)
+}

app/interface/live/lottery-interface/internal/model/BUILD

@@ -0,0 +1,28 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["model.go"],
+    importpath = "go-common/app/interface/live/lottery-interface/internal/model",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)
+
+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/live/lottery-interface/internal/model/model.go

@@ -0,0 +1 @@
+package model

app/interface/live/lottery-interface/internal/server/http/BUILD

@@ -0,0 +1,40 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["http.go"],
+    importpath = "go-common/app/interface/live/lottery-interface/internal/server/http",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/live/lottery-interface/internal/conf:go_default_library",
+        "//app/interface/live/lottery-interface/internal/service:go_default_library",
+        "//app/interface/live/lottery-interface/internal/service/v1:go_default_library",
+        "//app/service/live/live_riskcontrol/api/grpc/v1:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+        "//library/net/http/blademaster/middleware/auth:go_default_library",
+        "//library/net/http/blademaster/middleware/verify:go_default_library",
+        "//library/net/http/blademaster/render:go_default_library",
+        "//library/net/metadata: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/live/lottery-interface/internal/server/http/http.go

@@ -0,0 +1,116 @@
+package http
+
+import (
+	"net/http"
+
+	"go-common/app/interface/live/lottery-interface/internal/conf"
+	"go-common/app/interface/live/lottery-interface/internal/service"
+	v1 "go-common/app/interface/live/lottery-interface/internal/service/v1"
+	risk "go-common/app/service/live/live_riskcontrol/api/grpc/v1"
+	"go-common/library/log"
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/http/blademaster/middleware/auth"
+	"go-common/library/net/http/blademaster/middleware/verify"
+	"go-common/library/net/metadata"
+
+	"go-common/library/net/http/blademaster/render"
+)
+
+var (
+	vfy *verify.Verify
+)
+
+// Init init
+func Init(c *conf.Config) {
+
+	vfy = verify.New(c.Verify)
+	engine := bm.DefaultServer(c.BM)
+	route(engine)
+	if err := engine.Start(); err != nil {
+		log.Error("bm Start error(%v)", err)
+		panic(err)
+	}
+}
+
+func route(e *bm.Engine) {
+	e.Ping(ping)
+	e.Register(register)
+	auth := auth.New(&auth.Config{
+		Identify:    conf.Conf.ShortClient,
+		DisableCSRF: true,
+	})
+
+	g := e.Group("/xlive/lottery-interface")
+	{
+		g.POST("/v1/storm/Join", abortGET, auth.Guest, riskControl, v1.StormJoin)
+		g.GET("/v1/storm/Check", auth.Guest, v1.StormCheck)
+	}
+}
+
+func abortGET(c *bm.Context) {
+	if c.Request.Method == http.MethodGet {
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 0,
+			"msg":  "",
+			"data": []int{},
+		})
+		c.Abort()
+		return
+	}
+}
+func riskControl(c *bm.Context) {
+	mid, ok := c.Get("mid")
+	if !ok {
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 401,
+			"msg":  "未登录",
+			"data": []int{},
+		})
+		c.Abort()
+		return
+	}
+	midi := mid.(int64)
+	header := make(map[string]string)
+	for k := range c.Request.Header {
+		if v := c.Request.Header[k]; len(v) > 0 {
+			header[k] = v[0]
+		}
+
+	}
+	header["Platform"] = c.Request.URL.Query().Get("platform")
+	resp, err := service.ServiceInstance.IsForbiddenClient.GetForbidden(c, &risk.GetForbiddenReq{
+		Uid:    midi,
+		Uri:    c.Request.URL.Path,
+		Ip:     metadata.String(c, metadata.RemoteIP),
+		Method: c.Request.Method,
+		Header: header,
+	})
+	if err != nil {
+		log.Info("#call_IsForbiddenClient_fail %s", err.Error())
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 400,
+			"msg":  "没抢到",
+			"data": []int{},
+		})
+		c.Abort()
+		return
+	}
+	if resp.GetIsForbidden() == risk.GetForbiddenReply_FORBIDDEN {
+		log.Info("#IsForbidden# mid= %d", midi)
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 400,
+			"msg":  "访问被拒绝",
+			"data": []int{},
+		})
+		c.Abort()
+		return
+	}
+}
+
+func ping(ctx *bm.Context) {
+	ctx.String(http.StatusOK, "PONG")
+}
+
+func register(c *bm.Context) {
+	c.JSON(map[string]interface{}{}, nil)
+}

app/interface/live/lottery-interface/internal/service/BUILD

@@ -0,0 +1,37 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["service.go"],
+    importpath = "go-common/app/interface/live/lottery-interface/internal/service",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/live/lottery-interface/internal/conf:go_default_library",
+        "//app/service/live/live_riskcontrol/api/grpc/v1:go_default_library",
+        "//app/service/live/xlottery/api/grpc/v1:go_default_library",
+        "//library/log/infoc:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [
+        ":package-srcs",
+        "//app/interface/live/lottery-interface/internal/service/v1:all-srcs",
+    ],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/interface/live/lottery-interface/internal/service/service.go

@@ -0,0 +1,48 @@
+package service
+
+import (
+	"go-common/app/interface/live/lottery-interface/internal/conf"
+	risk "go-common/app/service/live/live_riskcontrol/api/grpc/v1"
+	storm "go-common/app/service/live/xlottery/api/grpc/v1"
+	"go-common/library/log/infoc"
+)
+
+// Service struct
+type Service struct {
+	c                 *conf.Config
+	Infoc             *infoc.Infoc
+	StormClient       storm.StormClient
+	IsForbiddenClient risk.IsForbiddenClient
+}
+
+// New init
+func New(c *conf.Config) (s *Service) {
+	sc, err := storm.NewClient(c.LongClient)
+	if err != nil {
+		panic(err)
+	}
+	isForbiddenClient, err := risk.NewClient(c.ShortClient)
+	if err != nil {
+		panic(err)
+	}
+	s = &Service{
+		c:                 c,
+		Infoc:             infoc.New(c.Infoc),
+		StormClient:       sc.StormClient,
+		IsForbiddenClient: isForbiddenClient,
+	}
+	return s
+}
+
+// Close Service
+func (s *Service) Close() {
+	s.Infoc.Close()
+}
+
+// ServiceInstance instance
+var ServiceInstance *Service
+
+// Init init
+func Init(c *conf.Config) {
+	ServiceInstance = New(c)
+}

app/interface/live/lottery-interface/internal/service/v1/BUILD

@@ -0,0 +1,37 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["storm.go"],
+    importpath = "go-common/app/interface/live/lottery-interface/internal/service/v1",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/live/lottery-interface/internal/service:go_default_library",
+        "//app/service/live/xlottery/api/grpc/v1:go_default_library",
+        "//library/log:go_default_library",
+        "//library/log/infoc:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+        "//library/net/http/blademaster/render:go_default_library",
+        "//library/net/metadata: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/live/lottery-interface/internal/service/v1/storm.go

@@ -0,0 +1,162 @@
+package v1
+
+import (
+	"go-common/app/interface/live/lottery-interface/internal/service"
+	xlottery "go-common/app/service/live/xlottery/api/grpc/v1"
+	"go-common/library/log"
+	"go-common/library/log/infoc"
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/metadata"
+	"net/http"
+	"time"
+
+	"go-common/library/net/http/blademaster/render"
+)
+
+//StormJoin  StormJoin
+func StormJoin(c *bm.Context) {
+	p := new(xlottery.JoinStormReq)
+	if err := c.Bind(p); err != nil {
+		c.JSON(nil, err)
+		return
+	}
+	// if p.GetRoomid() == 0 {
+	// 	c.Render(http.StatusOK, render.MapJSON{
+	// 		"code": 400,
+	// 		"msg":  "参数错误",
+	// 	})
+	// 	return
+	// }
+	mid, _ := c.Get("mid")
+	platform := c.Request.URL.Query().Get("platform")
+	if platform == "" {
+		platform = "web"
+	}
+	if p.Platform == "" {
+		p.Platform = platform
+	}
+	log.Info("mid:%s platform:%s", mid, platform)
+	p.Mid = mid.(int64)
+
+	//反作弊打点
+	lancer(c, p)
+	resp, err := service.ServiceInstance.StormClient.Join(c, p)
+	if err != nil {
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 400,
+			"msg":  "没抢到",
+			"data": []int{},
+		})
+		return
+	}
+
+	switch resp.GetCode() {
+	case 1005002:
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 429,
+			"msg":  resp.GetMsg(),
+		})
+		return
+	case 1005001:
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 401,
+			"msg":  resp.GetMsg(),
+		})
+		return
+	case 1005003, 1005004, 1005005, 1005016:
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": 400,
+			"msg":  resp.GetMsg(),
+		})
+		return
+	case 0:
+		c.JSON(resp.GetJoin(), nil)
+		return
+	default:
+		c.Render(http.StatusOK, render.MapJSON{
+			"code": int(resp.GetCode()),
+			"msg":  resp.GetMsg(),
+		})
+
+	}
+
+}
+
+// StormCheck   StormCheck 检查
+func StormCheck(c *bm.Context) {
+	p := new(xlottery.CheckStormReq)
+	if err := c.Bind(p); err != nil {
+		c.JSON(nil, err)
+		return
+	}
+	mid, isexit := c.Get("mid")
+	if isexit {
+		p.Uid = mid.(int64)
+		log.Info("StormCheck uid = %s", mid)
+	}
+	resp, err := service.ServiceInstance.StormClient.Check(c, p)
+	if err != nil {
+		c.JSON(nil, err)
+		return
+	}
+	if resp.GetCheck() != nil {
+		c.JSON(resp.GetCheck(), nil)
+		return
+	}
+	c.Render(http.StatusOK, render.MapJSON{
+		"code": 0,
+		"msg":  "",
+		"data": []int{},
+	})
+
+}
+
+var lacnertype = "rhythmic"
+
+func lancer(c *bm.Context, req *xlottery.JoinStormReq) {
+	uid := req.GetMid()
+	roomid := req.GetRoomid()
+	lotteryid := req.GetId()
+	action := "join"
+	ip := metadata.String(c, metadata.RemoteIP)
+	ts := time.Now().Unix()
+	platform := req.GetPlatform()
+	clientver := c.Request.URL.Query().Get("version")
+	buvid := c.Request.Header.Get("Buvid")
+	ua := c.Request.Header.Get("User-Agent")
+	referer := c.Request.Header.Get("Referer")
+	cookie := getCookie(c.Request)
+	abnoormal := 0
+	requesturi := c.Request.RequestURI
+
+	err := service.ServiceInstance.Infoc.Info(
+		uid,
+		roomid,
+		lacnertype,
+		lotteryid,
+		action,
+		ip,
+		ts,
+		platform,
+		clientver,
+		buvid,
+		ua,
+		referer,
+		cookie,
+		abnoormal,
+		requesturi)
+	if err != nil {
+		if err != infoc.ErrFull {
+			log.Error("Infoc.Info err: %s", err.Error())
+		}
+	}
+}
+func getCookie(req *http.Request) string {
+	var cookie string
+	for _, v := range req.Cookies() {
+		if v.Name != "SESSDATA" {
+			cookie = cookie + v.Name + "=" + v.Value + "; "
+		}
+	}
+	return cookie
+}