mirror of https://github.com/Xhofe/alist
53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
package halalcloud
|
|
|
|
import "google.golang.org/grpc"
|
|
|
|
func defaultOptions() halalOptions {
|
|
return halalOptions{
|
|
// onRefreshTokenRefreshed: func(string) {},
|
|
grpcOptions: []grpc.DialOption{
|
|
grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(1024 * 1024 * 32)),
|
|
// grpc.WithMaxMsgSize(1024 * 1024 * 1024),
|
|
},
|
|
}
|
|
}
|
|
|
|
type HalalOption interface {
|
|
apply(*halalOptions)
|
|
}
|
|
|
|
// halalOptions configure a RPC call. halalOptions are set by the HalalOption
|
|
// values passed to Dial.
|
|
type halalOptions struct {
|
|
onTokenRefreshed func(accessToken string, accessTokenExpiredAt int64, refreshToken string, refreshTokenExpiredAt int64)
|
|
grpcOptions []grpc.DialOption
|
|
}
|
|
|
|
// funcDialOption wraps a function that modifies halalOptions into an
|
|
// implementation of the DialOption interface.
|
|
type funcDialOption struct {
|
|
f func(*halalOptions)
|
|
}
|
|
|
|
func (fdo *funcDialOption) apply(do *halalOptions) {
|
|
fdo.f(do)
|
|
}
|
|
|
|
func newFuncDialOption(f func(*halalOptions)) *funcDialOption {
|
|
return &funcDialOption{
|
|
f: f,
|
|
}
|
|
}
|
|
|
|
func WithRefreshTokenRefreshedCallback(s func(accessToken string, accessTokenExpiredAt int64, refreshToken string, refreshTokenExpiredAt int64)) HalalOption {
|
|
return newFuncDialOption(func(o *halalOptions) {
|
|
o.onTokenRefreshed = s
|
|
})
|
|
}
|
|
|
|
func WithGrpcDialOptions(opts ...grpc.DialOption) HalalOption {
|
|
return newFuncDialOption(func(o *halalOptions) {
|
|
o.grpcOptions = opts
|
|
})
|
|
}
|