add helper options
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
		
							
								
								
									
										24
									
								
								options.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										24
									
								
								options.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,24 @@ | |||||||
|  | package http | ||||||
|  |  | ||||||
|  | import "context" | ||||||
|  |  | ||||||
|  | type rspCodeKey struct{} | ||||||
|  | type rspCodeVal struct { | ||||||
|  | 	code int | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetRspCode saves response code in context, must be used by handler to specify http code | ||||||
|  | func SetRspCode(ctx context.Context, code int) { | ||||||
|  | 	if rsp, ok := ctx.Value(rspCodeKey{}).(*rspCodeVal); ok { | ||||||
|  | 		rsp.code = code | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetRspCode used internally by generated http server handler | ||||||
|  | func GetRspCode(ctx context.Context) int { | ||||||
|  | 	var code int | ||||||
|  | 	if rsp, ok := ctx.Value(rspCodeKey{}).(*rspCodeVal); ok { | ||||||
|  | 		code = rsp.code | ||||||
|  | 	} | ||||||
|  | 	return code | ||||||
|  | } | ||||||
		Reference in New Issue
	
	Block a user