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

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