2023-01-13 17:17:17 +01:00
package requests
import (
2023-05-27 23:54:14 +02:00
ct "blackforestbytes.com/simplecloudnotifier/db/cursortoken"
2023-01-13 17:17:17 +01:00
"blackforestbytes.com/simplecloudnotifier/models"
"context"
"gogs.mikescher.com/BlackForestBytes/goext/sq"
"time"
)
2023-05-28 22:27:38 +02:00
func ( db * Database ) InsertRequestLog ( ctx context . Context , requestid models . RequestID , data models . RequestLog ) ( models . RequestLog , error ) {
2023-01-13 17:17:17 +01:00
2023-05-28 22:27:38 +02:00
entity := data . DB ( )
entity . RequestID = requestid
entity . TimestampCreated = time2DB ( time . Now ( ) )
2023-01-13 17:17:17 +01:00
2023-05-28 22:27:38 +02:00
_ , err := sq . InsertSingle ( ctx , db . db , "requests" , entity )
2023-01-13 17:17:17 +01:00
if err != nil {
2023-05-28 22:27:38 +02:00
return models . RequestLog { } , err
2023-01-13 17:17:17 +01:00
}
2023-05-28 22:27:38 +02:00
return entity . Model ( ) , nil
2023-01-13 17:17:17 +01:00
}
func ( db * Database ) Cleanup ( ctx context . Context , count int , duration time . Duration ) ( int64 , error ) {
2023-01-16 18:53:22 +01:00
res1 , err := db . db . Exec ( ctx , "DELETE FROM requests WHERE request_id NOT IN ( SELECT request_id FROM requests ORDER BY timestamp_created DESC LIMIT :keep ) " , sq . PP {
"keep" : count ,
2023-01-13 17:17:17 +01:00
} )
if err != nil {
return 0 , err
}
affected1 , err := res1 . RowsAffected ( )
if err != nil {
return 0 , err
}
res2 , err := db . db . Exec ( ctx , "DELETE FROM requests WHERE timestamp_created < :tslim" , sq . PP {
"tslim" : time . Now ( ) . Add ( - duration ) . UnixMilli ( ) ,
} )
if err != nil {
return 0 , err
}
affected2 , err := res2 . RowsAffected ( )
if err != nil {
return 0 , err
}
return affected1 + affected2 , nil
}
2023-05-27 23:54:14 +02:00
func ( db * Database ) ListRequestLogs ( ctx context . Context , filter models . RequestLogFilter , pageSize * int , inTok ct . CursorToken ) ( [ ] models . RequestLog , ct . CursorToken , error ) {
if inTok . Mode == ct . CTMEnd {
return make ( [ ] models . RequestLog , 0 ) , ct . End ( ) , nil
}
pageCond := "1=1"
if inTok . Mode == ct . CTMNormal {
pageCond = "timestamp_created < :tokts OR (timestamp_created = :tokts AND request_id < :tokid )"
}
filterCond , filterJoin , prepParams , err := filter . SQL ( )
orderClause := ""
if pageSize != nil {
orderClause = "ORDER BY timestamp_created DESC, request_id DESC LIMIT :lim"
prepParams [ "lim" ] = * pageSize + 1
} else {
orderClause = "ORDER BY timestamp_created DESC, request_id DESC"
}
sqlQuery := "SELECT " + "requests.*" + " FROM requests " + filterJoin + " WHERE ( " + pageCond + " ) AND ( " + filterCond + " ) " + orderClause
prepParams [ "tokts" ] = inTok . Timestamp
prepParams [ "tokid" ] = inTok . Id
rows , err := db . db . Query ( ctx , sqlQuery , prepParams )
if err != nil {
return nil , ct . CursorToken { } , err
}
2024-05-31 23:56:16 +02:00
data , err := models . DecodeRequestLogs ( ctx , db . db , rows )
2023-05-27 23:54:14 +02:00
if err != nil {
return nil , ct . CursorToken { } , err
}
if pageSize == nil || len ( data ) <= * pageSize {
return data , ct . End ( ) , nil
} else {
outToken := ct . Normal ( data [ * pageSize - 1 ] . TimestampCreated , data [ * pageSize - 1 ] . RequestID . String ( ) , "DESC" , filter . Hash ( ) )
return data [ 0 : * pageSize ] , outToken , nil
}
}