github/tai

公开

Tai · Windows 软件与网站使用时长统计。上游 Planshit/Tai 的源码镜像，非灵栈原创；MIT 许可。来源：https://github.com/Planshit/Tai 。本次为手动同步，暂不提供安装包。

完善数据写入逻辑

作者: noberumotto
时间: 2022/10/24 15:10
提交: 398eae6
上一个: 0e48e52

修改Core/Servicers/Instances/Data.cs+115 −68看文件

@@ -19,103 +19,150 @@ namespace Core.Servicers.Instances
19	19	public class Data : IData
20	20	{
21	21	private readonly IAppData appData;
	22	新增： private readonly object setLock = new object();
22	23	public Data(IAppData appData)
23	24	{
24	25	this.appData = appData;
25	26	}
26	27	public void Set(string processName, int seconds, DateTime? time_ = null)
27	28	{
	29	新增： lock (setLock)
	30	新增： {
	31	新增： DateTime time = !time_.HasValue ? DateTime.Now : time_.Value;
	32	新增： var today = time.Date;
	33	新增：
	34	新增： if (string.IsNullOrEmpty(processName) \|\| seconds <= 0)
	35	新增： {
	36	新增： return;
	37	新增： }
	38	新增：
	39	新增： AppModel app = appData.GetApp(processName);
	40	新增：
	41	新增： if (app == null)
	42	新增： {
	43	新增： return;
	44	新增： }
	45	新增： // 统计app累计使用时长
	46	新增： app.TotalTime += seconds;
	47	新增： appData.UpdateApp(app);
	48	新增：
	49	新增： using (var db = new TaiDbContext())
	50	新增： {
	51	新增： using (var transcation = db.Database.BeginTransaction())
	52	新增： {
	53	新增： try
	54	新增： {
	55	新增： var res = db.DailyLog.SingleOrDefault(m => m.Date == today && m.AppModelID == app.ID);
	56	新增： if (res == null)
	57	新增： {
	58	新增： //数据库中没有时则创建
	59	新增： db.DailyLog.Add(new Models.DailyLogModel()
	60	新增： {
	61	新增： Date = today,
	62	新增： AppModelID = app.ID,
	63	新增： Time = seconds,
	64	新增： });
	65	新增： }
	66	新增： else
	67	新增： {
	68	新增： res.Time += seconds;
	69	新增： }
	70	新增：
	71	新增： db.SaveChanges();
	72	新增： transcation.Commit();
	73	新增： }
	74	新增： catch (Exception e)
	75	新增： {
	76	新增： Logger.Error(e.ToString());
	77	新增： transcation.Rollback();
	78	新增： }
	79	新增： }
	80	新增： }
	81	新增：
	82	新增： SetHoursTime(app, seconds, time);
	83	新增： }
	84	新增： }
28	85
29		删除： DateTime time = !time_.HasValue ? DateTime.Now : time_.Value;
30		删除： var today = time.Date;
31	86
	87	新增： private void SetHoursTime(AppModel app, int seconds, DateTime time)
	88	新增： {
	89	新增： string processName = app.Name;
32	90	// 当前时段
33		删除： var nowtime = new DateTime(today.Year, today.Month, today.Day, time.Hour, 0, 0);
	91	新增： var nowtime = new DateTime(time.Year, time.Month, time.Day, time.Hour, 0, 0);
34	92
35		删除： if (string.IsNullOrEmpty(processName) \|\| seconds <= 0)
	93	新增： if (time.Minute == 59 && time.Second == 59)
36	94	{
	95	新增： SetHoursTime(app, seconds, nowtime.AddHours(1));
37	96	return;
38	97	}
39	98
40		删除： AppModel app = appData.GetApp(processName);
41		删除：
42		删除： if (app == null)
	99	新增： if (seconds <= 0 \|\| time > DateTime.Now)
43	100	{
44	101	return;
45	102	}
46	103
47		删除：
	104	新增：
48	105
49	106	using (var db = new TaiDbContext())
50	107	{
51		删除： var res = db.DailyLog.SingleOrDefault(m => m.Date == today && m.AppModelID == app.ID);
52		删除： if (res == null)
53		删除： {
54		删除： //数据库中没有时则创建
55		删除： db.DailyLog.Add(new Models.DailyLogModel()
56		删除： {
57		删除： Date = today,
58		删除： AppModelID = app.ID,
59		删除： Time = seconds,
60		删除： });
61		删除： }
62		删除： else
63		删除： {
64		删除： res.Time += seconds;
65		删除： }
66		删除：
67		删除： // 分时段记录数据
68		删除：
69		删除： var hourslog = db.HoursLog.SingleOrDefault(
70		删除： m =>
71		删除： m.DataTime == nowtime
72		删除： && m.AppModelID == app.ID
73		删除： );
74		删除： if (hourslog == null)
75		删除： {
76		删除： // 没有时创建
77		删除：
78		删除： if (seconds > 3600)
79		删除： {
80		删除： int overflowSeconds = seconds - 3600;
81		删除： Set(processName, overflowSeconds, nowtime.AddHours(1));
82		删除：
83		删除： seconds = 3600;
84		删除： }
85		删除：
86		删除： db.HoursLog.Add(new Models.HoursLogModel()
87		删除： {
88		删除： DataTime = nowtime,
89		删除： AppModelID = app.ID,
90		删除： Time = seconds
91		删除： });
92		删除： }
93		删除： else
	108	新增： using (var transcation = db.Database.BeginTransaction())
94	109	{
95		删除： if (hourslog.Time + seconds > 3600)
	110	新增： try
96	111	{
97		删除： hourslog.Time = 3600;
98		删除：
99		删除： int overflowSeconds = hourslog.Time + seconds - 3600;
100		删除：
101		删除： Set(processName, overflowSeconds, nowtime.AddHours(1));
	112	新增： var hourslog = db.HoursLog.SingleOrDefault(
	113	新增： m =>
	114	新增： m.DataTime == nowtime
	115	新增： && m.AppModelID == app.ID
	116	新增： );
	117	新增：
	118	新增： int overflowSeconds = 0;
	119	新增：
	120	新增： if (hourslog == null)
	121	新增： {
	122	新增： // 没有时创建
	123	新增：
	124	新增： if (seconds > 3600)
	125	新增： {
	126	新增： overflowSeconds = seconds - 3600;
	127	新增： seconds = 3600;
	128	新增： }
	129	新增：
	130	新增： db.HoursLog.Add(new Models.HoursLogModel()
	131	新增： {
	132	新增： DataTime = nowtime,
	133	新增： AppModelID = app.ID,
	134	新增： Time = seconds
	135	新增： });
	136	新增： }
	137	新增： else
	138	新增： {
	139	新增： if (hourslog.Time + seconds > 3600)
	140	新增： {
	141	新增： hourslog.Time = 3600;
	142	新增： overflowSeconds = hourslog.Time + seconds - 3600;
	143	新增： }
	144	新增： else
	145	新增： {
	146	新增： hourslog.Time += seconds;
	147	新增： }
	148	新增： }
	149	新增：
	150	新增： db.SaveChanges();
	151	新增： transcation.Commit();
	152	新增：
	153	新增： if (overflowSeconds > 0)
	154	新增： {
	155	新增： SetHoursTime(app, overflowSeconds, nowtime.AddHours(1));
	156	新增： }
102	157	}
103		删除： else
	158	新增： catch (Exception e)
104	159	{
105		删除： hourslog.Time += seconds;
	160	新增： Logger.Error(e.ToString());
	161	新增： transcation.Rollback();
106	162	}
107	163	}
108		删除：
109		删除： // 统计使用时长
110		删除： app.TotalTime += seconds;
111		删除：
112		删除： appData.UpdateApp(app);
113		删除：
114		删除： db.SaveChanges();
115	164	}
116	165	}
117		删除：
118		删除：
119	166	public List<DailyLogModel> GetTodaylogList()
120	167	{
121	168