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

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