Golang MiddlewareSimple Examples

Example #1

File: inject_config.go Project: lue828/osinstall-server

// InjectConfig 注入*config.Config
func InjectConfig(conf *config.Config) rest.Middleware {
	fn := func(h rest.HandlerFunc) rest.HandlerFunc {
		return func(ctx context.Context, w rest.ResponseWriter, r *rest.Request) {
			ctx = context.WithValue(ctx, &ctxConfigKey, conf)
			h(ctx, w, r)
		}
	}
	return rest.MiddlewareSimple(fn)
}

Example #2

File: inject_logger.go Project: lue828/osinstall-server

// InjectLogger 注入logger.Logger
func InjectLogger(logger logger.Logger) rest.Middleware {
	fn := func(h rest.HandlerFunc) rest.HandlerFunc {
		return func(ctx context.Context, w rest.ResponseWriter, r *rest.Request) {
			ctx = context.WithValue(ctx, &ctxLoggerKey, logger)
			h(ctx, w, r)
		}
	}
	return rest.MiddlewareSimple(fn)
}

Example #3

File: inject_repo.go Project: lue828/osinstall-server

// InjectRepo 注入model.Repo
func InjectRepo(repo model.Repo) rest.Middleware {
	fn := func(h rest.HandlerFunc) rest.HandlerFunc {
		return func(ctx context.Context, w rest.ResponseWriter, r *rest.Request) {
			ctx = context.WithValue(ctx, &ctxRepoKey, repo)
			h(ctx, w, r)
		}
	}
	return rest.MiddlewareSimple(fn)
}