feat: getter for underlined connection

.github/workflows/test.yml

@@ -37,9 +37,9 @@ jobs:
         run: make build-all
 
       - name: Lint
-        uses: golangci/golangci-lint-action@v2
+        uses: golangci/golangci-lint-action@v3
         with:
-          version: v1.32
+          version: v1.45
           skip-pkg-cache: true
           skip-build-cache: true
 

session.go

@@ -27,6 +27,9 @@ type Session interface {
 
 	// AllocateContext gets a Context ships with current session.
 	AllocateContext() Context
+
+	// Conn returns the underlined connection.
+	Conn() net.Conn
 }
 
 type session struct {
@@ -109,6 +112,11 @@ func (s *session) AllocateContext() Context {
 	return c
 }
 
+// Conn returns the underlined connection instance.
+func (s *session) Conn() net.Conn {
+	return s.conn
+}
+
 // readInbound reads message packet from connection in a loop.
 // And send unpacked message to reqQueue, which will be consumed in router.
 // The loop breaks if errors occurred or the session is closed.

session_test.go

@@ -419,3 +419,12 @@ func Test_session_SetID(t *testing.T) {
 	sess.SetID(123)
 	assert.Equal(t, sess.ID(), 123)
 }
+
+func Test_session_Conn(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	conn := mock.NewMockConn(ctrl)
+	s := newSession(conn, &sessionOption{})
+	assert.Equal(t, s.Conn(), conn)
+}