You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.1 KiB
46 lines
1.1 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"encoding/hex" |
|
"fmt" |
|
"time" |
|
|
|
"go-common/app/service/main/passport-auth/model" |
|
xsql "go-common/library/database/sql" |
|
"go-common/library/log" |
|
) |
|
|
|
const ( |
|
_getRefreshSQL = "SELECT mid,appid,refresh,token,expires FROM user_refresh_%s WHERE refresh = ? limit 1" |
|
) |
|
|
|
// Refresh get token by access_token |
|
func (d *Dao) Refresh(c context.Context, rk []byte, ct time.Time) (res *model.Refresh, err error) { |
|
row := d.db.QueryRow(c, fmt.Sprintf(_getRefreshSQL, formatRefreshSuffix(ct)), rk) |
|
res = new(model.Refresh) |
|
var refresh, token []byte |
|
if err = row.Scan(&res.Mid, &res.AppID, &refresh, &token, &res.Expires); err != nil { |
|
if err == xsql.ErrNoRows { |
|
res = nil |
|
err = nil |
|
} else { |
|
log.Error("row.Scan() error(%v)", err) |
|
} |
|
return |
|
} |
|
res.Refresh = hex.EncodeToString(refresh) |
|
res.Token = hex.EncodeToString(token) |
|
return |
|
} |
|
|
|
func formatRefreshSuffix(t time.Time) string { |
|
return formatByDate(t.Year(), int(t.Month())) |
|
} |
|
|
|
func formatByDate(year, month int) string { |
|
if month%2 == 1 { |
|
return fmt.Sprintf("%4d%02d", year, month) |
|
} |
|
return fmt.Sprintf("%4d%02d", year, month-1) |
|
}
|
|
|