You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
988 B
35 lines
988 B
4 years ago
|
package localize
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/BurntSushi/toml"
|
||
|
|
"github.com/nicksnyder/go-i18n/v2/i18n"
|
||
|
|
"golang.org/x/text/language"
|
||
|
|
|
||
|
|
"github.com/go-kratos/kratos/v2/middleware"
|
||
|
|
"github.com/go-kratos/kratos/v2/transport"
|
||
|
|
)
|
||
|
|
|
||
|
|
type localizerKey struct{}
|
||
|
|
|
||
|
|
func I18N() middleware.Middleware {
|
||
|
|
bundle := i18n.NewBundle(language.English)
|
||
|
|
bundle.RegisterUnmarshalFunc("toml", toml.Unmarshal)
|
||
|
|
bundle.MustLoadMessageFile("../../active.zh.toml")
|
||
|
|
|
||
|
|
return func(handler middleware.Handler) middleware.Handler {
|
||
|
|
return func(ctx context.Context, req interface{}) (reply interface{}, err error) {
|
||
|
|
if tr, ok := transport.FromServerContext(ctx); ok {
|
||
|
|
accept := tr.RequestHeader().Get("accept-language")
|
||
|
|
println(accept)
|
||
|
|
localizer := i18n.NewLocalizer(bundle, accept)
|
||
|
|
ctx = context.WithValue(ctx, localizerKey{}, localizer)
|
||
|
|
}
|
||
|
|
return handler(ctx, req)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func FromContext(ctx context.Context) *i18n.Localizer {
|
||
|
|
return ctx.Value(localizerKey{}).(*i18n.Localizer)
|
||
|
|
}
|