parent
e438abdba1
commit
cda459ae9c
@ -0,0 +1,53 @@
|
||||
package rtc
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"gowebsocket/common"
|
||||
"gowebsocket/controllers"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/qiniu/api.v7/v7/auth"
|
||||
)
|
||||
|
||||
var manager *Manager
|
||||
|
||||
func init() {
|
||||
accessKey := "dzumJl3gfsMSR3fvfABL4e0kDpo6FJmrlcuTu8TF"
|
||||
secretKey := "aqWegU2o8tTCe0JtIVfMDdOjC3-jvuv2eWFvKQOm"
|
||||
fmt.Println("accessKey: secretKey: ", accessKey, secretKey)
|
||||
mac := auth.New(accessKey, secretKey)
|
||||
manager = NewManager(mac)
|
||||
}
|
||||
|
||||
// TODO: 获取直播流地址
|
||||
func GetRoomToken(c *gin.Context) {
|
||||
if o := c.Request.Header.Get("Origin"); o != "" {
|
||||
c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
c.Writer.Header().Set("Access-Control-Allow-Methods", "GET,POST,HEAD,PUT,DELETE,OPTIONS")
|
||||
c.Writer.Header().Set("Access-Control-Expose-Headers", "Server,Range,Content-Length,Content-Range")
|
||||
c.Writer.Header().Set("Access-Control-Allow-Headers", "Origin,Range,Accept-Encoding,Referer,Cache-Control,X-Proxy-Authorization,X-Requested-With,Content-Type")
|
||||
}
|
||||
|
||||
q := c.Request.URL.Query()
|
||||
appID, roomName, userID, Permission := q.Get("app_id"), q.Get("room_name"), q.Get("user_id"), q.Get("permission")
|
||||
|
||||
roomAccess := RoomAccess{
|
||||
AppID: appID,
|
||||
RoomName: roomName,
|
||||
UserID: userID,
|
||||
ExpireAt: time.Now().Unix() + 3600,
|
||||
Permission: Permission,
|
||||
}
|
||||
|
||||
token, _ := manager.GetRoomToken(roomAccess)
|
||||
url := fmt.Sprintf("https://rtc.qiniuapi.com/v3/apps/%v/rooms/%v/auth?user=%v&token=%v", roomAccess.AppID, roomAccess.RoomName, roomAccess.UserID, token)
|
||||
fmt.Println("url: ", url)
|
||||
|
||||
data := gin.H{
|
||||
"token": token,
|
||||
}
|
||||
controllers.Response(c, common.OK, "", data)
|
||||
// url := fmt.Sprintf("https://rtc.qiniuapi.com/v3/apps/%v/rooms/%v/auth?user=%v&token=%v", roomAccess.AppID, roomAccess.RoomName, roomAccess.UserID, token)
|
||||
|
||||
}
|
||||
@ -0,0 +1,140 @@
|
||||
package rtc
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"github.com/qiniu/api.v7/v7/auth"
|
||||
)
|
||||
|
||||
// resInfo is httpresponse infomation
|
||||
type resInfo struct {
|
||||
Code int
|
||||
Err error
|
||||
}
|
||||
|
||||
func newResInfo() resInfo {
|
||||
info := resInfo{}
|
||||
return info
|
||||
}
|
||||
|
||||
func getReqid(src *http.Header) string {
|
||||
for k, v := range *src {
|
||||
K := strings.Title(k)
|
||||
if strings.Contains(K, "Reqid") {
|
||||
return strings.Join(v, ", ")
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func buildURL(path string) string {
|
||||
if strings.Index(path, "/") != 0 {
|
||||
path = "/" + path
|
||||
}
|
||||
return "https://" + RtcHost + path
|
||||
}
|
||||
|
||||
func postReq(httpClient *http.Client, mac *auth.Credentials, url string,
|
||||
reqParam interface{}, ret interface{}) *resInfo {
|
||||
info := newResInfo()
|
||||
var reqData []byte
|
||||
var err error
|
||||
|
||||
switch v := reqParam.(type) {
|
||||
case *string:
|
||||
reqData = []byte(*v)
|
||||
case string:
|
||||
reqData = []byte(v)
|
||||
case *[]byte:
|
||||
reqData = *v
|
||||
case []byte:
|
||||
reqData = v
|
||||
default:
|
||||
reqData, err = json.Marshal(reqParam)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
info.Err = err
|
||||
return &info
|
||||
}
|
||||
req, err := http.NewRequest("POST", url, bytes.NewReader(reqData))
|
||||
if err != nil {
|
||||
info.Err = err
|
||||
return &info
|
||||
}
|
||||
req.Header.Add("Content-Type", "application/json")
|
||||
return callReq(httpClient, req, mac, &info, ret)
|
||||
}
|
||||
|
||||
func getReq(httpClient *http.Client, mac *auth.Credentials, url string, ret interface{}) *resInfo {
|
||||
info := newResInfo()
|
||||
req, err := http.NewRequest("GET", url, nil)
|
||||
if err != nil {
|
||||
info.Err = err
|
||||
return &info
|
||||
}
|
||||
return callReq(httpClient, req, mac, &info, ret)
|
||||
}
|
||||
|
||||
func delReq(httpClient *http.Client, mac *auth.Credentials, url string, ret interface{}) *resInfo {
|
||||
info := newResInfo()
|
||||
req, err := http.NewRequest("DELETE", url, nil)
|
||||
if err != nil {
|
||||
info.Err = err
|
||||
return &info
|
||||
}
|
||||
return callReq(httpClient, req, mac, &info, ret)
|
||||
}
|
||||
|
||||
func callReq(httpClient *http.Client, req *http.Request, mac *auth.Credentials,
|
||||
info *resInfo, ret interface{}) (oinfo *resInfo) {
|
||||
oinfo = info
|
||||
accessToken, err := mac.SignRequestV2(req)
|
||||
if err != nil {
|
||||
info.Err = err
|
||||
return
|
||||
}
|
||||
req.Header.Add("Authorization", "Qiniu "+accessToken)
|
||||
client := httpClient
|
||||
if client == nil {
|
||||
client = http.DefaultClient
|
||||
}
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
info.Err = err
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
info.Code = resp.StatusCode
|
||||
reqid := getReqid(&resp.Header)
|
||||
rebuildErr := func(msg string) error {
|
||||
return fmt.Errorf("Code: %v, Reqid: %v, %v", info.Code, reqid, msg)
|
||||
}
|
||||
|
||||
if resp.ContentLength > 2*1024*1024 {
|
||||
err = rebuildErr(fmt.Sprintf("response is too long. Content-Length: %v", resp.ContentLength))
|
||||
info.Err = err
|
||||
return
|
||||
}
|
||||
resData, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
info.Err = rebuildErr(err.Error())
|
||||
return
|
||||
}
|
||||
if info.Code != 200 {
|
||||
info.Err = rebuildErr(string(resData))
|
||||
return
|
||||
}
|
||||
if ret != nil {
|
||||
err = json.Unmarshal(resData, ret)
|
||||
if err != nil {
|
||||
info.Err = rebuildErr(fmt.Sprintf("err: %v, res: %v", err, resData))
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
Loading…
Reference in new issue