update last_updated when calling the API with a token
This commit is contained in:
parent
dd536eebed
commit
256f082340
|
@ -74,6 +74,9 @@ func Start(conf common.Conf, dbO *sqlx.DB) *gin.Engine {
|
||||||
DB: conf.RedisDB,
|
DB: conf.RedisDB,
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// token updater
|
||||||
|
go tokenUpdater(db)
|
||||||
|
|
||||||
api := r.Group("/api")
|
api := r.Group("/api")
|
||||||
{
|
{
|
||||||
p := api.Group("/")
|
p := api.Group("/")
|
||||||
|
|
|
@ -4,20 +4,20 @@ import (
|
||||||
"crypto/md5"
|
"crypto/md5"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"time"
|
||||||
"github.com/jmoiron/sqlx"
|
|
||||||
|
|
||||||
"git.zxq.co/ripple/rippleapi/common"
|
"git.zxq.co/ripple/rippleapi/common"
|
||||||
|
"github.com/jmoiron/sqlx"
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetTokenFull retrieves an user ID and their token privileges knowing their API token.
|
// GetTokenFull retrieves an user ID and their token privileges knowing their API token.
|
||||||
func GetTokenFull(token string, db *sqlx.DB) (common.Token, bool) {
|
func GetTokenFull(token string, db *sqlx.DB) (common.Token, bool) {
|
||||||
var t common.Token
|
|
||||||
var (
|
var (
|
||||||
|
t common.Token
|
||||||
tokenPrivsRaw uint64
|
tokenPrivsRaw uint64
|
||||||
userPrivsRaw uint64
|
userPrivsRaw uint64
|
||||||
|
priv8 bool
|
||||||
)
|
)
|
||||||
var priv8 bool
|
|
||||||
err := db.QueryRow(`SELECT
|
err := db.QueryRow(`SELECT
|
||||||
t.id, t.user, t.privileges, t.private, u.privileges
|
t.id, t.user, t.privileges, t.private, u.privileges
|
||||||
FROM tokens t
|
FROM tokens t
|
||||||
|
@ -27,6 +27,7 @@ WHERE token = ? LIMIT 1`,
|
||||||
Scan(
|
Scan(
|
||||||
&t.ID, &t.UserID, &tokenPrivsRaw, &priv8, &userPrivsRaw,
|
&t.ID, &t.UserID, &tokenPrivsRaw, &priv8, &userPrivsRaw,
|
||||||
)
|
)
|
||||||
|
updateTokens <- t.ID
|
||||||
if priv8 {
|
if priv8 {
|
||||||
tokenPrivsRaw = common.PrivilegeReadConfidential | common.PrivilegeWrite
|
tokenPrivsRaw = common.PrivilegeReadConfidential | common.PrivilegeWrite
|
||||||
}
|
}
|
||||||
|
@ -42,3 +43,43 @@ WHERE token = ? LIMIT 1`,
|
||||||
return t, true
|
return t, true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var updateTokens = make(chan int, 100)
|
||||||
|
|
||||||
|
func tokenUpdater(db *sqlx.DB) {
|
||||||
|
for {
|
||||||
|
// prepare a queue of tokens to update.
|
||||||
|
tokensToUpdate := make([]int, 0, 50)
|
||||||
|
AwaitLoop:
|
||||||
|
for {
|
||||||
|
// if we got ten, move on and update
|
||||||
|
if len(tokensToUpdate) >= 50 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
// if we ain't got any, add what we get straight from updateTokens
|
||||||
|
if len(tokensToUpdate) == 0 {
|
||||||
|
x := <-updateTokens
|
||||||
|
tokensToUpdate = append(tokensToUpdate, x)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// otherwise, wait from updateTokens with a timeout of 10 seconds
|
||||||
|
select {
|
||||||
|
case x := <-updateTokens:
|
||||||
|
tokensToUpdate = append(tokensToUpdate, x)
|
||||||
|
case <-time.After(10 * time.Second):
|
||||||
|
// wondering what this means?
|
||||||
|
// https://golang.org/ref/spec#Break_statements
|
||||||
|
break AwaitLoop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
q, a, _ := sqlx.In("UPDATE tokens SET last_updated = ? WHERE id IN (?)", time.Now().Unix(), tokensToUpdate)
|
||||||
|
|
||||||
|
q = db.Rebind(q)
|
||||||
|
_, err := db.Exec(q, a...)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -4,6 +4,7 @@ import (
|
||||||
"crypto/md5"
|
"crypto/md5"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/jmoiron/sqlx"
|
"github.com/jmoiron/sqlx"
|
||||||
|
|
||||||
|
@ -121,7 +122,8 @@ func TokenNewPOST(md common.MethodData) common.CodeMessager {
|
||||||
return Err500
|
return Err500
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_, err = md.DB.Exec("INSERT INTO tokens(user, privileges, description, token, private) VALUES (?, ?, ?, ?, '0')", r.ID, r.Privileges, data.Description, tokenMD5)
|
_, err = md.DB.Exec("INSERT INTO tokens(user, privileges, description, token, private, last_updated) VALUES (?, ?, ?, ?, '0', ?)",
|
||||||
|
r.ID, r.Privileges, data.Description, tokenMD5, time.Now().Unix())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
md.Err(err)
|
md.Err(err)
|
||||||
return Err500
|
return Err500
|
||||||
|
@ -146,9 +148,10 @@ func TokenSelfDeletePOST(md common.MethodData) common.CodeMessager {
|
||||||
}
|
}
|
||||||
|
|
||||||
type token struct {
|
type token struct {
|
||||||
ID int `json:"id"`
|
ID int `json:"id"`
|
||||||
Privileges uint64 `json:"privileges"`
|
Privileges uint64 `json:"privileges"`
|
||||||
Description string `json:"description"`
|
Description string `json:"description"`
|
||||||
|
LastUpdated common.UnixTimestamp `json:"last_updated"`
|
||||||
}
|
}
|
||||||
type tokenResponse struct {
|
type tokenResponse struct {
|
||||||
common.ResponseBase
|
common.ResponseBase
|
||||||
|
@ -157,14 +160,14 @@ type tokenResponse struct {
|
||||||
|
|
||||||
// TokenGET retrieves a list listing all the user's public tokens.
|
// TokenGET retrieves a list listing all the user's public tokens.
|
||||||
func TokenGET(md common.MethodData) common.CodeMessager {
|
func TokenGET(md common.MethodData) common.CodeMessager {
|
||||||
rows, err := md.DB.Query("SELECT id, privileges, description FROM tokens WHERE user = ? AND private = '0'", md.ID())
|
rows, err := md.DB.Query("SELECT id, privileges, description, last_updated FROM tokens WHERE user = ? AND private = '0'", md.ID())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return Err500
|
return Err500
|
||||||
}
|
}
|
||||||
var r tokenResponse
|
var r tokenResponse
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var t token
|
var t token
|
||||||
err = rows.Scan(&t.ID, &t.Privileges, &t.Description)
|
err = rows.Scan(&t.ID, &t.Privileges, &t.Description, &t.LastUpdated)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
md.Err(err)
|
md.Err(err)
|
||||||
continue
|
continue
|
||||||
|
@ -187,9 +190,9 @@ func TokenSelfGET(md common.MethodData) common.CodeMessager {
|
||||||
}
|
}
|
||||||
var r tokenSingleResponse
|
var r tokenSingleResponse
|
||||||
// md.User.ID = token id, userid would have been md.User.UserID. what a clusterfuck
|
// md.User.ID = token id, userid would have been md.User.UserID. what a clusterfuck
|
||||||
err := md.DB.QueryRow("SELECT id, privileges, description FROM tokens WHERE id = ? "+
|
err := md.DB.QueryRow("SELECT id, privileges, description, last_updated FROM tokens WHERE id = ? "+
|
||||||
common.Paginate(md.Query("p"), md.Query("l"), 50), md.User.ID).Scan(
|
common.Paginate(md.Query("p"), md.Query("l"), 50), md.User.ID).Scan(
|
||||||
&r.ID, &r.Privileges, &r.Description,
|
&r.ID, &r.Privileges, &r.Description, &r.LastUpdated,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
md.Err(err)
|
md.Err(err)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user