Skip to content
Toggle navigation
Toggle navigation
This project
Loading...
Sign in
张磊
/
FileStorageBeego
Go to a project
Toggle navigation
Toggle navigation pinning
Projects
Groups
Snippets
Help
Project
Activity
Repository
Pipelines
Graphs
Issues
0
Merge Requests
0
Wiki
Network
Create a new issue
Builds
Commits
Issue Boards
Files
Commits
Network
Compare
Branches
Tags
f1e50596
authored
2013-07-08 15:13:51 +0800
by
astaxie
Browse Files
Options
Browse Files
Tag
Download
Email Patches
Plain Diff
fix #69 refer to
http://www.php.net/manual/zh/function.setcookie.php
1 parent
11977f4f
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
42 additions
and
10 deletions
context.go
context.go
View file @
f1e5059
package
beego
import
(
"bytes"
"fmt"
"mime"
"net/http"
"strings"
"time"
)
type
Context
struct
{
...
...
@@ -37,7 +37,7 @@ func (ctx *Context) NotFound(message string) {
ctx
.
ResponseWriter
.
Write
([]
byte
(
message
))
}
//Sets the content type by extension, as defined in the mime package.
//Sets the content type by extension, as defined in the mime package.
//For example, ctx.ContentType("json") sets the content-type to "application/json"
func
(
ctx
*
Context
)
ContentType
(
ext
string
)
{
if
!
strings
.
HasPrefix
(
ext
,
"."
)
{
...
...
@@ -58,14 +58,46 @@ func (ctx *Context) SetHeader(hdr string, val string, unique bool) {
}
//Sets a cookie -- duration is the amount of time in seconds. 0 = forever
func
(
ctx
*
Context
)
SetCookie
(
name
string
,
value
string
,
age
int64
)
{
var
utctime
time
.
Time
if
age
==
0
{
// 2^31 - 1 seconds (roughly 2038)
utctime
=
time
.
Unix
(
2147483647
,
0
)
//params:
//string name
//string value
//int64 expire = 0
//string $path
//string $domain
//bool $secure = false
//bool $httponly = false
func
(
ctx
*
Context
)
SetCookie
(
name
string
,
value
string
,
others
...
interface
{})
{
var
b
bytes
.
Buffer
fmt
.
Fprintf
(
&
b
,
"%s=%s"
,
sanitizeName
(
name
),
sanitizeValue
(
value
))
if
len
(
others
)
>
0
{
fmt
.
Fprintf
(
&
b
,
"; Max-Age=%d"
,
others
[
0
]
.
(
int64
))
}
else
{
utctime
=
time
.
Unix
(
time
.
Now
()
.
Unix
()
+
age
,
0
)
fmt
.
Fprintf
(
&
b
,
"; Max-Age=0"
)
}
if
len
(
others
)
>
1
{
fmt
.
Fprintf
(
&
b
,
"; Path=%s"
,
sanitizeValue
(
others
[
1
]
.
(
string
)))
}
if
len
(
others
)
>
2
{
fmt
.
Fprintf
(
&
b
,
"; Domain=%s"
,
sanitizeValue
(
others
[
2
]
.
(
string
)))
}
cookie
:=
fmt
.
Sprintf
(
"%s=%s; Expires=%s; Path=/"
,
name
,
value
,
webTime
(
utctime
))
ctx
.
SetHeader
(
"Set-Cookie"
,
cookie
,
true
)
if
len
(
others
)
>
3
{
fmt
.
Fprintf
(
&
b
,
"; Secure"
)
}
if
len
(
others
)
>
4
{
fmt
.
Fprintf
(
&
b
,
"; HttpOnly"
)
}
ctx
.
SetHeader
(
"Set-Cookie"
,
b
.
String
(),
true
)
}
var
cookieNameSanitizer
=
strings
.
NewReplacer
(
"
\n
"
,
"-"
,
"
\r
"
,
"-"
)
func
sanitizeName
(
n
string
)
string
{
return
cookieNameSanitizer
.
Replace
(
n
)
}
var
cookieValueSanitizer
=
strings
.
NewReplacer
(
"
\n
"
,
" "
,
"
\r
"
,
" "
,
";"
,
" "
)
func
sanitizeValue
(
v
string
)
string
{
return
cookieValueSanitizer
.
Replace
(
v
)
}
...
...
Write
Preview
Styling with
Markdown
is supported
Attach a file
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to post a comment