Избегая глобальных переменных в маршрутах для labstack/echo

Question

Я использую веб-сервер labstack/echo и gofight для модульного тестирования. В процессе обучения go хотел бы знать, существует ли идиома go для доступа к состоянию вне (встроенной) структуры эха. Например:

type WebPlusDB struct {
    web *echo.Echo
    db  *databaseInterface
}

func NewEngine() *echo.Echo {
    e := echo.New()
    e.GET("/hello", route_hello)
    return e    
}

func NewWebPlusDb() {
    e := NewEngine()
    db := database.New()   
    return WebPlusDB{e,db}
}

// for use in unit tests
func NewFakeEngine() *WebPlusDB {
    e := NewEngine()
    db := fakeDatabase.New()   
    return WebPlusDB{e,db}
}    

func route_hello(c echo.Context) error {
    log.Printf("Hello world\n")

    // how do I access WebPlusDB.db from here?

    return c.String(http.StatusOK, "hello world")
}

Тогда в тестовом коде я использую:

import (
    "github.com/labstack/echo"
    "github.com/appleboy/gofight"
    "github.com/stretchr/testify/assert"
    "testing"
)

func TestHelloWorld(t *testing.T) {
    r := gofight.New()

    r.GET("/hello").
          Run(NewFakeEngine(), func(r gofight.HTTPResponse, rq gofight.HTTPRequest) {
        assert.Equal(t, http.StatusOK, r.Code)
        assert.Equal(t, "hello world", r.Body.String())
        // test database access
    })

}

Самое простое решение - использовать глобальную переменную вместо того, чтобы встраивать echo в "WebPlusDB" и добавлять туда состояние. Я хотел бы лучше инкапсуляции. Я думаю, что я должен использовать что-то вроде структуры WebPlusDB, а не echo.Echo плюс глобальное состояние. Возможно, это не имеет большого значения для модульного тестирования, но в большей схеме правильной работы (в данном случае избегая глобальных) я хотел бы знать.

Есть ли решение или это слабое место в дизайне эха? У него есть точки расширения для промежуточного программного обеспечения, но серверная часть базы данных на самом деле не является промежуточным программным обеспечением, как определено здесь.

Примечание: я использую базу данных здесь, чтобы проиллюстрировать общий случай, но это может быть что угодно (на самом деле я использую amqp)

Похоже, вы можете расширить контекстный интерфейс, но где он создан? Похоже, он использует вид удручающего:

e.GET("/", func(c echo.Context) error {
    cc := c.(*CustomContext)
}

Я думаю (возможно, неправильно), что это разрешено только для интерфейса, и echo.Context.Echo() возвращает тип, а не интерфейс.

Answer 1

Вы можете передать метод экземпляра как значение функции, что, вероятно, является наиболее простым способом справиться с этим:

type WebPlusDB struct {
    web *echo.Echo
    db  *databaseInterface
}

func (w WebPlusDB) route_hello(c echo.Context) error {
    log.Printf("Hello world\n")

    // do whatever with w

    return c.String(http.StatusOK, "hello world")
}

func NewEngine() *echo.Echo {
    e := echo.New()
    w := NewWebPlusDb()
    e.GET("/hello", w.route_hello)
    return e    
}