| | | 1 | | using Elsa.Diagnostics.StructuredLogs.Models; |
| | | 2 | | using Elsa.Diagnostics.StructuredLogs.Options; |
| | | 3 | | using Elsa.Diagnostics.StructuredLogs.Persistence.Relational.Contracts; |
| | | 4 | | using Microsoft.Extensions.Options; |
| | | 5 | | |
| | | 6 | | namespace Elsa.Diagnostics.StructuredLogs.Persistence.Relational.Services; |
| | | 7 | | |
| | 25 | 8 | | public class RelationalStructuredLogSqlBuilder(IRelationalStructuredLogDialect dialect, IOptions<StructuredLogsOptions> |
| | | 9 | | { |
| | 25 | 10 | | private readonly string _table = dialect.QuoteIdentifier("StructuredLogEvents"); |
| | | 11 | | |
| | | 12 | | public string BuildInsert() |
| | | 13 | | { |
| | 1 | 14 | | var columns = Columns; |
| | 1 | 15 | | var columnList = string.Join(", ", columns.Select(dialect.QuoteIdentifier)); |
| | 21 | 16 | | var parameterList = string.Join(", ", columns.Select(x => $"{dialect.ParameterPrefix}{x}")); |
| | 1 | 17 | | return $"INSERT INTO {_table} ({columnList}) VALUES ({parameterList})"; |
| | | 18 | | } |
| | | 19 | | |
| | | 20 | | public QueryDefinition BuildQuery(StructuredLogFilter filter) |
| | | 21 | | { |
| | 14 | 22 | | var parameters = new Dictionary<string, object?>(); |
| | 14 | 23 | | var predicates = BuildFilterPredicates(filter, parameters); |
| | 14 | 24 | | var where = predicates.Count == 0 ? "" : $" WHERE {string.Join(" AND ", predicates)}"; |
| | 14 | 25 | | var limit = options.Value.ClampRecentLogQueryTake(filter.Take); |
| | 14 | 26 | | var sql = $"SELECT {string.Join(", ", Columns.Select(dialect.QuoteIdentifier))} FROM {_table}{where} ORDER BY {d |
| | 14 | 27 | | sql = dialect.ApplyLimit(sql, limit); |
| | 14 | 28 | | return new(sql, parameters); |
| | | 29 | | } |
| | | 30 | | |
| | | 31 | | public string BuildListSources() |
| | | 32 | | { |
| | 1 | 33 | | var sourceId = dialect.QuoteIdentifier("SourceId"); |
| | 1 | 34 | | var receivedAt = dialect.QuoteIdentifier("ReceivedAt"); |
| | 1 | 35 | | return $"SELECT {sourceId}, MAX({receivedAt}) AS {dialect.QuoteIdentifier("LastSeen")} FROM {_table} GROUP BY {s |
| | | 36 | | } |
| | | 37 | | |
| | | 38 | | public QueryDefinition BuildDeleteOlderThan(string cutoff) |
| | | 39 | | { |
| | 1 | 40 | | return new($"DELETE FROM {_table} WHERE {dialect.QuoteIdentifier("ReceivedAt")} < {dialect.ParameterPrefix}Cutof |
| | | 41 | | } |
| | | 42 | | |
| | | 43 | | public QueryDefinition BuildDeleteRowsBeyondMax(int maxRows) |
| | | 44 | | { |
| | 1 | 45 | | var id = dialect.QuoteIdentifier("Id"); |
| | 1 | 46 | | var receivedAt = dialect.QuoteIdentifier("ReceivedAt"); |
| | 1 | 47 | | var sequence = dialect.QuoteIdentifier("Sequence"); |
| | 1 | 48 | | var selectSql = $"SELECT {id} FROM {_table} ORDER BY {receivedAt} DESC, {sequence} DESC, {id} DESC"; |
| | 1 | 49 | | var sql = $"DELETE FROM {_table} WHERE {id} IN ({dialect.ApplyOffset(selectSql, maxRows)})"; |
| | 1 | 50 | | return new(sql, new Dictionary<string, object?>()); |
| | | 51 | | } |
| | | 52 | | |
| | | 53 | | private List<string> BuildFilterPredicates(StructuredLogFilter filter, IDictionary<string, object?> parameters) |
| | | 54 | | { |
| | 14 | 55 | | var predicates = new List<string>(); |
| | | 56 | | |
| | 14 | 57 | | if (filter.MinimumLevel is { } minimumLevel) |
| | 1 | 58 | | AddPredicate(predicates, parameters, "Level", ">=", (int)minimumLevel); |
| | | 59 | | |
| | 14 | 60 | | if (filter.Levels is { Count: > 0 }) |
| | | 61 | | { |
| | 1 | 62 | | var names = filter.Levels.Select((level, index) => |
| | 1 | 63 | | { |
| | 2 | 64 | | var name = $"Level{index}"; |
| | 2 | 65 | | parameters[name] = (int)level; |
| | 2 | 66 | | return $"{dialect.ParameterPrefix}{name}"; |
| | 1 | 67 | | }); |
| | 1 | 68 | | predicates.Add($"{dialect.QuoteIdentifier("Level")} IN ({string.Join(", ", names)})"); |
| | | 69 | | } |
| | | 70 | | |
| | 14 | 71 | | if (!string.IsNullOrWhiteSpace(filter.CategoryPrefix)) |
| | 1 | 72 | | AddPredicate(predicates, parameters, "Category", "LIKE", $"{filter.CategoryPrefix}%"); |
| | | 73 | | |
| | 14 | 74 | | if (!string.IsNullOrWhiteSpace(filter.Text)) |
| | | 75 | | { |
| | 2 | 76 | | parameters["Text"] = $"%{filter.Text}%"; |
| | 2 | 77 | | var textParameter = $"{dialect.ParameterPrefix}Text"; |
| | 2 | 78 | | predicates.Add($"({dialect.QuoteIdentifier("Message")} LIKE {textParameter} OR {dialect.QuoteIdentifier("Mes |
| | | 79 | | } |
| | | 80 | | |
| | 14 | 81 | | AddStringPredicate(predicates, parameters, "TenantId", filter.TenantId); |
| | 14 | 82 | | AddStringPredicate(predicates, parameters, "WorkflowDefinitionId", filter.WorkflowDefinitionId); |
| | 14 | 83 | | AddStringPredicate(predicates, parameters, "WorkflowInstanceId", filter.WorkflowInstanceId); |
| | 14 | 84 | | AddStringPredicate(predicates, parameters, "TraceId", filter.TraceId); |
| | 14 | 85 | | AddStringPredicate(predicates, parameters, "SpanId", filter.SpanId); |
| | 14 | 86 | | AddStringPredicate(predicates, parameters, "CorrelationId", filter.CorrelationId); |
| | 14 | 87 | | AddStringPredicate(predicates, parameters, "SourceId", filter.SourceId); |
| | | 88 | | |
| | 14 | 89 | | if (filter.From is { } from) |
| | 1 | 90 | | AddPredicate(predicates, parameters, "Timestamp", ">=", RelationalStructuredLogMapper.FormatTimestamp(from), |
| | | 91 | | |
| | 14 | 92 | | if (filter.To is { } to) |
| | 1 | 93 | | AddPredicate(predicates, parameters, "Timestamp", "<=", RelationalStructuredLogMapper.FormatTimestamp(to), " |
| | | 94 | | |
| | 14 | 95 | | return predicates; |
| | | 96 | | } |
| | | 97 | | |
| | | 98 | | private void AddStringPredicate(ICollection<string> predicates, IDictionary<string, object?> parameters, string colu |
| | | 99 | | { |
| | 98 | 100 | | if (string.IsNullOrWhiteSpace(value)) |
| | 89 | 101 | | return; |
| | | 102 | | |
| | 9 | 103 | | AddPredicate(predicates, parameters, column, "=", value); |
| | 9 | 104 | | } |
| | | 105 | | |
| | | 106 | | private void AddPredicate(ICollection<string> predicates, IDictionary<string, object?> parameters, string column, st |
| | | 107 | | { |
| | 13 | 108 | | var name = parameterName ?? column; |
| | 13 | 109 | | predicates.Add($"{dialect.QuoteIdentifier(column)} {op} {dialect.ParameterPrefix}{name}"); |
| | 13 | 110 | | parameters[name] = value; |
| | 13 | 111 | | } |
| | | 112 | | |
| | 1 | 113 | | private static readonly string[] Columns = |
| | 1 | 114 | | [ |
| | 1 | 115 | | "Id", |
| | 1 | 116 | | "Sequence", |
| | 1 | 117 | | "Timestamp", |
| | 1 | 118 | | "ReceivedAt", |
| | 1 | 119 | | "Level", |
| | 1 | 120 | | "Category", |
| | 1 | 121 | | "EventId", |
| | 1 | 122 | | "EventName", |
| | 1 | 123 | | "Message", |
| | 1 | 124 | | "MessageTemplate", |
| | 1 | 125 | | "ExceptionJson", |
| | 1 | 126 | | "ScopesJson", |
| | 1 | 127 | | "PropertiesJson", |
| | 1 | 128 | | "TraceId", |
| | 1 | 129 | | "SpanId", |
| | 1 | 130 | | "CorrelationId", |
| | 1 | 131 | | "TenantId", |
| | 1 | 132 | | "WorkflowDefinitionId", |
| | 1 | 133 | | "WorkflowInstanceId", |
| | 1 | 134 | | "SourceId" |
| | 1 | 135 | | ]; |
| | | 136 | | } |
| | | 137 | | |
| | | 138 | | public record QueryDefinition(string Sql, IReadOnlyDictionary<string, object?> Parameters); |