drop sqlite3 and use .log, optimize some code

This commit is contained in:
lzx
2026-01-29 16:51:06 +08:00
parent 194b062bb9
commit e9c4582e0d
47 changed files with 668 additions and 212 deletions

View File

@@ -1,185 +1,240 @@
import 'dart:async';
// import 'dart:io';
import 'dart:convert';
import 'dart:io';
import 'package:logging/logging.dart';
import 'package:sqflite_common_ffi/sqflite_ffi.dart';
import 'mining_task_info.dart';
import '../utils/path_utils.dart';
/// 数据库管理服务
/// 挖矿任务持久化服务(基于 .log 文件,而非 SQLite
///
/// 设计约定:
/// - 使用 `bin/mining_tasks.log` 记录当前(或最近)挖矿任务,一行一条 JSON 记录。
/// - 每次收到新的挖矿任务时,追加一条记录。
/// - 挖矿任务完成后,从 .log 中删除对应记录。
/// - 客户端启动时读取 .log
/// - 如果存在任务且 `endTimestamp` 尚未过期,则恢复该任务;
/// - 如果已过期,则删除该记录,并不恢复。
class DatabaseService {
static final DatabaseService _instance = DatabaseService._internal();
factory DatabaseService() => _instance;
DatabaseService._internal();
final Logger _logger = Logger('DatabaseService');
Database? _database;
/// 初始化数据库
/// 日志文件路径bin/mining_tasks.log
File get _logFile => File(PathUtils.binFile('mining_tasks.log'));
/// 初始化(确保 bin 目录存在)
Future<void> initialize() async {
try {
// Windows/Desktop: 使用 sqflite_common_ffi
sqfliteFfiInit();
databaseFactory = databaseFactoryFfi;
// 确保 bin 目录存在
final binDir = Directory(PathUtils.binDir);
if (!await binDir.exists()) {
await binDir.create(recursive: true);
}
// 与 Go 版一致,尽量落到 ./bin/mining_task.db
final dbPath = PathUtils.binFile('mining_task.db');
_database = await databaseFactory.openDatabase(
dbPath,
options: OpenDatabaseOptions(
version: 1,
onCreate: (db, version) async {
await db.execute('''
CREATE TABLE mining_tasks (
id INTEGER PRIMARY KEY AUTOINCREMENT,
coin TEXT NOT NULL,
algo TEXT NOT NULL,
pool TEXT NOT NULL,
pool_url TEXT NOT NULL,
wallet_address TEXT NOT NULL,
worker_id TEXT NOT NULL,
pool_user TEXT,
wallet_mining INTEGER NOT NULL,
end_timestamp INTEGER NOT NULL,
miner TEXT NOT NULL,
status TEXT NOT NULL,
created_at INTEGER NOT NULL,
updated_at INTEGER NOT NULL
)
''');
},
),
);
_logger.info('数据库初始化成功');
// 日志文件可懒创建,不强制在这里创建
_logger.info('任务日志初始化完成(使用文件存储,不再使用 SQLite');
} catch (e) {
_logger.severe('数据库初始化失败: $e');
_logger.severe('任务日志初始化失败: $e');
rethrow;
}
}
/// 插入挖矿任务
/// 插入挖矿任务(在 .log 文件中追加一条记录)
Future<int> insertMiningTask(MiningTaskInfo task) async {
if (_database == null) {
await initialize();
}
await initialize();
try {
final now = DateTime.now().millisecondsSinceEpoch ~/ 1000;
return await _database!.insert(
'mining_tasks',
{
'coin': task.coin,
'algo': task.algo,
'pool': task.pool,
'pool_url': task.poolUrl,
'wallet_address': task.walletAddress,
'worker_id': task.workerId,
'pool_user': task.poolUser,
'wallet_mining': task.walletMining ? 1 : 0,
'end_timestamp': task.endTimestamp,
'miner': task.miner,
'status': 'running',
'created_at': now,
'updated_at': now,
},
);
final record = <String, dynamic>{
'coin': task.coin,
'algo': task.algo,
'pool': task.pool,
'pool_url': task.poolUrl,
'wallet_address': task.walletAddress,
'worker_id': task.workerId,
'pool_user': task.poolUser,
'wallet_mining': task.walletMining,
'end_timestamp': task.endTimestamp,
'miner': task.miner,
'created_at': now,
};
final jsonLine = jsonEncode(record);
await _logFile.writeAsString('$jsonLine\n', mode: FileMode.append, flush: true);
return 1; // 返回值目前未被使用,保持兼容即可
} catch (e) {
_logger.severe('插入挖矿任务失败: $e');
rethrow;
}
}
/// 完成挖矿任务
Future<void> finishMiningTask(int taskId) async {
if (_database == null) {
await initialize();
}
/// 挖矿任务完成后,从 .log 文件中删除该任务
Future<void> finishMiningTask(MiningTaskInfo task) async {
await initialize();
try {
final now = DateTime.now().millisecondsSinceEpoch ~/ 1000;
await _database!.update(
'mining_tasks',
{
'status': 'finished',
'updated_at': now,
},
where: 'id = ?',
whereArgs: [taskId],
);
if (!await _logFile.exists()) {
return;
}
final lines = await _logFile.readAsLines();
if (lines.isEmpty) return;
final List<String> keptLines = [];
for (final line in lines) {
if (line.trim().isEmpty) continue;
try {
final Map<String, dynamic> data = jsonDecode(line) as Map<String, dynamic>;
final existing = _taskFromJson(data);
// 如果与当前任务匹配,则跳过(即删除)
if (_isSameTask(existing, task)) {
continue;
}
keptLines.add(line);
} catch (_) {
// 解析失败的行保留,避免误删
keptLines.add(line);
}
}
await _logFile.writeAsString(keptLines.join('\n') + (keptLines.isEmpty ? '' : '\n'));
} catch (e) {
_logger.severe('完成挖矿任务失败: $e');
_logger.severe('完成挖矿任务(从日志中删除)失败: $e');
}
}
/// 加载未完成的挖矿任务
///
/// - 读取 .log 中所有任务;
/// - 删除已过期endTimestamp <= now的任务
/// - 返回最新的、尚未过期的任务(如果有)。
Future<MiningTaskInfo?> loadUnfinishedTask() async {
if (_database == null) {
await initialize();
}
await initialize();
try {
final results = await _database!.query(
'mining_tasks',
where: 'status = ?',
whereArgs: ['running'],
orderBy: 'created_at DESC',
limit: 1,
);
if (results.isEmpty) {
if (!await _logFile.exists()) {
return null;
}
final row = results.first;
final currentTime = DateTime.now().millisecondsSinceEpoch ~/ 1000;
final endTimestamp = row['end_timestamp'] as int;
if (currentTime >= endTimestamp) {
// 任务已过期,标记为完成
await finishMiningTask(row['id'] as int);
final lines = await _logFile.readAsLines();
if (lines.isEmpty) {
return null;
}
return MiningTaskInfo(
coin: row['coin'] as String,
algo: row['algo'] as String,
pool: row['pool'] as String,
poolUrl: row['pool_url'] as String,
walletAddress: row['wallet_address'] as String,
workerId: row['worker_id'] as String,
poolUser: row['pool_user'] as String?,
walletMining: (row['wallet_mining'] as int) == 1,
endTimestamp: endTimestamp,
miner: row['miner'] as String,
);
final now = DateTime.now().millisecondsSinceEpoch ~/ 1000;
final List<_StoredTask> validTasks = [];
for (final line in lines) {
if (line.trim().isEmpty) continue;
try {
final Map<String, dynamic> data = jsonDecode(line) as Map<String, dynamic>;
final task = _taskFromJson(data);
final createdAt = (data['created_at'] as int?) ?? task.endTimestamp;
// 过滤掉已过期任务
if (now >= task.endTimestamp) {
continue;
}
validTasks.add(_StoredTask(task: task, createdAt: createdAt));
} catch (e) {
_logger.warning('解析任务日志行失败,已跳过: $e, 原始行: $line');
}
}
// 重新写回仅包含未过期的任务
if (validTasks.isEmpty) {
await _logFile.writeAsString('');
return null;
} else {
// 按创建时间排序,取最新一条作为恢复任务
validTasks.sort((a, b) => b.createdAt.compareTo(a.createdAt));
final toKeep = validTasks;
final buffer = StringBuffer();
for (final t in toKeep) {
final record = _taskToJson(t.task, createdAt: t.createdAt);
buffer.writeln(jsonEncode(record));
}
await _logFile.writeAsString(buffer.toString());
return validTasks.first.task;
}
} catch (e) {
_logger.severe('加载挖矿任务失败: $e');
return null;
}
}
/// 获取任务历史
/// 获取任务历史(目前基于 .log 仅保存“当前/最近”任务,这里返回空列表以保持接口兼容)
Future<List<Map<String, dynamic>>> getTaskHistory({int limit = 100}) async {
if (_database == null) {
await initialize();
}
try {
return await _database!.query(
'mining_tasks',
orderBy: 'created_at DESC',
limit: limit,
);
} catch (e) {
_logger.severe('获取任务历史失败: $e');
return [];
}
// 如有需要,可以在未来扩展为持久化历史记录
return [];
}
/// 关闭数据库
/// 关闭(对文件存储无实际操作,保留接口以兼容旧代码)
Future<void> close() async {
await _database?.close();
_database = null;
// no-op
}
// === 内部工具方法 ===
MiningTaskInfo _taskFromJson(Map<String, dynamic> data) {
return MiningTaskInfo(
coin: data['coin'] as String,
algo: data['algo'] as String,
pool: (data['pool'] as String?) ?? '',
poolUrl: data['pool_url'] as String,
walletAddress: data['wallet_address'] as String,
workerId: data['worker_id'] as String,
poolUser: data['pool_user'] as String?,
walletMining: (data['wallet_mining'] as bool?) ??
((data['wallet_mining'] is int) ? (data['wallet_mining'] as int) == 1 : false),
endTimestamp: data['end_timestamp'] as int,
miner: data['miner'] as String,
);
}
Map<String, dynamic> _taskToJson(MiningTaskInfo task, {required int createdAt}) {
return <String, dynamic>{
'coin': task.coin,
'algo': task.algo,
'pool': task.pool,
'pool_url': task.poolUrl,
'wallet_address': task.walletAddress,
'worker_id': task.workerId,
'pool_user': task.poolUser,
'wallet_mining': task.walletMining,
'end_timestamp': task.endTimestamp,
'miner': task.miner,
'created_at': createdAt,
};
}
bool _isSameTask(MiningTaskInfo a, MiningTaskInfo b) {
return a.coin == b.coin &&
a.algo == b.algo &&
a.pool == b.pool &&
a.poolUrl == b.poolUrl &&
a.walletAddress == b.walletAddress &&
a.workerId == b.workerId &&
(a.poolUser ?? '') == (b.poolUser ?? '') &&
a.walletMining == b.walletMining &&
a.endTimestamp == b.endTimestamp &&
a.miner == b.miner;
}
}
class _StoredTask {
final MiningTaskInfo task;
final int createdAt;
_StoredTask({required this.task, required this.createdAt});
}

View File

@@ -231,6 +231,7 @@ class ClientProvider with ChangeNotifier {
/// 挖矿任务变化回调
void _onMiningTaskChanged(MiningTaskInfo? task) async {
final previousTask = _currentMiningTask;
_currentMiningTask = task;
if (task != null) {
@@ -246,6 +247,10 @@ class ClientProvider with ChangeNotifier {
} else {
// 停止挖矿
await _miningManager.stopMining();
// 挖矿任务完成后,从日志中删除该任务
if (previousTask != null) {
await _database.finishMiningTask(previousTask);
}
// 恢复持续挖矿
await _sustainMiner.resume();