expose SetContextValue from ClientConn

Author: jkralik

dtls/server_test.go

@@ -136,7 +136,7 @@ func TestServer_SetContextValueWithPKI(t *testing.T) {
 		// set connection context certificate
 		clientCert, err := x509.ParseCertificate(dtlsConn.ConnectionState().PeerCertificates[0])
 		require.NoError(t, err)
-		cc.Session().SetContextValue("client-cert", clientCert)
+		cc.SetContextValue("client-cert", clientCert)
 	}
 	handle := func(w *client.ResponseWriter, r *pool.Message) {
 		// get certificate from connection context

dtls/session.go

@@ -80,6 +80,7 @@ func (s *Session) Context() context.Context {
 	return *s.ctx.Load().(*context.Context)
 }
 
+// SetContextValue stores the value associated with key to context of connection.
 func (s *Session) SetContextValue(key interface{}, val interface{}) {
 	s.mutex.Lock()
 	defer s.mutex.Unlock()

examples/dtls/pki/server/main.go

@@ -25,7 +25,7 @@ func onNewClientConn(cc *client.ClientConn, dtlsConn *piondtls.Conn) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	cc.Session().SetContextValue("client-cert", clientCert)
+	cc.SetContextValue("client-cert", clientCert)
 	cc.AddOnClose(func() {
 		log.Println("closed connection")
 	})

tcp/clientconn.go

@@ -441,3 +441,8 @@ func (cc *ClientConn) Client() *ClientTCP {
 func (cc *ClientConn) Sequence() uint64 {
 	return cc.session.Sequence()
 }
+
+// SetContextValue stores the value associated with key to context of connection.
+func (cc *ClientConn) SetContextValue(key interface{}, val interface{}) {
+	cc.session.SetContextValue(key, val)
+}

tcp/server_test.go

@@ -123,7 +123,7 @@ func TestServer_SetContextValueWithPKI(t *testing.T) {
 		// set connection context certificate
 		clientCert := tlscon.ConnectionState().PeerCertificates[0]
 		require.NoError(t, err)
-		cc.Session().SetContextValue("client-cert", clientCert)
+		cc.SetContextValue("client-cert", clientCert)
 	}
 	handle := func(w *tcp.ResponseWriter, r *pool.Message) {
 		// get certificate from connection context

tcp/session.go

@@ -91,6 +91,7 @@ func NewSession(
 	return s
 }
 
+// SetContextValue stores the value associated with key to context of connection.
 func (s *Session) SetContextValue(key interface{}, val interface{}) {
 	s.mutex.Lock()
 	defer s.mutex.Unlock()

udp/client/clientconn.go

@@ -524,3 +524,8 @@ func (cc *ClientConn) Process(datagram []byte) error {
 func (cc *ClientConn) Client() *Client {
 	return NewClient(cc)
 }
+
+// SetContextValue stores the value associated with key to context of connection.
+func (cc *ClientConn) SetContextValue(key interface{}, val interface{}) {
+	cc.session.SetContextValue(key, val)
+}

udp/session.go

@@ -46,6 +46,7 @@ func NewSession(
 	return s
 }
 
+// SetContextValue stores the value associated with key to context of connection.
 func (s *Session) SetContextValue(key interface{}, val interface{}) {
 	s.mutex.Lock()
 	defer s.mutex.Unlock()