Initial commit

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

@@ -0,0 +1,41 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "dm.go",
+        "service.go",
+    ],
+    importpath = "go-common/app/interface/live/open-interface/internal/service",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/live/open-interface/api/http/v1:go_default_library",
+        "//app/interface/live/open-interface/internal/dao:go_default_library",
+        "//app/service/live/broadcast-proxy/api/v1:go_default_library",
+        "//app/service/live/resource/sdk:go_default_library",
+        "//library/conf/paladin:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/log: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/open-interface/internal/service/dm.go

@@ -0,0 +1,101 @@
+package service
+
+import (
+	"context"
+	"crypto/md5"
+	"encoding/json"
+	"fmt"
+	"math"
+	"strconv"
+	"time"
+
+	dmv1pb "go-common/app/interface/live/open-interface/api/http/v1"
+	"go-common/app/interface/live/open-interface/internal/dao"
+	broadcasrtService "go-common/app/service/live/broadcast-proxy/api/v1"
+	titansSdk "go-common/app/service/live/resource/sdk"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	"go-common/library/net/metadata"
+)
+
+func checksign(ts string, sign string, group string) bool {
+	sts := time.Now().Unix()
+	cts, _ := strconv.ParseInt(ts, 10, 64)
+	dv := sts - cts
+	if math.Abs(float64(dv)) > 5 {
+		log.Info("[dm] ts  err: %+d", dv)
+		return false
+	}
+
+	userSecrets, terr := titansSdk.Get("dmUser")
+	if terr != nil {
+		log.Error("[dm] get titan conf err: %+v", terr)
+		return false
+	}
+	juserSecrets := make(map[string]string)
+	if jerr := json.Unmarshal([]byte(userSecrets), &juserSecrets); jerr != nil {
+		log.Error("[dm] JSON decode titan conf err: %+v", terr)
+		return false
+	}
+
+	secret, ok := juserSecrets[group]
+	if !ok {
+		log.Info("[dm] unknow  group: %+v", group)
+		return false
+	}
+
+	newSign := fmt.Sprintf("%x", md5.Sum([]byte(group+secret+ts)))
+	if newSign != sign {
+		log.Info("[dm] check sign err sign: %+s service sign: %s", sign, newSign)
+		return false
+	}
+	return true
+}
+
+//Sendmsg 发送弹幕消息
+func (s *Service) Sendmsg(ctx context.Context, req *dmv1pb.SendMsgReq) (resp *dmv1pb.SendMsgResp, err error) {
+	if ok := checksign(req.GetTs(), req.GetSign(), req.GetGroup()); !ok {
+		return nil, ecode.Error(-403, "sign or ts error")
+	}
+
+	var dmString = "{\"cmd\":\"DANMU_MSG\",\"info\":[[0,1,25,16777215,%d,0,0,\"\",0,0,0],\"%s\",[0,\"\",0,0,0,0,0,\"\"],[],[0,0,0,\"\"],[\"\",\"\"],0,0]}"
+
+	breq := &broadcasrtService.RoomMessageRequest{
+		RoomId:  int32(req.GetRoomID()),
+		Message: fmt.Sprintf(dmString, time.Now().Unix(), req.GetMsg()),
+	}
+	_, berr := dao.BcastClient.DanmakuClient.RoomMessage(ctx, breq)
+	if berr != nil {
+		log.Error("[dm] SendBroadCastGrpc err: %+v", berr)
+		return nil, ecode.Error(-400, "send msg err")
+	}
+	return
+}
+
+//GetConf 获取弹幕配置
+func (s *Service) GetConf(ctx context.Context, req *dmv1pb.GetConfReq) (resp *dmv1pb.GetConfResp, err error) {
+	if ok := checksign(req.GetTs(), req.GetSign(), req.GetGroup()); !ok {
+		return nil, ecode.Error(-403, "sign or ts error")
+	}
+
+	resp = &dmv1pb.GetConfResp{
+		WSPort:     []int64{2244},
+		WSSPort:    []int64{443},
+		TCPPort:    []int64{2243, 80},
+		DomianList: []string{},
+	}
+
+	breq := &broadcasrtService.DispatchRequest{
+		UserIp: metadata.String(ctx, metadata.RemoteIP),
+	}
+	bresp, berr := dao.BcastClient.Dispatch(ctx, breq)
+	if berr != nil {
+		resp.IPList = []string{"broadcastlv.chat.bilibili.com"}
+		resp.DomianList = []string{"broadcastlv.chat.bilibili.com"}
+		log.Error("[dm] get IPList by BcastClient Dispatch err:%+v", berr)
+	} else {
+		resp.IPList = bresp.Ip
+		resp.DomianList = bresp.Host
+	}
+	return resp, nil
+}

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

@@ -0,0 +1,50 @@
+package service
+
+import (
+	"context"
+
+	"go-common/app/interface/live/open-interface/internal/dao"
+	titansSdk "go-common/app/service/live/resource/sdk"
+	"go-common/library/conf/paladin"
+)
+
+// Service service.
+type Service struct {
+	ac  *paladin.Map
+	dao *dao.Dao
+}
+
+// New new a service and return.
+func New() (s *Service) {
+	var ac = new(paladin.TOML)
+	if err := paladin.Watch("application.toml", ac); err != nil {
+		panic(err)
+	}
+	s = &Service{
+		ac:  ac,
+		dao: dao.New(),
+	}
+
+	dao.InitGrpc()
+	InitTitan()
+	return s
+}
+
+//InitTitan 初始化kv配置
+func InitTitan() {
+	conf := &titansSdk.Config{
+		TreeId: 82686,
+		Expire: 1,
+	}
+	titansSdk.Init(conf)
+}
+
+// Ping ping the resource.
+func (s *Service) Ping(ctx context.Context) (err error) {
+	return s.dao.Ping(ctx)
+}
+
+// Close close the resource.
+func (s *Service) Close() {
+	s.dao.Close()
+}