A signaling server for GB28181

pkg/signaling/signaling.go

@@ -0,0 +1,75 @@
+package signaling
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"io"
+	"net/http"
+	"time"
+
+	"github.com/ossrs/go-oryx-lib/errors"
+	"github.com/ossrs/go-oryx-lib/logger"
+)
+
+type ISignaling interface {
+	Publish(id, ssrc string) (int, error)
+	Unpublish(id string) error
+	GetStreamStatus(id string) (bool, error)
+}
+
+// The r is HTTP API to request, like "http://localhost:1985/gb/v1/publish".
+// The req is the HTTP request body, will be marshal to JSON object. nil is no body
+// The res is the HTTP response body, already unmarshal to JSON object.
+func apiRequest(ctx context.Context, r string, req interface{}, res interface{}) error {
+	var buf bytes.Buffer
+	if req != nil {
+		if err := json.NewEncoder(&buf).Encode(req); err != nil {
+			return errors.Wrapf(err, "Marshal body %v", req)
+		}
+	}
+	logger.Tf(ctx, "Request url api=%v with %v bytes", r, buf.Len())
+
+	method := "POST"
+	if req == nil {
+		method = "GET"
+	}
+	reqObj, err := http.NewRequest(method, r, &buf)
+	if err != nil {
+		return errors.Wrapf(err, "HTTP request %v", buf.String())
+	}
+
+	client := &http.Client{Timeout: 10 * time.Second}
+	resObj, err := client.Do(reqObj.WithContext(ctx))
+	if err != nil {
+		return errors.Wrapf(err, "Do HTTP request %v", buf.String())
+	}
+	defer resObj.Body.Close()
+
+	if resObj.StatusCode != http.StatusOK {
+		return errors.Errorf("Server returned status code=%v", resObj.StatusCode)
+	}
+
+	b2, err := io.ReadAll(resObj.Body)
+	if err != nil {
+		return errors.Wrapf(err, "Read response for %v", buf.String())
+	}
+	logger.Tf(ctx, "Response from %v is %v bytes", r, len(b2))
+
+	errorCode := struct {
+		Code int `json:"code"`
+	}{}
+	if err := json.Unmarshal(b2, &errorCode); err != nil {
+		return errors.Wrapf(err, "Unmarshal %v", string(b2))
+	}
+	if errorCode.Code != 0 {
+		return errors.Errorf("Server fail code=%v %v", errorCode.Code, string(b2))
+	}
+
+	if err := json.Unmarshal(b2, res); err != nil {
+		return errors.Wrapf(err, "Unmarshal %v", string(b2))
+	}
+	logger.Tf(ctx, "Parse response to code=%v ok, %v", errorCode.Code, res)
+
+	return nil
+}