Mirrors/kararas_iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-12-17 09:57:01 +00:00
Code Releases Activity

(#1554) Add support for all common compressions (write and read)

Browse Source 
- Remove the context.Context interface and export the *context, the iris.Context now points to the pointer\nSupport compression and rate limiting in the FileServer\nBit of code organisation


Former-commit-id: ad1c61bf968059510c6be9e7f2cceec7da70ba17
This commit is contained in:
Gerasimos (Makis) Maropoulos
2020-07-10 23:21:09 +03:00
parent 645da2b2ef
commit 0f113dfcda
112 changed files with 2119 additions and 3390 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
websocket/websocket.go
View File

@@ -37,7 +37,7 @@ var (
New = neffos.New
// DefaultIDGenerator returns a universal unique identifier for a new connection.
// It's the default `IDGenerator` if missing.
DefaultIDGenerator = func(ctx context.Context) string {
DefaultIDGenerator = func(ctx *context.Context) string {
return neffos.DefaultIDGenerator(ctx.ResponseWriter(), ctx.Request())
}
@@ -138,10 +138,10 @@ func SetDefaultUnmarshaler(fn func(data []byte, v interface{}) error) {
}
// IDGenerator is an iris-specific IDGenerator for new connections.
type IDGenerator func(context.Context) string
type IDGenerator func(*context.Context) string
func wrapIDGenerator(idGen IDGenerator) func(ctx context.Context) neffos.IDGenerator {
return func(ctx context.Context) neffos.IDGenerator {
func wrapIDGenerator(idGen IDGenerator) func(ctx *context.Context) neffos.IDGenerator {
return func(ctx *context.Context) neffos.IDGenerator {
return func(w http.ResponseWriter, r *http.Request) string {
return idGen(ctx)
}
@@ -157,7 +157,7 @@ func Handler(s *neffos.Server, IDGenerator ...IDGenerator) context.Handler {
idGen = IDGenerator[0]
}
return func(ctx context.Context) {
return func(ctx *context.Context) {
if ctx.IsStopped() {
return
}
@@ -167,7 +167,7 @@ func Handler(s *neffos.Server, IDGenerator ...IDGenerator) context.Handler {
// Upgrade upgrades the request and returns a new websocket Conn.
// Use `Handler` for higher-level implementation instead.
func Upgrade(ctx context.Context, idGen IDGenerator, s *neffos.Server) *neffos.Conn {
func Upgrade(ctx *context.Context, idGen IDGenerator, s *neffos.Server) *neffos.Conn {
conn, _ := s.Upgrade(ctx.ResponseWriter(), ctx.Request(), func(socket neffos.Socket) neffos.Socket {
return &socketWrapper{
Socket: socket,
@@ -180,11 +180,11 @@ func Upgrade(ctx context.Context, idGen IDGenerator, s *neffos.Server) *neffos.C
type socketWrapper struct {
neffos.Socket
ctx context.Context
ctx *context.Context
}
// GetContext returns the Iris Context from a websocket connection.
func GetContext(c *neffos.Conn) context.Context {
func GetContext(c *neffos.Conn) *context.Context {
if sw, ok := c.Socket().(*socketWrapper); ok {
return sw.ctx
}