You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

user_data_analysis.go 24 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602
  1. package repo
  2. import (
  3. "fmt"
  4. "net/http"
  5. "net/url"
  6. "time"
  7. "code.gitea.io/gitea/models"
  8. "code.gitea.io/gitea/modules/context"
  9. "code.gitea.io/gitea/modules/git"
  10. "code.gitea.io/gitea/modules/log"
  11. "code.gitea.io/gitea/modules/setting"
  12. "code.gitea.io/gitea/services/mailer"
  13. "github.com/360EntSecGroup-Skylar/excelize/v2"
  14. )
  15. const (
  16. PAGE_SIZE = 2000
  17. )
  18. func getUserMetricsExcelHeader(ctx *context.Context) map[string]string {
  19. excelHeader := make([]string, 0)
  20. excelHeader = append(excelHeader, ctx.Tr("user.metrics.date"))
  21. excelHeader = append(excelHeader, ctx.Tr("user.metrics.newregistuser"))
  22. excelHeader = append(excelHeader, ctx.Tr("user.metrics.newregistandactiveuser"))
  23. excelHeader = append(excelHeader, ctx.Tr("user.metrics.hasactivateuser"))
  24. excelHeader = append(excelHeader, ctx.Tr("user.metrics.newregistnotactiveuser"))
  25. excelHeader = append(excelHeader, ctx.Tr("user.metrics.averageuser"))
  26. excelHeader = append(excelHeader, ctx.Tr("user.metrics.newuseractiveindex"))
  27. excelHeader = append(excelHeader, ctx.Tr("user.metrics.totalregistuser"))
  28. excelHeader = append(excelHeader, ctx.Tr("user.metrics.totalactiveduser"))
  29. excelHeader = append(excelHeader, ctx.Tr("user.metrics.totalhasactivityuser"))
  30. excelHeaderMap := make(map[string]string, 0)
  31. var i byte
  32. i = 0
  33. for _, value := range excelHeader {
  34. excelColumn := getColumn(i) + fmt.Sprint(1)
  35. excelHeaderMap[excelColumn] = value
  36. i++
  37. }
  38. return excelHeaderMap
  39. }
  40. func writeUserMetricsExcel(row int, xlsx *excelize.File, sheetName string, userMetrics *models.UserMetrics) {
  41. rows := fmt.Sprint(row)
  42. var tmp byte
  43. tmp = 0
  44. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.DisplayDate)
  45. tmp = tmp + 1
  46. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.ActivateRegistUser+userMetrics.NotActivateRegistUser)
  47. tmp = tmp + 1
  48. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.ActivateRegistUser)
  49. tmp = tmp + 1
  50. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.RegistActivityUser)
  51. tmp = tmp + 1
  52. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.NotActivateRegistUser)
  53. tmp = tmp + 1
  54. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, "")
  55. tmp = tmp + 1
  56. t := userMetrics.ActivateIndex * 100
  57. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", t)+"%")
  58. tmp = tmp + 1
  59. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.TotalUser)
  60. tmp = tmp + 1
  61. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.TotalActivateRegistUser)
  62. tmp = tmp + 1
  63. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userMetrics.TotalHasActivityUser)
  64. }
  65. func getExcelHeader(ctx *context.Context) map[string]string {
  66. excelHeader := make([]string, 0)
  67. excelHeader = append(excelHeader, ctx.Tr("user.static.id"))
  68. excelHeader = append(excelHeader, ctx.Tr("user.static.name"))
  69. excelHeader = append(excelHeader, ctx.Tr("user.static.UserIndex"))
  70. excelHeader = append(excelHeader, ctx.Tr("user.static.UserIndexPrimitive"))
  71. excelHeader = append(excelHeader, ctx.Tr("user.static.codemergecount"))
  72. excelHeader = append(excelHeader, ctx.Tr("user.static.commitcount"))
  73. excelHeader = append(excelHeader, ctx.Tr("user.static.issuecount"))
  74. excelHeader = append(excelHeader, ctx.Tr("user.static.commentcount"))
  75. excelHeader = append(excelHeader, ctx.Tr("user.static.focusrepocount"))
  76. excelHeader = append(excelHeader, ctx.Tr("user.static.starrepocount"))
  77. excelHeader = append(excelHeader, ctx.Tr("user.static.logincount"))
  78. excelHeader = append(excelHeader, ctx.Tr("user.static.watchedcount"))
  79. excelHeader = append(excelHeader, ctx.Tr("user.static.commitcodesize"))
  80. excelHeader = append(excelHeader, ctx.Tr("user.static.solveissuecount"))
  81. excelHeader = append(excelHeader, ctx.Tr("user.static.encyclopediascount"))
  82. excelHeader = append(excelHeader, ctx.Tr("user.static.createrepocount"))
  83. excelHeader = append(excelHeader, ctx.Tr("user.static.openiindex"))
  84. excelHeader = append(excelHeader, ctx.Tr("user.static.CloudBrainTaskNum"))
  85. excelHeader = append(excelHeader, ctx.Tr("user.static.CloudBrainRunTime"))
  86. excelHeader = append(excelHeader, ctx.Tr("user.static.CommitDatasetNum"))
  87. excelHeader = append(excelHeader, ctx.Tr("user.static.CommitModelCount"))
  88. excelHeader = append(excelHeader, ctx.Tr("user.static.FocusOtherUser"))
  89. excelHeader = append(excelHeader, ctx.Tr("user.static.CollectDataset"))
  90. excelHeader = append(excelHeader, ctx.Tr("user.static.CollectedDataset"))
  91. excelHeader = append(excelHeader, ctx.Tr("user.static.RecommendDataset"))
  92. excelHeader = append(excelHeader, ctx.Tr("user.static.CollectImage"))
  93. excelHeader = append(excelHeader, ctx.Tr("user.static.CollectedImage"))
  94. excelHeader = append(excelHeader, ctx.Tr("user.static.RecommendImage"))
  95. excelHeader = append(excelHeader, ctx.Tr("user.static.registdate"))
  96. excelHeader = append(excelHeader, ctx.Tr("user.static.countdate"))
  97. excelHeaderMap := make(map[string]string, 0)
  98. var i byte
  99. i = 0
  100. for _, value := range excelHeader {
  101. excelColumn := getColumn(i) + fmt.Sprint(1)
  102. log.Info("excelColumn=" + excelColumn)
  103. excelHeaderMap[excelColumn] = value
  104. i++
  105. }
  106. return excelHeaderMap
  107. }
  108. func writeExcel(row int, xlsx *excelize.File, sheetName string, userRecord *models.UserBusinessAnalysisAll) {
  109. rows := fmt.Sprint(row)
  110. var tmp byte
  111. tmp = 0
  112. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.ID)
  113. tmp = tmp + 1
  114. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.Name)
  115. tmp = tmp + 1
  116. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", userRecord.UserIndex))
  117. tmp = tmp + 1
  118. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", userRecord.UserIndexPrimitive))
  119. tmp = tmp + 1
  120. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CodeMergeCount)
  121. tmp = tmp + 1
  122. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitCount)
  123. tmp = tmp + 1
  124. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.IssueCount)
  125. tmp = tmp + 1
  126. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommentCount)
  127. tmp = tmp + 1
  128. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.FocusRepoCount)
  129. tmp = tmp + 1
  130. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.StarRepoCount)
  131. tmp = tmp + 1
  132. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.LoginCount)
  133. tmp = tmp + 1
  134. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.WatchedCount)
  135. tmp = tmp + 1
  136. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitCodeSize)
  137. tmp = tmp + 1
  138. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.SolveIssueCount)
  139. tmp = tmp + 1
  140. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.EncyclopediasCount)
  141. tmp = tmp + 1
  142. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CreateRepoCount)
  143. tmp = tmp + 1
  144. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", userRecord.OpenIIndex))
  145. tmp = tmp + 1
  146. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CloudBrainTaskNum)
  147. tmp = tmp + 1
  148. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", float64(userRecord.CloudBrainRunTime)/3600))
  149. tmp = tmp + 1
  150. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitDatasetNum)
  151. tmp = tmp + 1
  152. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitModelCount)
  153. tmp = tmp + 1
  154. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.FocusOtherUser)
  155. tmp = tmp + 1
  156. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectDataset)
  157. tmp = tmp + 1
  158. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectedDataset)
  159. tmp = tmp + 1
  160. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.RecommendDataset)
  161. tmp = tmp + 1
  162. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectImage)
  163. tmp = tmp + 1
  164. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectedImage)
  165. tmp = tmp + 1
  166. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.RecommendImage)
  167. tmp = tmp + 1
  168. formatTime := userRecord.RegistDate.Format("2006-01-02 15:04:05")
  169. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, formatTime[0:len(formatTime)-3])
  170. tmp = tmp + 1
  171. formatTime = userRecord.DataDate
  172. xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, formatTime)
  173. }
  174. func getColumn(tmp byte) string {
  175. var tmpA byte
  176. tmpA = 'A'
  177. if tmp < 26 {
  178. return string(tmpA + tmp)
  179. } else {
  180. return "A" + string(tmpA+(tmp-26))
  181. }
  182. }
  183. func queryUserDataPage(ctx *context.Context, tableName string, queryObj interface{}) {
  184. page := ctx.QueryInt("page")
  185. if page <= 0 {
  186. page = 1
  187. }
  188. pageSize := ctx.QueryInt("pageSize")
  189. if pageSize <= 0 {
  190. pageSize = setting.UI.IssuePagingNum
  191. }
  192. userName := ctx.Query("userName")
  193. IsReturnFile := ctx.QueryBool("IsReturnFile")
  194. if IsReturnFile {
  195. //writer exec file.
  196. xlsx := excelize.NewFile()
  197. sheetName := ctx.Tr("user.static.sheetname")
  198. index := xlsx.NewSheet(sheetName)
  199. xlsx.DeleteSheet("Sheet1")
  200. dataHeader := getExcelHeader(ctx)
  201. for k, v := range dataHeader {
  202. //设置单元格的值
  203. xlsx.SetCellValue(sheetName, k, v)
  204. }
  205. _, count := models.QueryUserStaticDataByTableName(1, 1, tableName, queryObj, userName)
  206. var indexTotal int64
  207. indexTotal = 0
  208. row := 1
  209. for {
  210. re, _ := models.QueryUserStaticDataByTableName(int(indexTotal), PAGE_SIZE, tableName, queryObj, "")
  211. log.Info("return count=" + fmt.Sprint(count))
  212. for _, userRecord := range re {
  213. row++
  214. writeExcel(row, xlsx, sheetName, userRecord)
  215. }
  216. indexTotal += PAGE_SIZE
  217. if indexTotal >= count {
  218. break
  219. }
  220. }
  221. //设置默认打开的表单
  222. xlsx.SetActiveSheet(index)
  223. filename := sheetName + "_" + ctx.Tr("user.static."+tableName) + ".xlsx"
  224. ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(filename))
  225. ctx.Resp.Header().Set("Content-Type", "application/octet-stream")
  226. if _, err := xlsx.WriteTo(ctx.Resp); err != nil {
  227. log.Info("writer exel error." + err.Error())
  228. }
  229. } else {
  230. re, count := models.QueryUserStaticDataByTableName((page-1)*pageSize, pageSize, tableName, queryObj, userName)
  231. mapInterface := make(map[string]interface{})
  232. mapInterface["data"] = re
  233. mapInterface["count"] = count
  234. ctx.JSON(http.StatusOK, mapInterface)
  235. }
  236. }
  237. func queryMetrics(ctx *context.Context, tableName string, startTime time.Time, endTime time.Time) {
  238. page := ctx.QueryInt("page")
  239. if page <= 0 {
  240. page = 1
  241. }
  242. pageSize := ctx.QueryInt("pageSize")
  243. if pageSize <= 0 {
  244. pageSize = setting.UI.IssuePagingNum
  245. }
  246. IsReturnFile := ctx.QueryBool("IsReturnFile")
  247. var count int64
  248. result := make([]*models.UserMetrics, 0)
  249. if tableName == "public.user_business_analysis_current_year" {
  250. result = models.QueryMetricsForYear(startTime.Unix(), endTime.Unix())
  251. count = int64(len(result))
  252. } else if tableName == "public.user_business_analysis_all" {
  253. result = models.QueryMetricsForAll(startTime.Unix(), endTime.Unix())
  254. count = int64(len(result))
  255. } else {
  256. result, count = models.QueryMetricsPage(startTime.Unix(), endTime.Unix(), page, pageSize)
  257. }
  258. if IsReturnFile {
  259. //writer exec file.
  260. xlsx := excelize.NewFile()
  261. sheetName := ctx.Tr("user.metrics.sheetname")
  262. index := xlsx.NewSheet(sheetName)
  263. xlsx.DeleteSheet("Sheet1")
  264. dataHeader := getUserMetricsExcelHeader(ctx)
  265. for k, v := range dataHeader {
  266. //设置单元格的值
  267. xlsx.SetCellValue(sheetName, k, v)
  268. }
  269. row := 1
  270. log.Info("return count=" + fmt.Sprint(count))
  271. for _, userRecord := range result {
  272. row++
  273. writeUserMetricsExcel(row, xlsx, sheetName, userRecord)
  274. }
  275. //设置默认打开的表单
  276. xlsx.SetActiveSheet(index)
  277. filename := sheetName + "_" + ctx.Tr("user.static."+tableName) + ".xlsx"
  278. ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(filename))
  279. ctx.Resp.Header().Set("Content-Type", "application/octet-stream")
  280. if _, err := xlsx.WriteTo(ctx.Resp); err != nil {
  281. log.Info("writer exel error." + err.Error())
  282. }
  283. } else {
  284. mapInterface := make(map[string]interface{})
  285. mapInterface["data"] = result
  286. mapInterface["count"] = count
  287. if tableName == "public.user_business_analysis_yesterday" {
  288. mapInterface["datarecordbegintime"] = setting.RadarMap.GrowthBeginTime
  289. if len(result) > 0 {
  290. dateTime := time.Unix(result[0].CountDate, 0)
  291. mapInterface["lastUpdatedTime"] = dateTime.Format("2006-01-02 15:04:05")
  292. } else {
  293. mapInterface["lastUpdatedTime"] = ""
  294. }
  295. }
  296. ctx.JSON(http.StatusOK, mapInterface)
  297. }
  298. }
  299. func QueryRankingList(ctx *context.Context) {
  300. key := ctx.Query("key")
  301. tableName := ctx.Query("tableName")
  302. limit := ctx.QueryInt("limit")
  303. result, count := models.QueryRankList(key, tableName, limit)
  304. mapInterface := make(map[string]interface{})
  305. mapInterface["data"] = result
  306. mapInterface["count"] = count
  307. ctx.JSON(http.StatusOK, mapInterface)
  308. }
  309. func QueryUserMetricsCurrentMonth(ctx *context.Context) {
  310. currentTimeNow := time.Now()
  311. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  312. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), 1, 0, 0, 0, 0, currentTimeNow.Location())
  313. pageStartTime = getStartTime(pageStartTime)
  314. queryMetrics(ctx, "public.user_business_analysis_current_month", pageStartTime, pageEndTime)
  315. }
  316. func QueryUserStaticCurrentMonth(ctx *context.Context) {
  317. queryUserDataPage(ctx, "public.user_business_analysis_current_month", new(models.UserBusinessAnalysisCurrentMonth))
  318. }
  319. func getStartTime(pageStartTime time.Time) time.Time {
  320. t, _ := time.Parse("2006-01-02", setting.RadarMap.GrowthBeginTime)
  321. if pageStartTime.Before(t) {
  322. pageStartTime = t
  323. }
  324. return pageStartTime
  325. }
  326. func QueryUserMetricsCurrentWeek(ctx *context.Context) {
  327. currentTimeNow := time.Now()
  328. offset := int(time.Monday - currentTimeNow.Weekday())
  329. if offset > 0 {
  330. offset = -6
  331. }
  332. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, offset)
  333. pageStartTime = getStartTime(pageStartTime)
  334. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  335. queryMetrics(ctx, "public.user_business_analysis_current_week", pageStartTime, pageEndTime)
  336. }
  337. func QueryUserStaticCurrentWeek(ctx *context.Context) {
  338. queryUserDataPage(ctx, "public.user_business_analysis_current_week", new(models.UserBusinessAnalysisCurrentWeek))
  339. }
  340. func QueryUserMetricsCurrentYear(ctx *context.Context) {
  341. currentTimeNow := time.Now()
  342. pageStartTime := time.Date(currentTimeNow.Year(), 1, 1, 0, 0, 0, 0, currentTimeNow.Location())
  343. pageStartTime = getStartTime(pageStartTime)
  344. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  345. queryMetrics(ctx, "public.user_business_analysis_current_year", pageStartTime, pageEndTime)
  346. }
  347. func QueryUserStaticCurrentYear(ctx *context.Context) {
  348. queryUserDataPage(ctx, "public.user_business_analysis_current_year", new(models.UserBusinessAnalysisCurrentYear))
  349. }
  350. func QueryUserMetricsLast30Day(ctx *context.Context) {
  351. currentTimeNow := time.Now()
  352. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, -30)
  353. pageStartTime = getStartTime(pageStartTime)
  354. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  355. queryMetrics(ctx, "public.user_business_analysis_last30_day", pageStartTime, pageEndTime)
  356. }
  357. func QueryUserStaticLast30Day(ctx *context.Context) {
  358. queryUserDataPage(ctx, "public.user_business_analysis_last30_day", new(models.UserBusinessAnalysisLast30Day))
  359. }
  360. func QueryUserMetricsLastMonth(ctx *context.Context) {
  361. currentTimeNow := time.Now()
  362. thisMonth := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), 1, 0, 0, 0, 0, currentTimeNow.Location())
  363. pageStartTime := thisMonth.AddDate(0, -1, 0)
  364. pageStartTime = getStartTime(pageStartTime)
  365. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), 1, 23, 59, 59, 0, currentTimeNow.Location()).AddDate(0, 0, -1)
  366. queryMetrics(ctx, "public.user_business_analysis_last_month", pageStartTime, pageEndTime)
  367. }
  368. func QueryUserStaticLastMonth(ctx *context.Context) {
  369. queryUserDataPage(ctx, "public.user_business_analysis_last_month", new(models.UserBusinessAnalysisLastMonth))
  370. }
  371. func QueryUserMetricsYesterday(ctx *context.Context) {
  372. currentTimeNow := time.Now()
  373. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, time.Local)
  374. pageStartTime = getStartTime(pageStartTime)
  375. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  376. queryMetrics(ctx, "public.user_business_analysis_yesterday", pageStartTime, pageEndTime)
  377. }
  378. func QueryUserStaticYesterday(ctx *context.Context) {
  379. queryUserDataPage(ctx, "public.user_business_analysis_yesterday", new(models.UserBusinessAnalysisYesterday))
  380. }
  381. func QueryUserMetricsAll(ctx *context.Context) {
  382. currentTimeNow := time.Now()
  383. pageStartTime := time.Date(2022, 4, 5, 0, 0, 0, 0, currentTimeNow.Location())
  384. pageStartTime = getStartTime(pageStartTime)
  385. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  386. queryMetrics(ctx, "public.user_business_analysis_all", pageStartTime, pageEndTime)
  387. }
  388. func QueryUserStaticAll(ctx *context.Context) {
  389. queryUserDataPage(ctx, "public.user_business_analysis_all", new(models.UserBusinessAnalysisAll))
  390. }
  391. func QueryUserMetricDataPage(ctx *context.Context) {
  392. startDate := ctx.Query("startDate")
  393. endDate := ctx.Query("endDate")
  394. startTime, _ := time.ParseInLocation("2006-01-02", startDate, time.Local)
  395. endTime, _ := time.ParseInLocation("2006-01-02", endDate, time.Local)
  396. page := ctx.QueryInt("page")
  397. if page <= 0 {
  398. page = 1
  399. }
  400. pageSize := ctx.QueryInt("pageSize")
  401. if pageSize <= 0 {
  402. pageSize = setting.UI.IssuePagingNum
  403. }
  404. result, count := models.QueryMetricsPage(startTime.Unix(), endTime.Unix(), page, pageSize)
  405. mapInterface := make(map[string]interface{})
  406. mapInterface["data"] = result
  407. mapInterface["count"] = count
  408. ctx.JSON(http.StatusOK, mapInterface)
  409. }
  410. func QueryUserStaticDataPage(ctx *context.Context) {
  411. startDate := ctx.Query("startDate")
  412. endDate := ctx.Query("endDate")
  413. page := ctx.QueryInt("page")
  414. if page <= 0 {
  415. page = 1
  416. }
  417. pageSize := ctx.QueryInt("pageSize")
  418. if pageSize <= 0 {
  419. pageSize = setting.UI.IssuePagingNum
  420. }
  421. userName := ctx.Query("userName")
  422. IsReturnFile := ctx.QueryBool("IsReturnFile")
  423. log.Info("startDate=" + startDate + " endDate=" + endDate + " userName=" + userName + " page=" + fmt.Sprint(page))
  424. var startTime time.Time
  425. var endTime time.Time
  426. var isAll bool
  427. if startDate == "all" {
  428. isAll = true
  429. startTime = time.Now()
  430. endTime = time.Now()
  431. } else {
  432. startTime, _ = time.ParseInLocation("2006-01-02", startDate, time.Local)
  433. startTime = time.Date(startTime.Year(), startTime.Month(), startTime.Day(), 12, 0, 0, 0, startTime.Location())
  434. settingStartTime, _ := time.Parse("2006-01-02", setting.RadarMap.RecordBeginTime)
  435. if startTime.Unix() < settingStartTime.Unix() {
  436. startTime = settingStartTime
  437. startDate = settingStartTime.Format("2006-01-02")
  438. }
  439. endTime, _ = time.ParseInLocation("2006-01-02", endDate, time.Local)
  440. endTime = endTime.AddDate(0, 0, 1)
  441. endTime = time.Date(endTime.Year(), endTime.Month(), endTime.Day(), 23, 59, 59, 0, startTime.Location())
  442. isAll = false
  443. log.Info("startTime=" + fmt.Sprint(startTime.Unix()) + " endDate=" + fmt.Sprint(endTime.Unix()))
  444. }
  445. if IsReturnFile {
  446. page = -1
  447. pageSize = -1
  448. }
  449. pageOpts := &models.UserBusinessAnalysisQueryOptions{
  450. ListOptions: models.ListOptions{
  451. Page: page,
  452. PageSize: pageSize,
  453. },
  454. UserName: userName,
  455. StartTime: startTime.Unix(),
  456. EndTime: endTime.Unix(),
  457. IsAll: isAll,
  458. }
  459. if IsReturnFile {
  460. re, count := models.QueryUserStaticDataAll(pageOpts)
  461. log.Info("return count=" + fmt.Sprint(count))
  462. //writer exec file.
  463. xlsx := excelize.NewFile()
  464. sheetName := ctx.Tr("user.static.sheetname")
  465. index := xlsx.NewSheet(sheetName)
  466. xlsx.DeleteSheet("Sheet1")
  467. dataHeader := getExcelHeader(ctx)
  468. for k, v := range dataHeader {
  469. //设置单元格的值
  470. xlsx.SetCellValue(sheetName, k, v)
  471. }
  472. for i, userRecord := range re {
  473. row := i + 2
  474. writeExcel(row, xlsx, sheetName, userRecord)
  475. }
  476. //设置默认打开的表单
  477. xlsx.SetActiveSheet(index)
  478. filename := sheetName + "_" + ctx.Tr("user.static.all") + ".xlsx"
  479. ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(filename))
  480. ctx.Resp.Header().Set("Content-Type", "application/octet-stream")
  481. if _, err := xlsx.WriteTo(ctx.Resp); err != nil {
  482. log.Info("writer exel error." + err.Error())
  483. }
  484. } else {
  485. mapInterface := make(map[string]interface{})
  486. re, count := models.QueryUserStaticDataPage(pageOpts)
  487. mapInterface["data"] = re
  488. mapInterface["count"] = count
  489. ctx.JSON(http.StatusOK, mapInterface)
  490. }
  491. }
  492. func TimingCountDataByDateAndReCount(date string, isReCount bool) {
  493. t, _ := time.Parse("2006-01-02", date)
  494. startTime := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, t.Location())
  495. endTime := time.Date(t.Year(), t.Month(), t.Day(), 23, 59, 59, 0, t.Location())
  496. //query wiki data
  497. log.Info("start to time count data")
  498. wikiMap := make(map[string]int)
  499. warnEmailMessage := "用户统计信息入库失败,请尽快定位。"
  500. repoList, err := models.GetAllRepositories()
  501. if err != nil {
  502. log.Error("query repo error." + err.Error())
  503. mailer.SendWarnNotifyMail(setting.Warn_Notify_Mails, warnEmailMessage)
  504. return
  505. }
  506. log.Info("start to query wiki data")
  507. for _, repoRecord := range repoList {
  508. wikiPath := models.WikiPath(repoRecord.OwnerName, repoRecord.Name)
  509. time, err := git.GetLatestCommitTime(wikiPath)
  510. if err == nil {
  511. log.Info("last commit time:" + time.Format("2006-01-02 15:04:05") + " wikiPath=" + wikiPath)
  512. if time.After(startTime) {
  513. wikiRepo, _, err := FindWikiRepoCommitByWikiPath(wikiPath)
  514. if err != nil {
  515. log.Error("wiki not exist. wikiPath=" + wikiPath)
  516. } else {
  517. log.Info("wiki exist, wikiPath=" + wikiPath)
  518. list, err := wikiRepo.GetCommitByPathAndDays(wikiPath, 1)
  519. if err != nil {
  520. log.Info("err,err=v%", err)
  521. } else {
  522. for logEntry := list.Front(); logEntry != nil; logEntry = logEntry.Next() {
  523. commit := logEntry.Value.(*git.Commit)
  524. log.Info("commit msg=" + commit.CommitMessage + " time=" + commit.Committer.When.Format("2006-01-02 15:04:05") + " user=" + commit.Committer.Name)
  525. if _, ok := wikiMap[commit.Committer.Name]; !ok {
  526. wikiMap[commit.Committer.Name] = 1
  527. } else {
  528. wikiMap[commit.Committer.Name] += 1
  529. }
  530. }
  531. }
  532. }
  533. }
  534. }
  535. }
  536. //other user info data
  537. err = models.CounDataByDateAndReCount(wikiMap, startTime, endTime, isReCount)
  538. if err != nil {
  539. log.Error("count user info error." + err.Error())
  540. mailer.SendWarnNotifyMail(setting.Warn_Notify_Mails, warnEmailMessage)
  541. }
  542. log.Info("start to count all user info data")
  543. //models.RefreshUserStaticAllTabel(wikiMap)
  544. log.Info("end to count all user info data")
  545. }
  546. func TimingCountDataByDate(date string) {
  547. TimingCountDataByDateAndReCount(date, true)
  548. }
  549. func TimingCountData() {
  550. log.Info("start to time count data")
  551. currentTimeNow := time.Now()
  552. log.Info("current time:" + currentTimeNow.Format("2006-01-02 15:04:05"))
  553. startTime := currentTimeNow.AddDate(0, 0, -1).Format("2006-01-02")
  554. TimingCountDataByDateAndReCount(startTime, false)
  555. }