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

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