refactor: remove method

Author: DarthPestilane

router_context.go

@@ -22,7 +22,7 @@ type Context interface {
 	Session() Session
 
 	// SetSession sets session.
-	SetSession(sess Session)
+	SetSession(sess Session) Context
 
 	// Request returns request message entry.
 	Request() *message.Entry
@@ -43,9 +43,6 @@ type Context interface {
 	// Response returns the response message entry.
 	Response() *message.Entry
 
-	// RawResponseData returns the not yet encoded response data.
-	RawResponseData() interface{}
-
 	// SetResponse encodes data with session's codec and sets response message entry.
 	SetResponse(id, data interface{}) error
 
@@ -77,12 +74,11 @@ type Context interface {
 
 // routeContext implements the Context interface.
 type routeContext struct {
-	mu          sync.RWMutex
-	storage     map[string]interface{}
-	session     Session
-	reqEntry    *message.Entry
-	respEntry   *message.Entry
-	rawRespData interface{}
+	mu        sync.RWMutex
+	storage   map[string]interface{}
+	session   Session
+	reqEntry  *message.Entry
+	respEntry *message.Entry
 }
 
 // Deadline implements the context.Context Deadline method.
@@ -115,8 +111,9 @@ func (c *routeContext) Session() Session {
 }
 
 // SetSession sets session.
-func (c *routeContext) SetSession(sess Session) {
+func (c *routeContext) SetSession(sess Session) Context {
 	c.session = sess
+	return c
 }
 
 // Request implements Context.Request method.
@@ -168,17 +165,6 @@ func (c *routeContext) Response() *message.Entry {
 	return c.respEntry
 }
 
-// RawResponseData returns the not yet encoded response data.
-func (c *routeContext) RawResponseData() interface{} {
-	if c.rawRespData != nil {
-		return c.rawRespData
-	}
-	if c.respEntry != nil {
-		return c.respEntry.Data
-	}
-	return nil
-}
-
 // SetResponse implements Context.SetResponse method.
 func (c *routeContext) SetResponse(id, data interface{}) error {
 	codec := c.session.Codec()
@@ -189,7 +175,6 @@ func (c *routeContext) SetResponse(id, data interface{}) error {
 	if err != nil {
 		return err
 	}
-	c.rawRespData = data
 	c.respEntry = &message.Entry{
 		ID:   id,
 		Data: dataRaw,

router_context_test.go

@@ -149,7 +149,6 @@ func Test_routeContext_SetResponse(t *testing.T) {
 		err := c.SetResponse(1, "test")
 		assert.NoError(t, err)
 		assert.Equal(t, c.respEntry, entry)
-		assert.Equal(t, c.rawRespData, "test")
 	})
 }
 
@@ -248,30 +247,10 @@ func Test_routeContext_MustSetResponse(t *testing.T) {
 	})
 }
 
-func Test_routeContext_RawResponseData(t *testing.T) {
-	t.Run("when raw resp data is not nil", func(t *testing.T) {
-		c := newContext(nil, nil)
-		c.rawRespData = 123
-		assert.Equal(t, c.RawResponseData(), 123)
-	})
-	t.Run("when resp entry is not nil", func(t *testing.T) {
-		c := newContext(nil, nil)
-		c.rawRespData = nil
-		c.respEntry = &message.Entry{
-			Data: []byte("123"),
-		}
-		assert.Equal(t, c.RawResponseData(), []byte("123"))
-	})
-	t.Run("when resp entry is nil", func(t *testing.T) {
-		c := newContext(nil, nil)
-		assert.Nil(t, c.RawResponseData())
-	})
-}
-
 func Test_routeContext_SetSession(t *testing.T) {
 	sess := newSession(nil, &sessionOption{})
 	c := newContext(nil, nil)
-	c.SetSession(sess)
+	assert.Equal(t, c.SetSession(sess), c)
 	assert.Equal(t, c.Session(), sess)
 }