|
| 1 | +package internal |
| 2 | + |
| 3 | +import ( |
| 4 | + "fmt" |
| 5 | + "strings" |
| 6 | + "time" |
| 7 | + |
| 8 | + "qiniu1024-mcp-server/pkg/common/config" |
| 9 | +) |
| 10 | + |
| 11 | +// IndexManager Elasticsearch索引管理器 |
| 12 | +type IndexManager struct { |
| 13 | + pattern string // 索引模式 |
| 14 | + dateFormat string // 日期格式 |
| 15 | + wildcardPattern string // 通配符模式 |
| 16 | +} |
| 17 | + |
| 18 | +// NewIndexManager 创建索引管理器 |
| 19 | +func NewIndexManager() *IndexManager { |
| 20 | + // 从全局配置获取索引配置 |
| 21 | + esConfig := config.GlobalConfig.ElasticSearch.Index |
| 22 | + |
| 23 | + // 设置默认值 |
| 24 | + pattern := esConfig.Pattern |
| 25 | + if pattern == "" { |
| 26 | + pattern = "mock-{service}-logs-{date}" |
| 27 | + } |
| 28 | + |
| 29 | + dateFormat := esConfig.DateFormat |
| 30 | + if dateFormat == "" { |
| 31 | + dateFormat = "2006.01.02" |
| 32 | + } |
| 33 | + |
| 34 | + wildcardPattern := esConfig.WildcardPattern |
| 35 | + if wildcardPattern == "" { |
| 36 | + wildcardPattern = "mock-*-logs-{date}" |
| 37 | + } |
| 38 | + |
| 39 | + return &IndexManager{ |
| 40 | + pattern: pattern, |
| 41 | + dateFormat: dateFormat, |
| 42 | + wildcardPattern: wildcardPattern, |
| 43 | + } |
| 44 | +} |
| 45 | + |
| 46 | +// GenerateIndexName 生成具体的索引名称 |
| 47 | +// service: 服务名称 |
| 48 | +// date: 日期，如果为空则使用当前日期 |
| 49 | +func (im *IndexManager) GenerateIndexName(service string, date time.Time) string { |
| 50 | + if date.IsZero() { |
| 51 | + date = time.Now() |
| 52 | + } |
| 53 | + |
| 54 | + dateStr := date.Format(im.dateFormat) |
| 55 | + |
| 56 | + // 替换模式中的占位符 |
| 57 | + indexName := im.pattern |
| 58 | + indexName = strings.ReplaceAll(indexName, "{service}", service) |
| 59 | + indexName = strings.ReplaceAll(indexName, "{date}", dateStr) |
| 60 | + |
| 61 | + return indexName |
| 62 | +} |
| 63 | + |
| 64 | +// GenerateWildcardPattern 生成通配符模式 |
| 65 | +// date: 日期，如果为空则使用当前日期 |
| 66 | +func (im *IndexManager) GenerateWildcardPattern(date time.Time) string { |
| 67 | + if date.IsZero() { |
| 68 | + date = time.Now() |
| 69 | + } |
| 70 | + |
| 71 | + dateStr := date.Format(im.dateFormat) |
| 72 | + |
| 73 | + // 替换模式中的占位符 |
| 74 | + pattern := im.wildcardPattern |
| 75 | + pattern = strings.ReplaceAll(pattern, "{date}", dateStr) |
| 76 | + |
| 77 | + return pattern |
| 78 | +} |
| 79 | + |
| 80 | +// GenerateIndexPatternsForDateRange 为日期范围生成索引模式列表 |
| 81 | +// startDate: 开始日期 |
| 82 | +// endDate: 结束日期 |
| 83 | +// service: 服务名称，如果为空则使用通配符 |
| 84 | +func (im *IndexManager) GenerateIndexPatternsForDateRange(startDate, endDate time.Time, service string) []string { |
| 85 | + var patterns []string |
| 86 | + |
| 87 | + // 计算日期范围内的所有日期 |
| 88 | + currentDate := startDate |
| 89 | + for !currentDate.After(endDate) { |
| 90 | + if service == "" { |
| 91 | + // 使用通配符模式 |
| 92 | + pattern := im.GenerateWildcardPattern(currentDate) |
| 93 | + patterns = append(patterns, pattern) |
| 94 | + } else { |
| 95 | + // 使用具体服务模式 |
| 96 | + indexName := im.GenerateIndexName(service, currentDate) |
| 97 | + patterns = append(patterns, indexName) |
| 98 | + } |
| 99 | + |
| 100 | + // 移动到下一天 |
| 101 | + currentDate = currentDate.AddDate(0, 0, 1) |
| 102 | + } |
| 103 | + |
| 104 | + return patterns |
| 105 | +} |
| 106 | + |
| 107 | +// GenerateIndexPatternsForTimeRange 为时间范围生成索引模式列表 |
| 108 | +// startTime: 开始时间字符串，格式为 "2025-08-20T00:00:00Z" |
| 109 | +// endTime: 结束时间字符串，格式为 "2025-08-20T23:59:59Z" |
| 110 | +// service: 服务名称，如果为空则使用通配符 |
| 111 | +func (im *IndexManager) GenerateIndexPatternsForTimeRange(startTime, endTime string, service string) ([]string, error) { |
| 112 | + // 解析时间字符串 |
| 113 | + startDate, err := time.Parse(time.RFC3339, startTime) |
| 114 | + if err != nil { |
| 115 | + return nil, fmt.Errorf("解析开始时间失败: %w", err) |
| 116 | + } |
| 117 | + |
| 118 | + endDate, err := time.Parse(time.RFC3339, endTime) |
| 119 | + if err != nil { |
| 120 | + return nil, fmt.Errorf("解析结束时间失败: %w", err) |
| 121 | + } |
| 122 | + |
| 123 | + // 提取日期部分（去掉时间） |
| 124 | + startDate = time.Date(startDate.Year(), startDate.Month(), startDate.Day(), 0, 0, 0, 0, startDate.Location()) |
| 125 | + endDate = time.Date(endDate.Year(), endDate.Month(), endDate.Day(), 0, 0, 0, 0, endDate.Location()) |
| 126 | + |
| 127 | + return im.GenerateIndexPatternsForDateRange(startDate, endDate, service), nil |
| 128 | +} |
| 129 | + |
| 130 | +// GetCurrentIndexPattern 获取当前日期的索引模式 |
| 131 | +// service: 服务名称，如果为空则使用通配符 |
| 132 | +func (im *IndexManager) GetCurrentIndexPattern(service string) string { |
| 133 | + if service == "" { |
| 134 | + return im.GenerateWildcardPattern(time.Time{}) // 使用当前日期 |
| 135 | + } |
| 136 | + return im.GenerateIndexName(service, time.Time{}) // 使用当前日期 |
| 137 | +} |
| 138 | + |
| 139 | +// ValidateIndexPattern 验证索引模式是否有效 |
| 140 | +func (im *IndexManager) ValidateIndexPattern(pattern string) error { |
| 141 | + if pattern == "" { |
| 142 | + return fmt.Errorf("索引模式不能为空") |
| 143 | + } |
| 144 | + |
| 145 | + // 检查是否包含必要的占位符 |
| 146 | + if !strings.Contains(pattern, "{date}") { |
| 147 | + return fmt.Errorf("索引模式必须包含{date}占位符") |
| 148 | + } |
| 149 | + |
| 150 | + return nil |
| 151 | +} |
| 152 | + |
| 153 | +// GetIndexConfig 获取索引配置信息 |
| 154 | +func (im *IndexManager) GetIndexConfig() map[string]interface{} { |
| 155 | + return map[string]interface{}{ |
| 156 | + "pattern": im.pattern, |
| 157 | + "date_format": im.dateFormat, |
| 158 | + "wildcard_pattern": im.wildcardPattern, |
| 159 | + } |
| 160 | +} |
0 commit comments