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

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586
  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. if tableName == "public.user_business_analysis_yesterday" {
  289. mapInterface["datarecordbegintime"] = setting.RadarMap.GrowthBeginTime
  290. if len(result) > 0 {
  291. dateTime := time.Unix(result[0].CountDate, 0)
  292. mapInterface["lastUpdatedTime"] = dateTime.Format("2006-01-02 15:04:05")
  293. } else {
  294. mapInterface["lastUpdatedTime"] = ""
  295. }
  296. }
  297. ctx.JSON(http.StatusOK, mapInterface)
  298. }
  299. }
  300. func QueryRankingList(ctx *context.Context) {
  301. key := ctx.Query("key")
  302. tableName := ctx.Query("tableName")
  303. limit := ctx.QueryInt("limit")
  304. result, count := models.QueryRankList(key, tableName, limit)
  305. mapInterface := make(map[string]interface{})
  306. mapInterface["data"] = result
  307. mapInterface["count"] = count
  308. ctx.JSON(http.StatusOK, mapInterface)
  309. }
  310. func QueryUserMetricsCurrentMonth(ctx *context.Context) {
  311. currentTimeNow := time.Now()
  312. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  313. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), 1, 0, 0, 0, 0, currentTimeNow.Location())
  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 QueryUserMetricsCurrentWeek(ctx *context.Context) {
  320. currentTimeNow := time.Now()
  321. offset := int(time.Monday - currentTimeNow.Weekday())
  322. if offset > 0 {
  323. offset = -6
  324. }
  325. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, offset)
  326. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  327. queryMetrics(ctx, "public.user_business_analysis_current_week", pageStartTime, pageEndTime)
  328. }
  329. func QueryUserStaticCurrentWeek(ctx *context.Context) {
  330. queryUserDataPage(ctx, "public.user_business_analysis_current_week", new(models.UserBusinessAnalysisCurrentWeek))
  331. }
  332. func QueryUserMetricsCurrentYear(ctx *context.Context) {
  333. currentTimeNow := time.Now()
  334. pageStartTime := time.Date(currentTimeNow.Year(), 1, 1, 0, 0, 0, 0, currentTimeNow.Location())
  335. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  336. queryMetrics(ctx, "public.user_business_analysis_current_year", pageStartTime, pageEndTime)
  337. }
  338. func QueryUserStaticCurrentYear(ctx *context.Context) {
  339. queryUserDataPage(ctx, "public.user_business_analysis_current_year", new(models.UserBusinessAnalysisCurrentYear))
  340. }
  341. func QueryUserMetricsLast30Day(ctx *context.Context) {
  342. currentTimeNow := time.Now()
  343. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, -30)
  344. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  345. queryMetrics(ctx, "public.user_business_analysis_last30_day", pageStartTime, pageEndTime)
  346. }
  347. func QueryUserStaticLast30Day(ctx *context.Context) {
  348. queryUserDataPage(ctx, "public.user_business_analysis_last30_day", new(models.UserBusinessAnalysisLast30Day))
  349. }
  350. func QueryUserMetricsLastMonth(ctx *context.Context) {
  351. currentTimeNow := time.Now()
  352. thisMonth := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), 1, 0, 0, 0, 0, currentTimeNow.Location())
  353. pageStartTime := thisMonth.AddDate(0, -1, 0)
  354. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), 1, 23, 59, 59, 0, currentTimeNow.Location()).AddDate(0, 0, -1)
  355. queryMetrics(ctx, "public.user_business_analysis_last_month", pageStartTime, pageEndTime)
  356. }
  357. func QueryUserStaticLastMonth(ctx *context.Context) {
  358. queryUserDataPage(ctx, "public.user_business_analysis_last_month", new(models.UserBusinessAnalysisLastMonth))
  359. }
  360. func QueryUserMetricsYesterday(ctx *context.Context) {
  361. currentTimeNow := time.Now()
  362. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, time.Local)
  363. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  364. queryMetrics(ctx, "public.user_business_analysis_yesterday", pageStartTime, pageEndTime)
  365. }
  366. func QueryUserStaticYesterday(ctx *context.Context) {
  367. queryUserDataPage(ctx, "public.user_business_analysis_yesterday", new(models.UserBusinessAnalysisYesterday))
  368. }
  369. func QueryUserMetricsAll(ctx *context.Context) {
  370. currentTimeNow := time.Now()
  371. pageStartTime := time.Date(2022, 4, 5, 0, 0, 0, 0, currentTimeNow.Location())
  372. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  373. queryMetrics(ctx, "public.user_business_analysis_all", pageStartTime, pageEndTime)
  374. }
  375. func QueryUserStaticAll(ctx *context.Context) {
  376. queryUserDataPage(ctx, "public.user_business_analysis_all", new(models.UserBusinessAnalysisAll))
  377. }
  378. func QueryUserMetricDataPage(ctx *context.Context) {
  379. startDate := ctx.Query("startDate")
  380. endDate := ctx.Query("endDate")
  381. startTime, _ := time.ParseInLocation("2006-01-02", startDate, time.Local)
  382. endTime, _ := time.ParseInLocation("2006-01-02", endDate, time.Local)
  383. page := ctx.QueryInt("page")
  384. if page <= 0 {
  385. page = 1
  386. }
  387. pageSize := ctx.QueryInt("pageSize")
  388. if pageSize <= 0 {
  389. pageSize = setting.UI.IssuePagingNum
  390. }
  391. result, count := models.QueryMetricsPage(startTime.Unix(), endTime.Unix(), page, pageSize)
  392. mapInterface := make(map[string]interface{})
  393. mapInterface["data"] = result
  394. mapInterface["count"] = count
  395. ctx.JSON(http.StatusOK, mapInterface)
  396. }
  397. func QueryUserStaticDataPage(ctx *context.Context) {
  398. startDate := ctx.Query("startDate")
  399. endDate := ctx.Query("endDate")
  400. page := ctx.QueryInt("page")
  401. if page <= 0 {
  402. page = 1
  403. }
  404. pageSize := ctx.QueryInt("pageSize")
  405. if pageSize <= 0 {
  406. pageSize = setting.UI.IssuePagingNum
  407. }
  408. userName := ctx.Query("userName")
  409. IsReturnFile := ctx.QueryBool("IsReturnFile")
  410. log.Info("startDate=" + startDate + " endDate=" + endDate + " userName=" + userName + " page=" + fmt.Sprint(page))
  411. var startTime time.Time
  412. var endTime time.Time
  413. var isAll bool
  414. if startDate == "all" {
  415. isAll = true
  416. startTime = time.Now()
  417. endTime = time.Now()
  418. } else {
  419. startTime, _ = time.ParseInLocation("2006-01-02", startDate, time.Local)
  420. startTime = time.Date(startTime.Year(), startTime.Month(), startTime.Day(), 12, 0, 0, 0, startTime.Location())
  421. settingStartTime, _ := time.Parse("2006-01-02", setting.RadarMap.RecordBeginTime)
  422. if startTime.Unix() < settingStartTime.Unix() {
  423. startTime = settingStartTime
  424. startDate = settingStartTime.Format("2006-01-02")
  425. }
  426. endTime, _ = time.ParseInLocation("2006-01-02", endDate, time.Local)
  427. endTime = endTime.AddDate(0, 0, 1)
  428. endTime = time.Date(endTime.Year(), endTime.Month(), endTime.Day(), 23, 59, 59, 0, startTime.Location())
  429. isAll = false
  430. log.Info("startTime=" + fmt.Sprint(startTime.Unix()) + " endDate=" + fmt.Sprint(endTime.Unix()))
  431. }
  432. if IsReturnFile {
  433. page = -1
  434. pageSize = -1
  435. }
  436. pageOpts := &models.UserBusinessAnalysisQueryOptions{
  437. ListOptions: models.ListOptions{
  438. Page: page,
  439. PageSize: pageSize,
  440. },
  441. UserName: userName,
  442. StartTime: startTime.Unix(),
  443. EndTime: endTime.Unix(),
  444. IsAll: isAll,
  445. }
  446. if IsReturnFile {
  447. re, count := models.QueryUserStaticDataAll(pageOpts)
  448. log.Info("return count=" + fmt.Sprint(count))
  449. //writer exec file.
  450. xlsx := excelize.NewFile()
  451. sheetName := ctx.Tr("user.static.sheetname")
  452. index := xlsx.NewSheet(sheetName)
  453. xlsx.DeleteSheet("Sheet1")
  454. dataHeader := getExcelHeader(ctx)
  455. for k, v := range dataHeader {
  456. //设置单元格的值
  457. xlsx.SetCellValue(sheetName, k, v)
  458. }
  459. for i, userRecord := range re {
  460. row := i + 2
  461. writeExcel(row, xlsx, sheetName, userRecord)
  462. }
  463. //设置默认打开的表单
  464. xlsx.SetActiveSheet(index)
  465. filename := sheetName + "_" + ctx.Tr("user.static.all") + ".xlsx"
  466. ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(filename))
  467. ctx.Resp.Header().Set("Content-Type", "application/octet-stream")
  468. if _, err := xlsx.WriteTo(ctx.Resp); err != nil {
  469. log.Info("writer exel error." + err.Error())
  470. }
  471. } else {
  472. mapInterface := make(map[string]interface{})
  473. re, count := models.QueryUserStaticDataPage(pageOpts)
  474. mapInterface["data"] = re
  475. mapInterface["count"] = count
  476. ctx.JSON(http.StatusOK, mapInterface)
  477. }
  478. }
  479. func TimingCountDataByDateAndReCount(date string, isReCount bool) {
  480. t, _ := time.Parse("2006-01-02", date)
  481. startTime := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, t.Location())
  482. endTime := time.Date(t.Year(), t.Month(), t.Day(), 23, 59, 59, 0, t.Location())
  483. //query wiki data
  484. log.Info("start to time count data")
  485. wikiMap := make(map[string]int)
  486. warnEmailMessage := "用户统计信息入库失败,请尽快定位。"
  487. repoList, err := models.GetAllRepositories()
  488. if err != nil {
  489. log.Error("query repo error." + err.Error())
  490. mailer.SendWarnNotifyMail(setting.Warn_Notify_Mails, warnEmailMessage)
  491. return
  492. }
  493. log.Info("start to query wiki data")
  494. for _, repoRecord := range repoList {
  495. wikiPath := models.WikiPath(repoRecord.OwnerName, repoRecord.Name)
  496. time, err := git.GetLatestCommitTime(wikiPath)
  497. if err == nil {
  498. log.Info("last commit time:" + time.Format("2006-01-02 15:04:05") + " wikiPath=" + wikiPath)
  499. if time.After(startTime) {
  500. wikiRepo, _, err := FindWikiRepoCommitByWikiPath(wikiPath)
  501. if err != nil {
  502. log.Error("wiki not exist. wikiPath=" + wikiPath)
  503. } else {
  504. log.Info("wiki exist, wikiPath=" + wikiPath)
  505. list, err := wikiRepo.GetCommitByPathAndDays(wikiPath, 1)
  506. if err != nil {
  507. log.Info("err,err=v%", err)
  508. } else {
  509. for logEntry := list.Front(); logEntry != nil; logEntry = logEntry.Next() {
  510. commit := logEntry.Value.(*git.Commit)
  511. log.Info("commit msg=" + commit.CommitMessage + " time=" + commit.Committer.When.Format("2006-01-02 15:04:05") + " user=" + commit.Committer.Name)
  512. if _, ok := wikiMap[commit.Committer.Name]; !ok {
  513. wikiMap[commit.Committer.Name] = 1
  514. } else {
  515. wikiMap[commit.Committer.Name] += 1
  516. }
  517. }
  518. }
  519. }
  520. }
  521. }
  522. }
  523. //other user info data
  524. err = models.CounDataByDateAndReCount(wikiMap, startTime, endTime, isReCount)
  525. if err != nil {
  526. log.Error("count user info error." + err.Error())
  527. mailer.SendWarnNotifyMail(setting.Warn_Notify_Mails, warnEmailMessage)
  528. }
  529. log.Info("start to count all user info data")
  530. //models.RefreshUserStaticAllTabel(wikiMap)
  531. log.Info("end to count all user info data")
  532. }
  533. func TimingCountDataByDate(date string) {
  534. TimingCountDataByDateAndReCount(date, true)
  535. }
  536. func TimingCountData() {
  537. log.Info("start to time count data")
  538. currentTimeNow := time.Now()
  539. log.Info("current time:" + currentTimeNow.Format("2006-01-02 15:04:05"))
  540. startTime := currentTimeNow.AddDate(0, 0, -1).Format("2006-01-02")
  541. TimingCountDataByDateAndReCount(startTime, false)
  542. }