TestMu Conference is Back! Future of Testing + AI LambdaTest G2 Announcement

How to use Metadata method of websocket Package

Best Testkube code snippet using websocket.Metadata

websocket_service.go

Source:websocket_service.go

...52	}53	c.participant.Name = claims.Name54	// default set false55	c.kws.SetAttribute("isAdmin", false)56	metadata := new(plugnmeet.UserMetadata)57	err = json.Unmarshal([]byte(claims.Metadata), metadata)58	if err != nil {59		_ = c.kws.EmitTo(c.kws.UUID, []byte("can't Unmarshal metadata!"), ikisocket.TextMessage)60		return false61	}62	if metadata.IsAdmin {63		c.kws.SetAttribute("isAdmin", true)64		c.participant.IsAdmin = true65	}66	return true67}68func (c *websocketController) addUser() {69	config.AppCnf.AddChatUser(c.participant.RoomId, c.participant)70	c.kws.SetAttribute("userId", c.participant.UserId)71	c.kws.SetAttribute("roomId", c.participant.RoomId)...

websocket.go

Source:websocket.go

1package communicator2import (3	"context"4	"encoding/json"5	"fmt"6	"net/url"7	"os"8	"github.com/gorilla/websocket"9	log "github.com/sirupsen/logrus"10	"github.com/zobinHuang/BrosCloud/provider/streamer/model"11)12/*13	@struct: WebsocketCommunicator14	@description: communicator to daemon and scheduler15*/16type WebsocketCommunicator struct {17	SchedulerWSConnection *model.Websocket18	DaemonWSConnection    *model.Websocket19	InstanceDAL           model.InstanceDAL20	SchedulerDAL          model.SchedulerDAL21	DaemonDAL             model.DaemonDAL22	WebRTCStreamDAL       model.WebRTCStreamDAL23}24/*25	@struct: WebsocketCommunicatorConfig26	@description: used for config instance of struct WebsocketCommunicator27*/28type WebsocketCommunicatorConfig struct {29	InstanceDAL     model.InstanceDAL30	SchedulerDAL    model.SchedulerDAL31	DaemonDAL       model.DaemonDAL32	WebRTCStreamDAL model.WebRTCStreamDAL33}34/*35	@func: NewWebsocketCommunicator36	@description:37		create, config and return an instance of struct WebsocketCommunicator38*/39func NewWebsocketCommunicator(c *WebsocketCommunicatorConfig) model.WebsocketCommunicator {40	wsCommunicator := &WebsocketCommunicator{41		SchedulerDAL:    c.SchedulerDAL,42		DaemonDAL:       c.DaemonDAL,43		InstanceDAL:     c.InstanceDAL,44		WebRTCStreamDAL: c.WebRTCStreamDAL,45	}46	err := wsCommunicator.InitDaemonConnection()47	if err != nil {48		log.WithFields(log.Fields{49			"error": err,50		}).Fatalln("Failed to build connection to provider daemon")51	}52	// wsCommunicator.InitSchdulerConnectionForDebug()53	return wsCommunicator54}55func (s *WebsocketCommunicator) InitDaemonConnection() error {56	ctx := context.Background()57	// obtain websocket metadata to daemon58	wsScheme := os.Getenv("DAEMON_WS_SCHEME")59	wsHostname := os.Getenv("DAEMON_WS_HOSTNAME")60	wsPort := os.Getenv("DAEMON_WS_PORT")61	wsPath := os.Getenv("DAEMON_WS_PATH")62	completeHostname := fmt.Sprintf("%s:%s", wsHostname, wsPort)63	daemonURL := url.URL{64		Scheme:   wsScheme,65		Host:     completeHostname,66		Path:     wsPath,67		RawQuery: "type=provider",68	}69	conn, _, err := websocket.DefaultDialer.Dial(daemonURL.String(), nil)70	if err != nil {71		log.WithFields(log.Fields{72			"error": err,73		}).Warn("Failed to build websocket connection to daemon")74		return err75	}76	// store websocket to daemon, and start to serve it77	s.NewDaemonConnection(ctx, conn)78	// register recv callbacks79	s.InitDaemonRecvRoute(ctx)80	// start keep alive81	s.KeepDaemonConnAlive(ctx)82	return nil83}84/*85	@func: InitSchdulerConnectionForDebug86	@description:87		initialize websocket communication just for debugging88*/89func (s *WebsocketCommunicator) InitSchdulerConnectionForDebug() {90	wsScheme := os.Getenv("SCHEDULER_WS_SCHEME")91	wsHostname := os.Getenv("SCHEDULER_WS_HOSTNAME")92	wsPort := os.Getenv("SCHEDULER_WS_PORT")93	wsPath := os.Getenv("SCHEDULER_WS_PATH")94	ctx := context.Background()95	// connect to scheduler96	err := s.ConnectToScheduler(ctx,97		wsScheme,98		wsHostname,99		wsPort,100		wsPath,101	)102	if err != nil {103		log.Fatalln(err.Error())104	}105	// start keep alive routine to scheduler106	s.KeepSchedulerConnAlive(ctx)107	// initilize scheduler recv callbacks route108	s.InitSchedulerRecvRoute(ctx)109	// generate request websocket to scheduler to register provider metadata110	reqToScheduler := struct {111		ProviderType string `json:"provider_type"`112	}{113		ProviderType: "stream",114	}115	reqToSchedulerString, err := json.Marshal(reqToScheduler)116	if err != nil {117		log.WithFields(log.Fields{118			"Warn Type":        "Recv Callback Error",119			"Recv Packet Type": "register_provider_metadata",120			"error":            err,121		}).Fatal("Failed to marshal provider metadata, abandoned")122	}123	// send metadata124	s.SchedulerWSConnection.Send(model.WSPacket{125		PacketType: "init_provider_metadata",126		Data:       string(reqToSchedulerString),127	}, nil)128}...

msg_test.go

Source:msg_test.go

...5	"io"6	"testing"7	"github.com/stretchr/testify/assert"8)9func TestAddAmountMetadata(t *testing.T) {10	message := &DIDCommMsg{}11	if message.GetAmountMetadata() != int64(0) {12		t.Error("empty amount not set to 0")13	}14	message.AddAmountMetadata(1)15	if message.InternalMetadata["amount"] != int64(1) {16		t.Error("amount did not set correctly")17	}18	if message.GetAmountMetadata() != int64(1) {19		t.Error("amount did not get correctly")20	}21	message.AddAmountMetadata(2)22	if message.GetAmountMetadata() != int64(2) {23		t.Error("amount did not overwrite correctly")24	}25}26func TestAddWebsocketIdMetadata(t *testing.T) {27	message := &DIDCommMsg{}28	if message.GetWebsocketId() != "" {29		t.Error("empty websocket id not set to empty string")30	}31	message.AddWebsocketId("1")32	if message.InternalMetadata["websocket_id"] != "1" {33		t.Error("websocket id did not set correctly")34	}35	if message.GetWebsocketId() != "1" {36		t.Error("websocket id did not get correctly")37	}38	message.AddWebsocketId("2")39	if message.GetWebsocketId() != "2" {40		t.Error("websocket id did not overwrite correctly")41	}42}43func TestParseMsgFromReqSuccess(t *testing.T) {44	message := NewBasicMsg("hello world")45	marshaledMessage, err := json.Marshal(message)46	assert.Nil(t, err)47	reader := bytes.NewReader(marshaledMessage)48	parsedMsgInterface, err := ParseMsgFromReq(io.NopCloser(reader))49	assert.Nil(t, err)50	parsedMsg, err := ParseDIDCommMsg(parsedMsgInterface)51	assert.Nil(t, err)52	// parse body as a basic message53	parsedBody, err := ParseBasicMsgBody(parsedMsg.Body)54	assert.Nil(t, err)55	parsedMsg.Body = parsedBody56	assert.Equal(t, message, parsedMsg)57}58func TestParseMsgFromBytesSuccess(t *testing.T) {59	message := NewBasicMsg("hello world")60	marshaledMessage, err := json.Marshal(message)61	assert.Nil(t, err)62	parsedMsgInterface, err := ParseMsgFromBytes(marshaledMessage, 0)63	assert.Nil(t, err)64	parsedMsg, err := ParseDIDCommMsg(parsedMsgInterface)65	assert.Nil(t, err)66	// parse body as a basic message67	parsedBody, err := ParseBasicMsgBody(parsedMsg.Body)68	assert.Nil(t, err)69	parsedMsg.Body = parsedBody70	// ignore internal metadata from parsing71	parsedMsg.InternalMetadata = make(map[string]interface{})72	assert.Equal(t, message, parsedMsg)73}...

Metadata

Using AI Code Generation

1import (2var upgrader = websocket.Upgrader{3    CheckOrigin: func(r *http.Request) bool { return true },4}5func handler(w http.ResponseWriter, r *http.Request) {6    c, err := upgrader.Upgrade(w, r, nil)7    if err != nil {8        fmt.Println("upgrade err:", err)9    }10    defer c.Close()11    for {12        mType, msg, err := c.ReadMessage()13        if err != nil {14            fmt.Println("read err:", err)15        }16        fmt.Println("recv:", string(msg))17        if err = c.WriteMessage(mType, msg); err != nil {18            fmt.Println("write err:", err)19        }20    }21}22func main() {23    http.HandleFunc("/", handler)24    http.ListenAndServe(":8080", nil)25}26import (27var upgrader = websocket.Upgrader{28    CheckOrigin: func(r *http.Request) bool { return true },29}30func main() {31    u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/"}32    c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)33    if err != nil {34        fmt.Println("dial err:", err)35    }36    defer c.Close()37    for {38        _, msg, err := c.ReadMessage()39        if err != nil {40            fmt.Println("read err:", err)41        }42        fmt.Println("recv:", string(msg))43        if err = c.WriteMessage(websocket.TextMessage, msg); err != nil {44            fmt.Println("write err:", err)45        }46    }47}

Metadata

Using AI Code Generation

1import (2func main() {3    if err != nil {4        panic(err)5    }6    fmt.Println(ws.Config().Location)7    fmt.Println(ws.Config().Origin)8    fmt.Println(ws.Config().Version)9    fmt.Println(ws.Config().Header)10    fmt.Println(ws.Config().Subprotocols)11    fmt.Println(ws.Config().TlsConfig)12    fmt.Println(ws.Config().HandshakeTimeout)13    fmt.Println(ws.Config().ReadBufferSize)14    fmt.Println(ws.Config().WriteBufferSize)15    fmt.Println(ws.Config().WriteBufferPool)16    fmt.Println(ws.Config().ReadBufferPool)17    fmt.Println(ws.Config().CompressionLevel)18    fmt.Println(ws.Config().CompressionN

Metadata

Using AI Code Generation

1import (2func main() {3    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {4        ws, err := websocket.Upgrade(w, r, w.Header(), 1024, 1024)5        if _, ok := err.(websocket.HandshakeError); ok {6            http.Error(w, "Not a websocket handshake", 400)7        } else if err != nil {8        }9        defer ws.Close()10        for {11            err := websocket.Message.Receive(ws, &msg)12            if err != nil {13            }14            fmt.Printf("Received: %s15            for k, v := range ws.Request().Header {16                fmt.Printf("%s: %s17            }18        }19    })20    http.ListenAndServe(":8080", nil)21}22import (23func main() {24    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {25        ws, err := websocket.Upgrade(w, r, w.Header(), 1024, 1024)26        if _, ok := err.(websocket.HandshakeError); ok {27            http.Error(w, "Not a websocket handshake", 400)28        } else if err != nil {29        }30        defer ws.Close()31        for {32            _, msg, err := ws.ReadMessage()33            if err != nil {34            }35            fmt.Printf("Received: %s36            for k, v := range ws.Request().Header {37                fmt.Printf("%s: %s38            }39        }40    })41    http.ListenAndServe(":8080", nil)42}43import (44func main() {45    http.HandleFunc("/", func(w http.ResponseWriter

Metadata

Using AI Code Generation

1ws.onopen = function() {2    console.log("connected");3    ws.send("hello");4};5ws.onmessage = function(msg) {6    console.log(msg);7};8ws.onclose = function() {9    console.log("disconnected");10};

Metadata

Using AI Code Generation

1import (2func main() {3 metadata := websocket.Metadata{4 }5 fmt.Println(metadata)6}

Metadata

Using AI Code Generation

1import (2func main() {3    dialer := websocket.Dialer{}4    conn, _, err := dialer.Dial(url, nil)5    if err != nil {6        fmt.Println("Error:", err)7    }8    defer conn.Close()9    fmt.Println(conn.Metadata())10}11map[Sec-Websocket-Accept:u1Q2l2+GwPQ2eZGJQbX+XsI9H1Y= Sec-Websocket-Extensions:permessage-deflate; client_max_window_bits Sec-Websocket-Protocol:chat Sec-Websocket-Version:13 Upgrade:websocket]

Metadata

Using AI Code Generation

1import (2func main() {3	if err != nil {4		fmt.Println(err)5	}6	metadata := ws.Metadata()7	fmt.Println("Metadata: ", metadata)8}9import (10func main() {11	if err != nil {12		fmt.Println(err)13	}14	remoteAddr := ws.RemoteAddr()15	fmt.Println("Remote Address: ", remoteAddr)16}17import (18func main() {19	if err != nil {20		fmt.Println(err)21	}22	request := ws.Request()23	fmt.Println("Request: ", request)24}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.