change module name 3

pkg/linedb/line_db.go

@@ -171,6 +171,81 @@ func (db *LineDb) Init(force bool, initOptions *LineDbInitOptions) error {
 		go db.indexRebuildTimerLoop(interval)
 	}
 
+	if err := db.seedLastIDsFromData(dbFolder); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// seedLastIDsFromData выставляет LastIDManager по максимальному id в существующих данных
+// (после рестарта процесса счётчик не начинается с 1).
+func (db *LineDb) seedLastIDsFromData(dbFolder string) error {
+	for i, opt := range db.initOptions.Collections {
+		collectionName := opt.CollectionName
+		if collectionName == "" {
+			collectionName = fmt.Sprintf("collection_%d", i+1)
+		}
+		if db.isCollectionPartitioned(collectionName) {
+			if err := db.seedLastIDPartitioned(dbFolder, collectionName); err != nil {
+				return err
+			}
+			continue
+		}
+		adapter := db.adapters[collectionName]
+		if adapter == nil {
+			continue
+		}
+		records, err := adapter.Read(LineDbAdapterOptions{})
+		if err != nil {
+			return fmt.Errorf("seed last id: read %s: %w", collectionName, err)
+		}
+		db.lastIDManager.SetLastID(collectionName, db.GetMaxID(records))
+	}
+	return nil
+}
+
+// seedLastIDPartitioned — максимальный id по базовому файлу и всем events_*.jsonl на диске.
+// NextID для партиций вызывается с логическим именем (например events), ключ тот же.
+func (db *LineDb) seedLastIDPartitioned(dbFolder, baseName string) error {
+	maxID := 0
+	if a := db.adapters[baseName]; a != nil {
+		recs, err := a.Read(LineDbAdapterOptions{})
+		if err != nil {
+			return fmt.Errorf("seed last id: read base %s: %w", baseName, err)
+		}
+		if m := db.GetMaxID(recs); m > maxID {
+			maxID = m
+		}
+	}
+	pattern := filepath.Join(dbFolder, baseName+"_*.jsonl")
+	matches, err := filepath.Glob(pattern)
+	if err != nil {
+		return fmt.Errorf("seed last id: glob %s: %w", pattern, err)
+	}
+	for _, path := range matches {
+		base := filepath.Base(path)
+		partName := strings.TrimSuffix(base, ".jsonl")
+		var adapter *JSONLFile
+		if existing, ok := db.adapters[partName]; ok {
+			adapter = existing
+		} else {
+			adapter = NewJSONLFile(path, "", JSONLFileOptions{CollectionName: partName})
+			if err := adapter.Init(false, LineDbAdapterOptions{}); err != nil {
+				continue
+			}
+			db.adapters[partName] = adapter
+			db.collections[partName] = path
+		}
+		recs, err := adapter.Read(LineDbAdapterOptions{})
+		if err != nil {
+			continue
+		}
+		if m := db.GetMaxID(recs); m > maxID {
+			maxID = m
+		}
+	}
+	db.lastIDManager.SetLastID(baseName, maxID)
 	return nil
 }
 
@@ -1285,8 +1360,9 @@ func (db *LineDb) GetMaxID(records []any) int {
 	for _, record := range records {
 		if recordMap, ok := record.(map[string]any); ok {
 			if id, ok := recordMap["id"]; ok {
-				if idInt, ok := id.(int); ok && idInt > maxID {
-					maxID = idInt
+				n := idToIntForMax(id)
+				if n > maxID {
+					maxID = n
 				}
 			}
 		}
@@ -1294,6 +1370,21 @@ func (db *LineDb) GetMaxID(records []any) int {
 	return maxID
 }
 
+func idToIntForMax(id any) int {
+	switch v := id.(type) {
+	case int:
+		return v
+	case int64:
+		return int(v)
+	case float64:
+		return int(v)
+	case float32:
+		return int(v)
+	default:
+		return 0
+	}
+}
+
 func (db *LineDb) matchesFilter(record any, filter any, strictCompare bool) bool {
 	if recordMap, ok := record.(map[string]any); ok {
 		if filterMap, ok := filter.(map[string]any); ok {
@@ -1383,10 +1474,32 @@ func (db *LineDb) isInvalidID(id any) bool {
 	if id == nil {
 		return true
 	}
-	if idNum, ok := id.(int); ok {
-		return idNum <= -1
+	switch v := id.(type) {
+	case int:
+		return v <= -1
+	case int8:
+		return v <= -1
+	case int16:
+		return v <= -1
+	case int32:
+		return v <= -1
+	case int64:
+		return v <= -1
+	case uint, uint8, uint16, uint32, uint64:
+		return false
+	case float32:
+		if v != v { // NaN
+			return true
+		}
+		return float64(v) <= -1
+	case float64:
+		if v != v { // NaN
+			return true
+		}
+		return v <= -1
+	default:
+		return false
 	}
-	return false
 }
 
 func (db *LineDb) compareIDs(a, b any) bool {