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

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