Как реализовать интерфейс, в котором можно регистрировать функции на подобии http.HandleFunc?

Question

[Bekkazy]

Необходимо сделать почти то же самое что и в пакете net/http

Сначала зарегистрировать хэндлеры типа:
mypackage.RegisterHandler("/path", handler)

Нашёл код но как это реализовать, чтобы можно было регистрировать хэндлеры вне пакета, чтобы они потом вызывались?

Answer 1

[igorzakhar]

Такой вариант (если я правильно понял твой вопрос):

package main

import (
	"fmt"
)

type HandlersMap map[string]func(string) string

func (hm HandlersMap) RegisterHandler(path string, handler func(string) string) {
	hm[path] = handler
}

func main() {
	m := make(HandlersMap, 0)

	m.RegisterHandler("1", myFunc1)
	m.RegisterHandler("2", myFunc2)

	handler1 := m["1"]
	handler2 := m["2"]

	fmt.Println(handler1("handler 1 output"))
	fmt.Println(handler2("handler 2 output"))
}

func myFunc1(s string) string {
	return fmt.Sprintf("MyFunc1: %s", s)
}

func myFunc2(s string) string {
	return fmt.Sprintf("MyFunc2: %s", s)
}

$ go run main.go 
MyFunc1: handler 1 output
MyFunc2: handler 2 output

Comments

[Bekkazy]

В пакете net/http нет необходимости инициализировать map, можно просто написать:
http.HandleFunc("/", indexHandler)

Можно ли сделать что то подобное?

Или внутри пакета заранее как то инициализировать?

[igorzakhar]

Bekkazy, если посмотреть в пакет http там объявляется DefaultServeMux который представляет из себя следующую структуру:

var DefaultServeMux = &defaultServeMux

var defaultServeMux ServeMux

type ServeMux struct {
	mu    sync.RWMutex
	m     map[string]muxEntry
	hosts bool // whether any patterns contain hostnames
}

type muxEntry struct {
	h       Handler
	pattern string
}

func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
	DefaultServeMux.HandleFunc(pattern, handler)
}

func (mux *ServeMux) HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
	if handler == nil {
		panic("http: nil handler")
	}
	mux.Handle(pattern, HandlerFunc(handler))
}

func (mux *ServeMux) Handle(pattern string, handler Handler) {
	mux.mu.Lock()
	defer mux.mu.Unlock()

	if pattern == "" {
		panic("http: invalid pattern")
	}
	if handler == nil {
		panic("http: nil handler")
	}
	if _, exist := mux.m[pattern]; exist {
		panic("http: multiple registrations for " + pattern)
	}

	if mux.m == nil {
		mux.m = make(map[string]muxEntry)
	}
	mux.m[pattern] = muxEntry{h: handler, pattern: pattern}

	if pattern[0] != '/' {
		mux.hosts = true
	}
}

[igorzakhar]

Bekkazy, мне до сих пор не понятна фраза:

... регистрировать хэндлеры вне пакета, чтобы они потом вызывались?

[Bekkazy]

igorzakhar, Ну моя первая мысль о реализации этого, примерно то же самое что и вы написали в первом примере. Но используя пакет net/http нет необходимости сначала как то инициализировать новый тип для хэндлеров. Хотелось примерно такой же реализации

[igorzakhar]

Bekkazy, у структуры ServeMux( есть поле map[string]muxEntry, где хранятся хэндлеры) По умолчанию все хэндлеры попадают в DefaultServeMux, в пакете "net/http" уже все определено:

// код из пакета net/http
var DefaultServeMux = &defaultServeMux
var defaultServeMux ServeMux

А по твоему где хранятся обработчики когда ты используешь "net/http"? По умолчанию они хранятся в DefaultServeMux. В моем примере роль DefaultServeMux исполняет HandlersMap.

[Алексей]

Но используя пакет net/http нет необходимости сначала как то инициализировать новый тип для хэндлеров

Bekkazy поместите инициализацию в функцию init вашего пакета.

[igorzakhar]

Алексей, в пакете "net/http" уже объявлена переменная DefaultServeMux тип ServeMux:

// код из пакета net/http
var DefaultServeMux = &defaultServeMux
var defaultServeMux ServeMux

которая имеет вид:

// код из пакета net/http
type ServeMux struct {
  mu    sync.RWMutex
  m     map[string]muxEntry
  hosts bool // whether any patterns contain hostnames
}

type muxEntry struct {
  h       Handler
  pattern string
}

Не нужен никакой init(). Просто объявляем переменную в пакете (имя должно начинаться с заглавной буквы) и она будет доступна после импорта.

[Алексей]

Насколько помню map надо инициализировать, недостаточно просто объявить перемнную.

[igorzakhar]

Алексей, еще раз смотрим в пакет "net/http":

var DefaultServeMux = &defaultServeMux

var defaultServeMux ServeMux

...

type ServeMux struct {
	mu    sync.RWMutex
	m     map[string]muxEntry
	hosts bool // whether any patterns contain hostnames
}

type muxEntry struct {
	h       Handler
	pattern string
}

func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
  DefaultServeMux.HandleFunc(pattern, handler)
}

func (mux *ServeMux) HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
  if handler == nil {
    panic("http: nil handler")
  }
  mux.Handle(pattern, HandlerFunc(handler))
}

func (mux *ServeMux) Handle(pattern string, handler Handler) {
  mux.mu.Lock()
  defer mux.mu.Unlock()

  if pattern == "" {
    panic("http: invalid pattern")
  }
  if handler == nil {
    panic("http: nil handler")
  }
  if _, exist := mux.m[pattern]; exist {
    panic("http: multiple registrations for " + pattern)
  }

  if mux.m == nil {
    mux.m = make(map[string]muxEntry)
  }
  mux.m[pattern] = muxEntry{h: handler, pattern: pattern}

  if pattern[0] != '/' {
    mux.hosts = true
  }
}

В методе Handle есть такой кусок кода:

...
  if mux.m == nil {
    mux.m = make(map[string]muxEntry)
  }
...

Answer 2

[stratosmi]

Все три указанные библиотеки умеют регистрировать Handler из net/http:

1. https://github.com/go-ozzo/ozzo-routing
   
2. www.gorillatoolkit.org/pkg/pat
   
3. www.gorillatoolkit.org/pkg/mux
   

Comments

[Bekkazy]

Мне не нужно использовать пакет http, тем более пакеты которые работают с net/http

Мне нужно реализовать это самому

[stratosmi]

Bekkazy,
погляди ВНУТРЬ приведенных пакетов.
Исходники-то открыты.

Я тебе зачем их так много накидал?
Чтобы ты посмотреть мог различные способы, как сие реализуется.