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

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