LoginSignup
0
0

More than 3 years have passed since last update.

http sample with go.uber.org/dig

Posted at

package main

import (
    "fmt"
    "net/http"

    "go.uber.org/dig"
)

type Handler struct {
    Greeting string
    Path     string
}

func (h Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "%s from %s", h.Greeting, h.Path)
}

func NewHello1Handler() HandlerResult {
    return HandlerResult{
        Handler: Handler{
            Path:     "/hello1",
            Greeting: "welcome",
        },
    }
}

func NewHello2Handler() HandlerResult {
    return HandlerResult{
        Handler: Handler{
            Path:     "/hello2",
            Greeting: "😄",
        },
    }
}

type HandlerResult struct {
    dig.Out

    Handler Handler `group:"server"`
}

type HandlerParams struct {
    dig.In

    Handlers []Handler `group:"server"`
}

func RunServer(params HandlerParams) error {
    mux := http.NewServeMux()
    for _, h := range params.Handlers {
        mux.Handle(h.Path, h)
    }

    server := &http.Server{
        Addr:    ":8080",
        Handler: mux,
    }
    if err := server.ListenAndServe(); err != nil {
        return err
    }

    return nil
}

func main() {
    container := dig.New()

    container.Provide(NewHello1Handler)
    container.Provide(NewHello2Handler)

    container.Invoke(RunServer)
}

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0