Merge pull request #49 from LyricTian/develop

update refreshing token config
go-oauth2 · May 8, 2017 · 38556ca · 38556ca
2 parents 4c69c58 + d85b5f3
commit 38556ca
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/manage/config.go b/manage/config.go
@@ -14,6 +14,10 @@ type Config struct {
 
 // RefreshingConfig refreshing token config
 type RefreshingConfig struct {
+	// access token expiration time
+	AccessTokenExp time.Duration
+	// refresh token expiration time
+	RefreshTokenExp time.Duration
 	// whether to generate the refreshing token
 	IsGenerateRefresh bool
 	// whether to reset the refreshing create time

diff --git a/manage/manager.go b/manage/manager.go
@@ -373,6 +373,14 @@ func (m *Manager) RefreshAccessToken(tgr *oauth2.TokenGenerateRequest) (accessTo
 		ti.SetAccess(tv)
 		ti.SetAccessCreateAt(td.CreateAt)
 
+		if v := rcfg.AccessTokenExp; v > 0 {
+			ti.SetAccessExpiresIn(v)
+		}
+
+		if v := rcfg.RefreshTokenExp; v > 0 {
+			ti.SetRefreshExpiresIn(v)
+		}
+
 		if rcfg.IsResetRefreshTime {
 			ti.SetRefreshCreateAt(td.CreateAt)
 		}