70 lines
2.8 KiB
Go
70 lines
2.8 KiB
Go
package cassandradb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/authorizerdev/authorizer/server/db/models"
|
|
"github.com/authorizerdev/authorizer/server/graph/model"
|
|
"github.com/gocql/gocql"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// AddWebhookLog to add webhook log
|
|
func (p *provider) AddWebhookLog(ctx context.Context, webhookLog *models.WebhookLog) (*model.WebhookLog, error) {
|
|
if webhookLog.ID == "" {
|
|
webhookLog.ID = uuid.New().String()
|
|
}
|
|
|
|
webhookLog.Key = webhookLog.ID
|
|
webhookLog.CreatedAt = time.Now().Unix()
|
|
webhookLog.UpdatedAt = time.Now().Unix()
|
|
|
|
insertQuery := fmt.Sprintf("INSERT INTO %s (id, http_status, response, request, webhook_id, created_at, updated_at) VALUES ('%s', %d,'%s', '%s', '%s', %d, %d)", KeySpace+"."+models.Collections.WebhookLog, webhookLog.ID, webhookLog.HttpStatus, webhookLog.Response, webhookLog.Request, webhookLog.WebhookID, webhookLog.CreatedAt, webhookLog.UpdatedAt)
|
|
err := p.db.Query(insertQuery).Exec()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return webhookLog.AsAPIWebhookLog(), nil
|
|
}
|
|
|
|
// ListWebhookLogs to list webhook logs
|
|
func (p *provider) ListWebhookLogs(ctx context.Context, pagination *model.Pagination, webhookID string) (*model.WebhookLogs, error) {
|
|
webhookLogs := []*model.WebhookLog{}
|
|
paginationClone := pagination
|
|
totalCountQuery := fmt.Sprintf(`SELECT COUNT(*) FROM %s`, KeySpace+"."+models.Collections.WebhookLog)
|
|
// there is no offset in cassandra
|
|
// so we fetch till limit + offset
|
|
// and return the results from offset to limit
|
|
query := fmt.Sprintf("SELECT id, http_status, response, request, webhook_id, created_at, updated_at FROM %s LIMIT %d", KeySpace+"."+models.Collections.WebhookLog, pagination.Limit+pagination.Offset)
|
|
if webhookID != "" {
|
|
totalCountQuery = fmt.Sprintf(`SELECT COUNT(*) FROM %s WHERE webhook_id='%s' ALLOW FILTERING`, KeySpace+"."+models.Collections.WebhookLog, webhookID)
|
|
query = fmt.Sprintf("SELECT id, http_status, response, request, webhook_id, created_at, updated_at FROM %s WHERE webhook_id = '%s' LIMIT %d ALLOW FILTERING", KeySpace+"."+models.Collections.WebhookLog, webhookID, pagination.Limit+pagination.Offset)
|
|
}
|
|
|
|
err := p.db.Query(totalCountQuery).Consistency(gocql.One).Scan(&paginationClone.Total)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
scanner := p.db.Query(query).Iter().Scanner()
|
|
counter := int64(0)
|
|
for scanner.Next() {
|
|
if counter >= pagination.Offset {
|
|
var webhookLog models.WebhookLog
|
|
err := scanner.Scan(&webhookLog.ID, &webhookLog.HttpStatus, &webhookLog.Response, &webhookLog.Request, &webhookLog.WebhookID, &webhookLog.CreatedAt, &webhookLog.UpdatedAt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
webhookLogs = append(webhookLogs, webhookLog.AsAPIWebhookLog())
|
|
}
|
|
counter++
|
|
}
|
|
|
|
return &model.WebhookLogs{
|
|
Pagination: paginationClone,
|
|
WebhookLogs: webhookLogs,
|
|
}, nil
|
|
}
|