mirror of
https://github.com/QingdaoU/JudgeServer.git
synced 2025-01-15 21:12:54 +00:00
46 lines
830 B
Go
46 lines
830 B
Go
package judge
|
||
|
||
import (
|
||
"time"
|
||
"crypto/sha256"
|
||
"encoding/hex"
|
||
)
|
||
|
||
type options struct {
|
||
// scheme://host:port .
|
||
EndpointURL string
|
||
Token string
|
||
sha256Token string
|
||
// 请求超时时间
|
||
// 如果 Timeout 为 0，那么意味着不会超时
|
||
Timeout time.Duration
|
||
}
|
||
|
||
var DefaultOptions = &options{
|
||
EndpointURL: "http://127.0.0.1:12358",
|
||
Token: "YOUR_TOKEN_HERE",
|
||
Timeout: 10 * time.Second,
|
||
}
|
||
|
||
type Option func(*options)
|
||
|
||
func WithEndpointURL(u string) Option {
|
||
return func(o *options) {
|
||
o.EndpointURL = u
|
||
}
|
||
}
|
||
|
||
func WithToken(token string) Option {
|
||
return func(o *options) {
|
||
o.Token = token
|
||
sha256Token := sha256.Sum256([]byte(token))
|
||
o.sha256Token = hex.EncodeToString(sha256Token[:])
|
||
}
|
||
}
|
||
|
||
func WithTimeout(timeout time.Duration) Option {
|
||
return func(o *options) {
|
||
o.Timeout = timeout
|
||
}
|
||
}
|