Skip to content

Commit

Permalink
Json api responses
Browse files Browse the repository at this point in the history
  • Loading branch information
@GNURub
GNURub committed Apr 8, 2020
1 parent 170a1fa commit 24928da
Showing 1 changed file with 109 additions and 48 deletions.
157 changes: 109 additions & 48 deletions protocol/httpopera/http_opera.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@ package httpopera
import (
"encoding/json"
"fmt"
"io"
"log"
"net"
"net/http"
Expand All @@ -18,15 +17,15 @@ import (
)

type Response struct {
w http.ResponseWriter
Status int `json:"status"`
Message string `json:"message"`
w http.ResponseWriter
Status int `json:"status"`
Data interface{} `json:"data"`
}

func (r *Response) SendJson() (int, error) {
resp, _ := json.Marshal(r)
r.w.WriteHeader(r.Status)
r.w.Header().Set("Content-Type", "application/json")
r.w.WriteHeader(r.Status)
return r.w.Write(resp)
}

Expand Down Expand Up @@ -135,9 +134,18 @@ type streams struct {

//http://127.0.0.1:8090/stat/livestat
func (server *Server) GetLiveStatics(w http.ResponseWriter, req *http.Request) {
res := &Response{
w: w,
Data: nil,
Status: 200,
}

defer res.SendJson()

rtmpStream := server.handler.(*rtmp.RtmpStream)
if rtmpStream == nil {
io.WriteString(w, "<h1>Get rtmp stream information error</h1>")
res.Status = 500
res.Data = "Get rtmp stream information error"
return
}

Expand Down Expand Up @@ -172,18 +180,27 @@ func (server *Server) GetLiveStatics(w http.ResponseWriter, req *http.Request) {
}
}
}

resp, _ := json.Marshal(msgs)
w.Header().Set("Content-Type", "application/json")
w.Write(resp)
res.Data = resp
}

//http://127.0.0.1:8090/control/pull?&oper=start&app=live&name=123456&url=rtmp://192.168.16.136/live/123456
func (s *Server) handlePull(w http.ResponseWriter, req *http.Request) {
var retString string
var err error

res := &Response{
w: w,
Data: nil,
Status: 200,
}

defer res.SendJson()

if req.ParseForm() != nil {
fmt.Fprintf(w, "url: /control/pull?&oper=start&app=live&name=123456&url=rtmp://192.168.16.136/live/123456")
res.Status = 400
res.Data = "url: /control/pull?&oper=start&app=live&name=123456&url=rtmp://192.168.16.136/live/123456"
return
}

Expand All @@ -194,7 +211,8 @@ func (s *Server) handlePull(w http.ResponseWriter, req *http.Request) {

log.Printf("control pull: oper=%v, app=%v, name=%v, url=%v", oper, app, name, url)
if (len(app) <= 0) || (len(name) <= 0) || (len(url) <= 0) {
io.WriteString(w, "control push parameter error, please check them.</br>")
res.Status = 400
res.Data = "control push parameter error, please check them."
return
}

Expand All @@ -207,15 +225,17 @@ func (s *Server) handlePull(w http.ResponseWriter, req *http.Request) {

if !found {
retString = fmt.Sprintf("session key[%s] not exist, please check it again.", keyString)
io.WriteString(w, retString)
res.Status = 400
res.Data = retString
return
}
log.Printf("rtmprelay stop push %s from %s", remoteurl, localurl)
pullRtmprelay.Stop()

delete(s.session, keyString)
retString = fmt.Sprintf("<h1>push url stop %s ok</h1></br>", url[0])
io.WriteString(w, retString)
res.Status = 400
res.Data = retString
log.Printf("pull stop return %s", retString)
} else {
pullRtmprelay := rtmprelay.NewRtmpRelay(&localurl, &remoteurl)
Expand All @@ -227,7 +247,8 @@ func (s *Server) handlePull(w http.ResponseWriter, req *http.Request) {
s.session[keyString] = pullRtmprelay
retString = fmt.Sprintf("<h1>push url start %s ok</h1></br>", url[0])
}
io.WriteString(w, retString)
res.Status = 400
res.Data = retString
log.Printf("pull start return %s", retString)
}
}
Expand All @@ -237,8 +258,16 @@ func (s *Server) handlePush(w http.ResponseWriter, req *http.Request) {
var retString string
var err error

res := &Response{
w: w,
Data: nil,
Status: 200,
}

defer res.SendJson()

if req.ParseForm() != nil {
fmt.Fprintf(w, "url: /control/push?&oper=start&app=live&name=123456&url=rtmp://192.168.16.136/live/123456")
res.Data = "url: /control/push?&oper=start&app=live&name=123456&url=rtmp://192.168.16.136/live/123456"
return
}

Expand All @@ -249,7 +278,7 @@ func (s *Server) handlePush(w http.ResponseWriter, req *http.Request) {

log.Printf("control push: oper=%v, app=%v, name=%v, url=%v", oper, app, name, url)
if (len(app) <= 0) || (len(name) <= 0) || (len(url) <= 0) {
io.WriteString(w, "control push parameter error, please check them.</br>")
res.Data = "control push parameter error, please check them."
return
}

Expand All @@ -261,15 +290,15 @@ func (s *Server) handlePush(w http.ResponseWriter, req *http.Request) {
pushRtmprelay, found := s.session[keyString]
if !found {
retString = fmt.Sprintf("<h1>session key[%s] not exist, please check it again.</h1>", keyString)
io.WriteString(w, retString)
res.Data = retString
return
}
log.Printf("rtmprelay stop push %s from %s", remoteurl, localurl)
pushRtmprelay.Stop()

delete(s.session, keyString)
retString = fmt.Sprintf("<h1>push url stop %s ok</h1></br>", url[0])
io.WriteString(w, retString)
res.Data = retString
log.Printf("push stop return %s", retString)
} else {
pushRtmprelay := rtmprelay.NewRtmpRelay(&localurl, &remoteurl)
Expand All @@ -282,69 +311,101 @@ func (s *Server) handlePush(w http.ResponseWriter, req *http.Request) {
s.session[keyString] = pushRtmprelay
}

io.WriteString(w, retString)
res.Data = retString
log.Printf("push start return %s", retString)
}
}

//http://127.0.0.1:8090/control/reset?room=ROOM_NAME
func (s *Server) handleReset(w http.ResponseWriter, r *http.Request) {
if r.ParseForm() != nil {
fmt.Fprintf(w, "url: /control/reset?room=ROOM_NAME")
res := &Response{
w: w,
Data: nil,
Status: 200,
}
defer res.SendJson()

if err := r.ParseForm(); err != nil {
res.Status = 400
res.Data = "url: /control/reset?room=ROOM_NAME"
return
}
room := r.Form.Get("room")

if len(room) == 0 {
res.Status = 400
res.Data = "url: /control/get?room=ROOM_NAME"
return
}
room := r.Form["room"][0]

status := 200
msg, err := configure.RoomKeys.SetKey(room)

if err != nil {
msg = err.Error()
status = 400
res.Status = 400
}

res := &Response{
w: w,
Message: msg,
Status: status,
}
res.SendJson()
res.Data = msg
}

//http://127.0.0.1:8090/control/get?room=ROOM_NAME
func (s *Server) handleGet(w http.ResponseWriter, r *http.Request) {
if r.ParseForm() != nil {
fmt.Fprintf(w, "url: /control/get?room=ROOM_NAME")
res := &Response{
w: w,
Data: nil,
Status: 200,
}
defer res.SendJson()

if err := r.ParseForm(); err != nil {
res.Status = 400
res.Data = "url: /control/get?room=ROOM_NAME"
return
}
room := r.Form["room"][0]

status := 200
msg, err := configure.RoomKeys.GetKey(room)
room := r.Form.Get("room")

if err != nil {
msg = err.Error()
status = 400
if len(room) == 0 {
res.Status = 400
res.Data = "url: /control/get?room=ROOM_NAME"
return
}

res := &Response{
w: w,
Message: msg,
Status: status,
msg, err := configure.RoomKeys.GetKey(room)
if err != nil {
msg = err.Error()
res.Status = 400
}
res.SendJson()
res.Data = msg
}

//http://127.0.0.1:8090/control/delete?room=ROOM_NAME
func (s *Server) handleDelete(w http.ResponseWriter, r *http.Request) {
if r.ParseForm() != nil {
fmt.Fprintf(w, "url: /control/delete?room=ROOM_NAME")
res := &Response{
w: w,
Data: nil,
Status: 200,
}
defer res.SendJson()

if err := r.ParseForm(); err != nil {
res.Status = 400
res.Data = "url: /control/delete?room=ROOM_NAME"
return
}
room := r.Form["room"][0]

room := r.Form.Get("room")

if len(room) == 0 {
res.Status = 400
res.Data = "url: /control/get?room=ROOM_NAME"
return
}

if configure.RoomKeys.DeleteChannel(room) {
fmt.Fprintf(w, "OK")
} else {
fmt.Fprintf(w, "Room Not Found")
res.Data = "Ok"
return
}
res.Status = 404
res.Data = "Room not found"
}

0 comments on commit 24928da

Please sign in to comment.