Create & Init Project...

app/interface/live/web-room/service/v1/dm/BUILD

@@ -0,0 +1,44 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "dM.go",
+        "data.go",
+        "validate.go",
+    ],
+    importpath = "go-common/app/interface/live/web-room/service/v1/dm",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/live/web-room/api/http/v1:go_default_library",
+        "//app/interface/live/web-room/conf:go_default_library",
+        "//app/interface/live/web-room/model:go_default_library",
+        "//app/service/live/live-dm/api/grpc/v1:go_default_library",
+        "//app/service/live/live_riskcontrol/api/grpc/v1:go_default_library",
+        "//app/service/live/xcaptcha/api/grpc/v1:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/http/blademaster: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/web-room/service/v1/dm/dM.go

@@ -0,0 +1,136 @@
+package v1
+
+import (
+	v1pb "go-common/app/interface/live/web-room/api/http/v1"
+	"go-common/app/interface/live/web-room/conf"
+	dmrpc "go-common/app/service/live/live-dm/api/grpc/v1"
+	risk "go-common/app/service/live/live_riskcontrol/api/grpc/v1"
+	xcaptcha "go-common/app/service/live/xcaptcha/api/grpc/v1"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/metadata"
+	"strings"
+)
+
+// DMService struct
+type DMService struct {
+	conf *conf.Config
+}
+
+var (
+	//dmClient 弹幕clietn
+	dmClient dmrpc.DMClient
+	//riskClient 风控Client
+	riskClient risk.IsForbiddenClient
+	//验证码client
+	xcaptchaClient xcaptcha.XCaptchaClient
+)
+
+//NewDMService init
+func NewDMService(c *conf.Config) (s *DMService) {
+	s = &DMService{
+		conf: c,
+	}
+
+	var err error
+	if dmClient, err = dmrpc.NewClient(c.DM); err != nil {
+		panic(err)
+	}
+
+	if riskClient, err = risk.NewClient(c.DM); err != nil {
+		panic(err)
+	}
+
+	if xcaptchaClient, err = xcaptcha.NewClient(c.VerifyConf); err != nil {
+		panic(err)
+	}
+	return s
+}
+
+func sendMsg(ctx *bm.Context, req *v1pb.SendDMReq, uid int64) (resp *v1pb.SendMsgResp, err error) {
+	resp = &v1pb.SendMsgResp{}
+	ip := metadata.String(ctx, metadata.RemoteIP)
+
+	var ck = make([]string, 0, 10)
+	for _, v := range ctx.Request.Cookies() {
+		if v.Name != "SESSDATA" {
+			ck = append(ck, v.Name+"="+v.Value)
+		}
+	}
+
+	var dmReq = &dmrpc.SendMsgReq{
+		Uid:      uid,
+		Roomid:   req.Roomid,
+		Msg:      req.Msg,
+		Rnd:      req.Rnd,
+		Ip:       ip,
+		Fontsize: req.Fontsize,
+		Mode:     req.Mode,
+		Platform: "web",
+		Msgtype:  0,
+		Bubble:   req.Bubble,
+		Lancer: &dmrpc.Lancer{
+			Build:     0,
+			Buvid:     "",
+			Refer:     ctx.Request.Header.Get("Referer"),
+			UserAgent: ctx.Request.Header.Get("User-Agent"),
+			Cookie:    strings.Join(ck, ";"),
+		},
+	}
+	gresp, gerr := dmClient.SendMsg(ctx, dmReq)
+	if gerr != nil {
+		log.Error("DM GRPC ERR: %v", gerr)
+		err = ecode.Error(1003218, "系统正在维护中,请稍后尝试")
+		return nil, err
+	}
+	if gresp.IsLimit {
+		err = ecode.Error(ecode.Code(gresp.Code), gresp.LimitMsg)
+		return nil, err
+	}
+	return resp, nil
+}
+
+// SendMsg implementation
+// `method:"POST"`
+func (s *DMService) SendMsg(ctx *bm.Context, req *v1pb.SendDMReq) (resp *v1pb.SendMsgResp, err error) {
+	uid, ok := ctx.Get("mid")
+	if !ok {
+		err = ecode.Error(1003218, "未登录")
+		return nil, err
+	}
+
+	//验证码
+	if req.Anti != "" {
+		result := checkVerify(ctx, req.Anti, uid.(int64), req.Roomid)
+		if !result {
+			return nil, ecode.Error(1990001, "验证码验证失败")
+		}
+		return sendMsg(ctx, req, uid.(int64))
+	}
+
+	//风控校验
+	ifb, ferr := isriskcontrol(ctx, uid.(int64), req)
+	if ifb {
+		return nil, ferr
+	}
+	//发送弹幕
+	return sendMsg(ctx, req, uid.(int64))
+}
+
+// GetHistory implementation
+// `method:"POST"`
+func (s *DMService) GetHistory(ctx *bm.Context, req *v1pb.HistoryReq) (resp *v1pb.HistoryResp, err error) {
+	resp = &v1pb.HistoryResp{}
+	var hreq = &dmrpc.HistoryReq{
+		Roomid: req.Roomid,
+	}
+	gresp, err := dmClient.GetHistory(ctx, hreq)
+	if err != nil {
+		log.Error("DM GRPC ERR: %v", err)
+		return
+	}
+	resp.Admin = gresp.Admin
+	resp.Room = gresp.Room
+	return
+}

app/interface/live/web-room/service/v1/dm/data.go

@@ -0,0 +1,32 @@
+package v1
+
+import (
+	"encoding/json"
+	v1pb "go-common/app/interface/live/web-room/api/http/v1"
+	"go-common/app/interface/live/web-room/model"
+)
+
+//HistoryData 历史数据处理
+func HistoryData(data *v1pb.HistoryResp) map[string][]*model.History {
+	var result = make(map[string][]*model.History)
+	result["admin"] = make([]*model.History, 0, 10)
+	result["room"] = make([]*model.History, 0, 10)
+
+	for i := 0; i < len(data.Admin); i++ {
+		var h = &model.History{}
+		err := json.Unmarshal([]byte(data.Admin[i]), h)
+		if err != nil {
+			break
+		}
+		result["admin"] = append(result["admin"], h)
+	}
+	for i := 0; i < len(data.Room); i++ {
+		var h = &model.History{}
+		err := json.Unmarshal([]byte(data.Room[i]), h)
+		if err != nil {
+			break
+		}
+		result["room"] = append(result["room"], h)
+	}
+	return result
+}

app/interface/live/web-room/service/v1/dm/validate.go

@@ -0,0 +1,79 @@
+package v1
+
+import (
+	"encoding/json"
+
+	v1pb "go-common/app/interface/live/web-room/api/http/v1"
+	risk "go-common/app/service/live/live_riskcontrol/api/grpc/v1"
+	xcaptcha "go-common/app/service/live/xcaptcha/api/grpc/v1"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/metadata"
+)
+
+type rbody struct {
+	Roomid   int64  `json:"roomid"`
+	Msg      string `json:"msg" `
+	Rnd      string `json:"rnd" `
+	Fontsize int64  `json:"fontsize"`
+	Mode     int64  `json:"mode" `
+	Color    int64  `json:"color"`
+	Bubble   int64  `json:"bubble"`
+}
+
+func isriskcontrol(ctx *bm.Context, uid int64, r *v1pb.SendDMReq) (forbid bool, err *ecode.Status) {
+	req := &risk.GetForbiddenReq{
+		Uid:    uid,
+		Uri:    "/xlive/app-room/v1/dM/sendmsg",
+		Ip:     metadata.String(ctx, metadata.RemoteIP),
+		Method: "POST",
+		Header: make(map[string]string),
+	}
+
+	for k := range ctx.Request.Header {
+		req.Header[k] = ctx.Request.Header.Get(k)
+	}
+	rb := &rbody{
+		Roomid:   r.Roomid,
+		Msg:      r.Msg,
+		Rnd:      r.Rnd,
+		Fontsize: r.Fontsize,
+		Mode:     r.Mode,
+		Color:    r.Color,
+		Bubble:   r.Bubble,
+	}
+	jb, _ := json.Marshal(rb)
+	req.Body = string(jb)
+
+	resp, rerr := riskClient.GetForbidden(ctx, req)
+	if rerr != nil {
+		log.Error("DM: riskcontrol err:%+v", rerr)
+		return false, nil
+	}
+
+	switch resp.IsForbidden {
+	case 0:
+		return false, nil
+	case 1:
+		return true, ecode.Error(400, "访问被拒绝")
+	case 2:
+		return true, ecode.Error(1990000, "need a second time verify")
+	}
+	return false, nil
+}
+
+func checkVerify(ctx *bm.Context, anti string, uid int64, roomid int64) bool {
+	req := &xcaptcha.XVerifyReq{
+		Uid:      uid,
+		ClientIp: metadata.String(ctx, metadata.RemoteIP),
+		XAnti:    anti,
+		RoomId:   roomid,
+	}
+
+	if _, err := xcaptchaClient.Verify(ctx, req); err != nil {
+		return false
+	}
+
+	return true
+}