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 23 kB

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