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_business_analysis.go 22 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643
  1. package models
  2. import (
  3. "fmt"
  4. "time"
  5. "code.gitea.io/gitea/modules/log"
  6. "code.gitea.io/gitea/modules/setting"
  7. "code.gitea.io/gitea/modules/timeutil"
  8. "xorm.io/builder"
  9. )
  10. type UserBusinessAnalysis struct {
  11. ID int64 `xorm:"pk"`
  12. CountDate int64 `xorm:"pk"`
  13. //action :ActionMergePullRequest // 11
  14. CodeMergeCount int `xorm:"NOT NULL DEFAULT 0"`
  15. //action :ActionCommitRepo // 5
  16. CommitCount int `xorm:"NOT NULL DEFAULT 0"`
  17. //action :ActionCommentIssue // 10
  18. IssueCount int `xorm:"NOT NULL DEFAULT 0"`
  19. //comment table current date
  20. CommentCount int `xorm:"NOT NULL DEFAULT 0"`
  21. //watch table current date
  22. FocusRepoCount int `xorm:"NOT NULL DEFAULT 0"`
  23. //star table current date
  24. StarRepoCount int `xorm:"NOT NULL DEFAULT 0"`
  25. //follow table
  26. WatchedCount int `xorm:"NOT NULL DEFAULT 0"`
  27. // user table
  28. GiteaAgeMonth int `xorm:"NOT NULL DEFAULT 0"`
  29. //
  30. CommitCodeSize int `xorm:"NOT NULL DEFAULT 0"`
  31. //attachement table
  32. CommitDatasetSize int `xorm:"NOT NULL DEFAULT 0"`
  33. //0
  34. CommitModelCount int `xorm:"NOT NULL DEFAULT 0"`
  35. //issue, issueassignees
  36. SolveIssueCount int `xorm:"NOT NULL DEFAULT 0"`
  37. //baike
  38. EncyclopediasCount int `xorm:"NOT NULL DEFAULT 0"`
  39. //user
  40. RegistDate timeutil.TimeStamp `xorm:"NOT NULL"`
  41. //repo
  42. CreateRepoCount int `xorm:"NOT NULL DEFAULT 0"`
  43. //login count, from elk
  44. LoginCount int `xorm:"NOT NULL DEFAULT 0"`
  45. //openi index
  46. OpenIIndex int `xorm:"NOT NULL DEFAULT 0"`
  47. //user
  48. Email string `xorm:"NOT NULL"`
  49. //user
  50. Name string `xorm:"NOT NULL"`
  51. }
  52. type UserBusinessAnalysisQueryOptions struct {
  53. ListOptions
  54. UserName string
  55. SortType string
  56. StartTime int64
  57. EndTime int64
  58. }
  59. func QueryUserStaticData(startTime int64, endTime int64) []*UserBusinessAnalysis {
  60. log.Info("query startTime =" + fmt.Sprint(startTime) + " endTime=" + fmt.Sprint(endTime))
  61. statictisSess := xStatistic.NewSession()
  62. defer statictisSess.Close()
  63. statictisSess.Select("*").Table("user_business_analysis").Where(" count_date>=" + fmt.Sprint(startTime) + " and count_date<=" + fmt.Sprint(endTime)).OrderBy("count_date desc")
  64. userBusinessAnalysisList := make([]*UserBusinessAnalysis, 0)
  65. statictisSess.Find(&userBusinessAnalysisList)
  66. resultMap := make(map[int64]*UserBusinessAnalysis)
  67. log.Info("query result size=" + fmt.Sprint(len(userBusinessAnalysisList)))
  68. for _, userRecord := range userBusinessAnalysisList {
  69. if _, ok := resultMap[userRecord.ID]; !ok {
  70. resultMap[userRecord.ID] = userRecord
  71. } else {
  72. resultMap[userRecord.ID].CodeMergeCount += userRecord.CodeMergeCount
  73. resultMap[userRecord.ID].CommitCount += userRecord.CommitCount
  74. resultMap[userRecord.ID].IssueCount += userRecord.IssueCount
  75. resultMap[userRecord.ID].CommentCount += userRecord.CommentCount
  76. resultMap[userRecord.ID].FocusRepoCount += userRecord.FocusRepoCount
  77. resultMap[userRecord.ID].StarRepoCount += userRecord.StarRepoCount
  78. resultMap[userRecord.ID].WatchedCount += userRecord.WatchedCount
  79. resultMap[userRecord.ID].CommitCodeSize += userRecord.CommitCodeSize
  80. resultMap[userRecord.ID].CommitDatasetSize += userRecord.CommitDatasetSize
  81. resultMap[userRecord.ID].CommitModelCount += userRecord.CommitModelCount
  82. resultMap[userRecord.ID].SolveIssueCount += userRecord.SolveIssueCount
  83. resultMap[userRecord.ID].EncyclopediasCount += userRecord.EncyclopediasCount
  84. resultMap[userRecord.ID].CreateRepoCount += userRecord.CreateRepoCount
  85. resultMap[userRecord.ID].LoginCount += userRecord.LoginCount
  86. }
  87. }
  88. userBusinessAnalysisReturnList := make([]*UserBusinessAnalysis, len(resultMap))
  89. index := 0
  90. for _, v := range resultMap {
  91. userBusinessAnalysisReturnList[index] = v
  92. index += 1
  93. }
  94. log.Info("return size=" + fmt.Sprint(len(userBusinessAnalysisReturnList)))
  95. return userBusinessAnalysisReturnList
  96. }
  97. func QueryUserStaticDataPage(opts *UserBusinessAnalysisQueryOptions) ([]*UserBusinessAnalysis, int64) {
  98. log.Info("query startTime =" + fmt.Sprint(opts.StartTime) + " endTime=" + fmt.Sprint(opts.EndTime))
  99. statictisSess := xStatistic.NewSession()
  100. defer statictisSess.Close()
  101. currentTimeNow := time.Now()
  102. pageStartTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, currentTimeNow.Location())
  103. pageEndTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 23, 59, 59, 0, currentTimeNow.Location())
  104. var cond = builder.NewCond()
  105. if len(opts.UserName) > 0 {
  106. cond = cond.And(
  107. builder.Eq{"name": opts.UserName},
  108. )
  109. }
  110. cond = cond.And(
  111. builder.Gte{"count_date": fmt.Sprint(pageStartTime)},
  112. )
  113. cond = cond.And(
  114. builder.Lte{"count_date": fmt.Sprint(pageEndTime)},
  115. )
  116. count, err := statictisSess.Where(cond).Count(new(UserBusinessAnalysis))
  117. if err != nil {
  118. log.Info("query error." + err.Error())
  119. return nil, 0
  120. }
  121. if opts.Page >= 0 && opts.PageSize > 0 {
  122. var start int
  123. if opts.Page == 0 {
  124. start = 0
  125. } else {
  126. start = (opts.Page - 1) * opts.PageSize
  127. }
  128. statictisSess.Limit(opts.PageSize, start)
  129. }
  130. statictisSess.OrderBy("count_date desc")
  131. userBusinessAnalysisList := make([]*UserBusinessAnalysis, 0, setting.UI.IssuePagingNum)
  132. if err := statictisSess.Table("user_business_analysis").Where(cond).
  133. Find(&userBusinessAnalysisList); err != nil {
  134. return nil, 0
  135. }
  136. resultMap := make(map[int64]*UserBusinessAnalysis)
  137. var newAndCond = builder.NewCond()
  138. var newOrCond = builder.NewCond()
  139. for _, userRecord := range userBusinessAnalysisList {
  140. newOrCond.Or(
  141. builder.Eq{"id": userRecord.ID},
  142. )
  143. }
  144. newAndCond = newAndCond.And(
  145. newOrCond,
  146. )
  147. newAndCond = newAndCond.And(
  148. builder.Gte{"count_date": fmt.Sprint(opts.StartTime)},
  149. )
  150. newAndCond = newAndCond.And(
  151. builder.Lte{"count_date": fmt.Sprint(opts.EndTime)},
  152. )
  153. userBusinessAnalysisList = make([]*UserBusinessAnalysis, 0)
  154. if err := statictisSess.Table("user_business_analysis").Where(newAndCond).
  155. Find(&userBusinessAnalysisList); err != nil {
  156. return nil, 0
  157. }
  158. log.Info("query result size=" + fmt.Sprint(len(userBusinessAnalysisList)))
  159. for _, userRecord := range userBusinessAnalysisList {
  160. if _, ok := resultMap[userRecord.ID]; !ok {
  161. resultMap[userRecord.ID] = userRecord
  162. } else {
  163. resultMap[userRecord.ID].CodeMergeCount += userRecord.CodeMergeCount
  164. resultMap[userRecord.ID].CommitCount += userRecord.CommitCount
  165. resultMap[userRecord.ID].IssueCount += userRecord.IssueCount
  166. resultMap[userRecord.ID].CommentCount += userRecord.CommentCount
  167. resultMap[userRecord.ID].FocusRepoCount += userRecord.FocusRepoCount
  168. resultMap[userRecord.ID].StarRepoCount += userRecord.StarRepoCount
  169. resultMap[userRecord.ID].WatchedCount += userRecord.WatchedCount
  170. resultMap[userRecord.ID].CommitCodeSize += userRecord.CommitCodeSize
  171. resultMap[userRecord.ID].CommitDatasetSize += userRecord.CommitDatasetSize
  172. resultMap[userRecord.ID].CommitModelCount += userRecord.CommitModelCount
  173. resultMap[userRecord.ID].SolveIssueCount += userRecord.SolveIssueCount
  174. resultMap[userRecord.ID].EncyclopediasCount += userRecord.EncyclopediasCount
  175. resultMap[userRecord.ID].CreateRepoCount += userRecord.CreateRepoCount
  176. resultMap[userRecord.ID].LoginCount += userRecord.LoginCount
  177. }
  178. }
  179. userBusinessAnalysisReturnList := make([]*UserBusinessAnalysis, len(resultMap))
  180. index := 0
  181. for _, v := range resultMap {
  182. userBusinessAnalysisReturnList[index] = v
  183. index += 1
  184. }
  185. log.Info("return size=" + fmt.Sprint(len(userBusinessAnalysisReturnList)))
  186. return userBusinessAnalysisReturnList, count
  187. }
  188. func CounDataByDate(wikiCountMap map[string]int, startTime time.Time, endTime time.Time) {
  189. log.Info("start to count other user info data")
  190. sess := x.NewSession()
  191. defer sess.Close()
  192. sess.Select("`user`.*").Table("user").Where("type != 1 and is_active=true")
  193. userList := make([]*User, 0)
  194. sess.Find(&userList)
  195. currentTimeNow := time.Now()
  196. log.Info("current time:" + currentTimeNow.Format("2006-01-02 15:04:05"))
  197. //yesterday := currentTimeNow.AddDate(0, 0, -1)
  198. //startTime := time.Date(yesterday.Year(), yesterday.Month(), yesterday.Day(), 0, 0, 0, 0, yesterday.Location())
  199. start_unix := startTime.Unix()
  200. log.Info("DB query time:" + startTime.Format("2006-01-02 15:04:05"))
  201. //endTime := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 0, 0, 0, currentTimeNow.Location())
  202. end_unix := endTime.Unix()
  203. CountDate := time.Date(currentTimeNow.Year(), currentTimeNow.Month(), currentTimeNow.Day(), 0, 1, 0, 0, currentTimeNow.Location())
  204. CodeMergeCountMap := queryPullRequest(start_unix, end_unix)
  205. CommitCountMap := queryAction(start_unix, end_unix, 5)
  206. IssueCountMap := queryAction(start_unix, end_unix, 10)
  207. CommentCountMap := queryComment(start_unix, end_unix)
  208. FocusRepoCountMap := queryWatch(start_unix, end_unix)
  209. StarRepoCountMap := queryStar(start_unix, end_unix)
  210. WatchedCountMap := queryFollow(start_unix, end_unix)
  211. CommitCodeSizeMap, err := GetAllUserKPIStats()
  212. if err != nil {
  213. log.Info("query commit code errr.")
  214. } else {
  215. log.Info("query commit code size, len=" + fmt.Sprint(len(CommitCodeSizeMap)))
  216. }
  217. CommitDatasetSizeMap := queryDatasetSize(start_unix, end_unix)
  218. SolveIssueCountMap := querySolveIssue(start_unix, end_unix)
  219. CreateRepoCountMap := queryUserCreateRepo(start_unix, end_unix)
  220. LoginCountMap := queryLoginCount(start_unix, end_unix)
  221. OpenIIndexMap := queryUserRepoOpenIIndex(start_unix, end_unix)
  222. statictisSess := xStatistic.NewSession()
  223. defer statictisSess.Close()
  224. for i, userRecord := range userList {
  225. var dateRecord UserBusinessAnalysis
  226. dateRecord.ID = userRecord.ID
  227. log.Info("i=" + fmt.Sprint(i) + " userName=" + userRecord.Name)
  228. dateRecord.CountDate = CountDate.Unix()
  229. statictisSess.Delete(&dateRecord)
  230. dateRecord.Email = userRecord.Email
  231. dateRecord.RegistDate = userRecord.CreatedUnix
  232. dateRecord.Name = userRecord.Name
  233. dateRecord.GiteaAgeMonth = subMonth(currentTimeNow, userRecord.CreatedUnix.AsTime())
  234. if _, ok := CodeMergeCountMap[dateRecord.ID]; !ok {
  235. dateRecord.CodeMergeCount = 0
  236. } else {
  237. dateRecord.CodeMergeCount = CodeMergeCountMap[dateRecord.ID]
  238. }
  239. if _, ok := CommitCountMap[dateRecord.ID]; !ok {
  240. dateRecord.CommitCount = 0
  241. } else {
  242. dateRecord.CommitCount = CommitCountMap[dateRecord.ID]
  243. }
  244. if _, ok := IssueCountMap[dateRecord.ID]; !ok {
  245. dateRecord.IssueCount = 0
  246. } else {
  247. dateRecord.IssueCount = IssueCountMap[dateRecord.ID]
  248. }
  249. if _, ok := CommentCountMap[dateRecord.ID]; !ok {
  250. dateRecord.CommentCount = 0
  251. } else {
  252. dateRecord.CommentCount = CommentCountMap[dateRecord.ID]
  253. }
  254. if _, ok := FocusRepoCountMap[dateRecord.ID]; !ok {
  255. dateRecord.FocusRepoCount = 0
  256. } else {
  257. dateRecord.FocusRepoCount = FocusRepoCountMap[dateRecord.ID]
  258. }
  259. if _, ok := StarRepoCountMap[dateRecord.ID]; !ok {
  260. dateRecord.StarRepoCount = 0
  261. } else {
  262. dateRecord.StarRepoCount = StarRepoCountMap[dateRecord.ID]
  263. }
  264. if _, ok := WatchedCountMap[dateRecord.ID]; !ok {
  265. dateRecord.WatchedCount = 0
  266. } else {
  267. dateRecord.WatchedCount = WatchedCountMap[dateRecord.ID]
  268. }
  269. if _, ok := CommitCodeSizeMap[dateRecord.Email]; !ok {
  270. dateRecord.CommitCodeSize = 0
  271. } else {
  272. dateRecord.CommitCodeSize = int(CommitCodeSizeMap[dateRecord.Email].CommitLines)
  273. }
  274. if _, ok := CommitDatasetSizeMap[dateRecord.ID]; !ok {
  275. dateRecord.CommitDatasetSize = 0
  276. } else {
  277. dateRecord.CommitDatasetSize = CommitDatasetSizeMap[dateRecord.ID]
  278. }
  279. if _, ok := SolveIssueCountMap[dateRecord.ID]; !ok {
  280. dateRecord.SolveIssueCount = 0
  281. } else {
  282. dateRecord.SolveIssueCount = SolveIssueCountMap[dateRecord.ID]
  283. }
  284. if _, ok := wikiCountMap[dateRecord.Name]; !ok {
  285. dateRecord.EncyclopediasCount = 0
  286. } else {
  287. dateRecord.EncyclopediasCount = wikiCountMap[dateRecord.Name]
  288. }
  289. if _, ok := CreateRepoCountMap[dateRecord.ID]; !ok {
  290. dateRecord.CreateRepoCount = 0
  291. } else {
  292. dateRecord.CreateRepoCount = CreateRepoCountMap[dateRecord.ID]
  293. }
  294. if _, ok := LoginCountMap[dateRecord.ID]; !ok {
  295. dateRecord.LoginCount = 0
  296. } else {
  297. dateRecord.LoginCount = LoginCountMap[dateRecord.ID]
  298. }
  299. if _, ok := OpenIIndexMap[dateRecord.ID]; !ok {
  300. dateRecord.OpenIIndex = 0
  301. } else {
  302. dateRecord.OpenIIndex = int(OpenIIndexMap[dateRecord.ID] * 100)
  303. }
  304. dateRecord.CommitModelCount = 0
  305. statictisSess.Insert(&dateRecord)
  306. }
  307. }
  308. func querySolveIssue(start_unix int64, end_unix int64) map[int64]int {
  309. //select issue_assignees.* from issue_assignees,issue where issue.is_closed=true and issue.id=issue_assignees.issue_id
  310. sess := x.NewSession()
  311. defer sess.Close()
  312. sess.Select("issue_assignees.*").Table("issue_assignees").
  313. Join("inner", "issue", "issue.id=issue_assignees.issue_id").
  314. Where("issue.is_closed=true and issue.closed_unix>=" + fmt.Sprint(start_unix) + " and issue.closed_unix<=" + fmt.Sprint(end_unix))
  315. issueAssigneesList := make([]*IssueAssignees, 0)
  316. sess.Find(&issueAssigneesList)
  317. resultMap := make(map[int64]int)
  318. log.Info("query IssueAssignees size=" + fmt.Sprint(len(issueAssigneesList)))
  319. for _, issueAssigneesRecord := range issueAssigneesList {
  320. if _, ok := resultMap[issueAssigneesRecord.AssigneeID]; !ok {
  321. resultMap[issueAssigneesRecord.AssigneeID] = 1
  322. } else {
  323. resultMap[issueAssigneesRecord.AssigneeID] += 1
  324. }
  325. }
  326. return resultMap
  327. }
  328. func queryPullRequest(start_unix int64, end_unix int64) map[int64]int {
  329. sess := x.NewSession()
  330. defer sess.Close()
  331. sess.Select("issue.*").Table("issue").
  332. Join("inner", "pull_request", "issue.id=pull_request.issue_id").
  333. Where("pull_request.merged_unix>=" + fmt.Sprint(start_unix) + " and pull_request.merged_unix<=" + fmt.Sprint(end_unix))
  334. issueList := make([]*Issue, 0)
  335. sess.Find(&issueList)
  336. resultMap := make(map[int64]int)
  337. log.Info("query issue(PR) size=" + fmt.Sprint(len(issueList)))
  338. for _, issueRecord := range issueList {
  339. if _, ok := resultMap[issueRecord.PosterID]; !ok {
  340. resultMap[issueRecord.PosterID] = 1
  341. } else {
  342. resultMap[issueRecord.PosterID] += 1
  343. }
  344. }
  345. return resultMap
  346. }
  347. func queryAction(start_unix int64, end_unix int64, actionType int64) map[int64]int {
  348. sess := x.NewSession()
  349. defer sess.Close()
  350. sess.Select("id,user_id,op_type,act_user_id").Table("action").Where("op_type=" + fmt.Sprint(actionType) + " and created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  351. actionList := make([]*Action, 0)
  352. sess.Find(&actionList)
  353. resultMap := make(map[int64]int)
  354. log.Info("query action size=" + fmt.Sprint(len(actionList)))
  355. for _, actionRecord := range actionList {
  356. if _, ok := resultMap[actionRecord.UserID]; !ok {
  357. resultMap[actionRecord.UserID] = 1
  358. } else {
  359. resultMap[actionRecord.UserID] += 1
  360. }
  361. }
  362. return resultMap
  363. }
  364. func queryComment(start_unix int64, end_unix int64) map[int64]int {
  365. sess := x.NewSession()
  366. defer sess.Close()
  367. sess.Select("id,type,poster_id").Table("comment").Where(" created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  368. commentList := make([]*Comment, 0)
  369. sess.Find(&commentList)
  370. resultMap := make(map[int64]int)
  371. log.Info("query Comment size=" + fmt.Sprint(len(commentList)))
  372. for _, commentRecord := range commentList {
  373. if _, ok := resultMap[commentRecord.PosterID]; !ok {
  374. resultMap[commentRecord.PosterID] = 1
  375. } else {
  376. resultMap[commentRecord.PosterID] += 1
  377. }
  378. }
  379. return resultMap
  380. }
  381. func queryWatch(start_unix int64, end_unix int64) map[int64]int {
  382. sess := x.NewSession()
  383. defer sess.Close()
  384. sess.Select("id,user_id,repo_id").Table("watch").Where(" created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  385. watchList := make([]*Watch, 0)
  386. sess.Find(&watchList)
  387. resultMap := make(map[int64]int)
  388. log.Info("query Watch size=" + fmt.Sprint(len(watchList)))
  389. for _, watchRecord := range watchList {
  390. if _, ok := resultMap[watchRecord.UserID]; !ok {
  391. resultMap[watchRecord.UserID] = 1
  392. } else {
  393. resultMap[watchRecord.UserID] += 1
  394. }
  395. }
  396. return resultMap
  397. }
  398. func queryStar(start_unix int64, end_unix int64) map[int64]int {
  399. sess := x.NewSession()
  400. defer sess.Close()
  401. sess.Select("id,uid,repo_id").Table("star").Where(" created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  402. starList := make([]*Star, 0)
  403. sess.Find(&starList)
  404. resultMap := make(map[int64]int)
  405. log.Info("query Star size=" + fmt.Sprint(len(starList)))
  406. for _, starRecord := range starList {
  407. if _, ok := resultMap[starRecord.UID]; !ok {
  408. resultMap[starRecord.UID] = 1
  409. } else {
  410. resultMap[starRecord.UID] += 1
  411. }
  412. }
  413. return resultMap
  414. }
  415. func queryFollow(start_unix int64, end_unix int64) map[int64]int {
  416. sess := x.NewSession()
  417. defer sess.Close()
  418. sess.Select("id,user_id,follow_id").Table("follow").Where(" created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  419. followList := make([]*Follow, 0)
  420. sess.Find(&followList)
  421. resultMap := make(map[int64]int)
  422. log.Info("query Follow size=" + fmt.Sprint(len(followList)))
  423. for _, followRecord := range followList {
  424. if _, ok := resultMap[followRecord.UserID]; !ok {
  425. resultMap[followRecord.UserID] = 1
  426. } else {
  427. resultMap[followRecord.UserID] += 1
  428. }
  429. }
  430. return resultMap
  431. }
  432. func queryDatasetSize(start_unix int64, end_unix int64) map[int64]int {
  433. sess := x.NewSession()
  434. defer sess.Close()
  435. sess.Select("id,uploader_id,size").Table("attachment").Where(" created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  436. attachmentList := make([]*Attachment, 0)
  437. sess.Find(&attachmentList)
  438. resultMap := make(map[int64]int)
  439. log.Info("query Attachment size=" + fmt.Sprint(len(attachmentList)))
  440. for _, attachRecord := range attachmentList {
  441. if _, ok := resultMap[attachRecord.UploaderID]; !ok {
  442. resultMap[attachRecord.UploaderID] = int(attachRecord.Size / (1024 * 1024)) //MB
  443. } else {
  444. resultMap[attachRecord.UploaderID] += int(attachRecord.Size / (1024 * 1024)) //MB
  445. }
  446. }
  447. return resultMap
  448. }
  449. func queryUserCreateRepo(start_unix int64, end_unix int64) map[int64]int {
  450. sess := x.NewSession()
  451. defer sess.Close()
  452. sess.Select("id,owner_id,name").Table("repository").Where("is_fork=false and created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  453. repoList := make([]*Repository, 0)
  454. sess.Find(&repoList)
  455. resultMap := make(map[int64]int)
  456. log.Info("query Repository size=" + fmt.Sprint(len(repoList)))
  457. for _, repoRecord := range repoList {
  458. if _, ok := resultMap[repoRecord.OwnerID]; !ok {
  459. resultMap[repoRecord.OwnerID] = 1
  460. } else {
  461. resultMap[repoRecord.OwnerID] += 1
  462. }
  463. }
  464. return resultMap
  465. }
  466. func queryUserRepoOpenIIndex(start_unix int64, end_unix int64) map[int64]float64 {
  467. statictisSess := xStatistic.NewSession()
  468. defer statictisSess.Close()
  469. statictisSess.Select("repo_id,radar_total").Table("repo_statistic").Where("created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  470. repoStatisticList := make([]*RepoStatistic, 0)
  471. statictisSess.Find(&repoStatisticList)
  472. repoOpenIIndexMap := make(map[int64]float64)
  473. log.Info("query repo_statistic size=" + fmt.Sprint(len(repoStatisticList)))
  474. for _, repoRecord := range repoStatisticList {
  475. if _, ok := repoOpenIIndexMap[repoRecord.RepoID]; !ok {
  476. repoOpenIIndexMap[repoRecord.RepoID] = repoRecord.RadarTotal
  477. }
  478. }
  479. sess := x.NewSession()
  480. defer sess.Close()
  481. sess.Select("id,owner_id,name").Table("repository").Where("is_fork=false and created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  482. repoList := make([]*Repository, 0)
  483. sess.Find(&repoList)
  484. userMap := make(map[int64]float64)
  485. log.Info("query Repository size=" + fmt.Sprint(len(repoList)))
  486. for _, repoRecord := range repoList {
  487. if _, ok := userMap[repoRecord.OwnerID]; !ok {
  488. if _, ok := repoOpenIIndexMap[repoRecord.ID]; !ok {
  489. userMap[repoRecord.OwnerID] = repoOpenIIndexMap[repoRecord.ID]
  490. }
  491. }
  492. }
  493. //query collaboration
  494. sess.Select("repo_id,user_id,mode").Table("collaboration")
  495. collaborationList := make([]*Collaboration, 0)
  496. sess.Find(&collaborationList)
  497. for _, collaborationRecord := range collaborationList {
  498. if _, ok := userMap[collaborationRecord.UserID]; !ok {
  499. if _, ok := repoOpenIIndexMap[collaborationRecord.RepoID]; !ok {
  500. userMap[collaborationRecord.UserID] = repoOpenIIndexMap[collaborationRecord.RepoID]
  501. }
  502. } else {
  503. if _, ok := repoOpenIIndexMap[collaborationRecord.RepoID]; !ok {
  504. userMap[collaborationRecord.UserID] += repoOpenIIndexMap[collaborationRecord.RepoID]
  505. }
  506. }
  507. }
  508. return userMap
  509. }
  510. func queryLoginCount(start_unix int64, end_unix int64) map[int64]int {
  511. statictisSess := xStatistic.NewSession()
  512. defer statictisSess.Close()
  513. statictisSess.Select("id,u_id").Table("user_login_log").Where("created_unix>=" + fmt.Sprint(start_unix) + " and created_unix<=" + fmt.Sprint(end_unix))
  514. userLoginLogList := make([]*UserLoginLog, 0)
  515. statictisSess.Find(&userLoginLogList)
  516. resultMap := make(map[int64]int)
  517. log.Info("query user login size=" + fmt.Sprint(len(userLoginLogList)))
  518. for _, loginRecord := range userLoginLogList {
  519. if _, ok := resultMap[loginRecord.UId]; !ok {
  520. resultMap[loginRecord.UId] = 1
  521. } else {
  522. resultMap[loginRecord.UId] += 1
  523. }
  524. }
  525. return resultMap
  526. }
  527. func subMonth(t1, t2 time.Time) (month int) {
  528. y1 := t1.Year()
  529. y2 := t2.Year()
  530. m1 := int(t1.Month())
  531. m2 := int(t2.Month())
  532. d1 := t1.Day()
  533. d2 := t2.Day()
  534. yearInterval := y1 - y2
  535. // 如果 d1的 月-日 小于 d2的 月-日 那么 yearInterval-- 这样就得到了相差的年数
  536. if m1 < m2 || m1 == m2 && d1 < d2 {
  537. yearInterval--
  538. }
  539. // 获取月数差值
  540. monthInterval := (m1 + 12) - m2
  541. if d1 < d2 {
  542. monthInterval--
  543. }
  544. monthInterval %= 12
  545. month = yearInterval*12 + monthInterval
  546. return month
  547. }