feat: update CreateTempToken,add CreateTempTokenByStyle

This commit is contained in:
weloe
2023-10-28 00:35:02 +08:00
parent 046128c502
commit 7b920a1a59
3 changed files with 29 additions and 3 deletions

View File

@@ -208,7 +208,7 @@ func (e *Enforcer) CloseSafe(token string, service string) error {
return nil
}
func (e *Enforcer) CreateTempToken(style string, service string, value string, timeout int64) (string, error) {
func (e *Enforcer) CreateTempTokenByStyle(style string, service string, value string, timeout int64) (string, error) {
token, err := e.generateFunc.Exec(style)
if err != nil {
return "", err
@@ -220,6 +220,17 @@ func (e *Enforcer) CreateTempToken(style string, service string, value string, t
return token, nil
}
func (e *Enforcer) CreateTempToken(token string, service string, value string, timeout int64) (string, error) {
if token == "" {
return "", fmt.Errorf("token must not be empty")
}
err := e.setTempToken(service, token, value, timeout)
if err != nil {
return "", err
}
return token, nil
}
func (e *Enforcer) GetTempTokenTimeout(service string, tempToken string) int64 {
if tempToken == "" {
return constant.NotValueExpire