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.
48 lines
1.7 KiB
48 lines
1.7 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"database/sql" |
|
"fmt" |
|
"time" |
|
|
|
"go-common/app/job/main/passport-auth/model" |
|
"go-common/library/log" |
|
) |
|
|
|
const ( |
|
_addCookieSQL = "INSERT IGNORE INTO user_cookie_%s (mid,session,csrf,type,expires) VALUES (?,?,?,?,?)" |
|
_delCookieBySessionSQL = "DELETE FROM user_cookie_%s where session = ?" |
|
|
|
_addCookieDeletedSQL = "INSERT IGNORE INTO user_cookie_deleted_%s (mid,session,csrf,type,expires,ctime) VALUES (?,?,?,?,?,?)" |
|
) |
|
|
|
// AddCookie save cookie |
|
func (d *Dao) AddCookie(c context.Context, cookie *model.Cookie, session, csrf []byte, ct time.Time) (affected int64, err error) { |
|
row, err := d.db.Exec(c, fmt.Sprintf(_addCookieSQL, formatSuffix(ct)), cookie.Mid, session, csrf, cookie.Type, cookie.Expires) |
|
if err != nil { |
|
log.Error("dao.db.Exec(%v) err(%v)", cookie, err) |
|
return |
|
} |
|
return row.RowsAffected() |
|
} |
|
|
|
// DelCookie del cookie by session |
|
func (d *Dao) DelCookie(c context.Context, session []byte, ct time.Time) (affected int64, err error) { |
|
var res sql.Result |
|
if res, err = d.db.Exec(c, fmt.Sprintf(_delCookieBySessionSQL, formatSuffix(ct)), session); err != nil { |
|
log.Error("del cookie by session , dao.db.Exec(%s) error(%v)", session, err) |
|
return |
|
} |
|
return res.RowsAffected() |
|
} |
|
|
|
// AddCookieDeleted save cookie deleted |
|
func (d *Dao) AddCookieDeleted(c context.Context, cookie *model.Cookie, session, csrf []byte, ct time.Time) (affected int64, err error) { |
|
row, err := d.db.Exec(c, fmt.Sprintf(_addCookieDeletedSQL, formatSuffix(ct)), cookie.Mid, session, csrf, cookie.Type, cookie.Expires, cookie.Ctime) |
|
if err != nil { |
|
log.Error("fail to add cookie deleted, cookie(%+v), tx.Exec() error(%+v)", cookie, err) |
|
return |
|
} |
|
return row.RowsAffected() |
|
}
|
|
|