floraachy
/
gitlink-forgeplus
mirror of https://gitlink.org.cn/Gitlink/forgeplus.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

add: project trace tasks count

This commit is contained in:
yystopf 2022-05-11 11:20:18 +08:00
parent 1f13bba52e
commit 7750cbe272
4 changed files with 20 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/controllers/traces/projects_controller.rb
View File

@ -5,13 +5,13 @@ class Traces::ProjectsController < Traces::BaseController
def tasks def tasks
branch_name = params[:branch_name] branch_name = params[:branch_name]
return render_error("无可用检测次数") if current_user.trace_user&.tasks_count >= 5 return render_error("无可用检测次数") if @project&.trace_tasks_count >= 5
return render_error("分支名不能为空！") if branch_name.blank? return render_error("分支名不能为空！") if branch_name.blank?
@all_branches = Gitea::Repository::Branches::ListNameService.call(@project&.owner, @project.identifier) @all_branches = Gitea::Repository::Branches::ListNameService.call(@project&.owner, @project.identifier)
return render_error("请输入正确的分支名！") unless @all_branches["branch_name"].include?(branch_name) return render_error("请输入正确的分支名！") unless @all_branches["branch_name"].include?(branch_name)
code, data, error = Trace::CheckService.call(current_user.trace_token, @project, "1", branch_name) code, data, error = Trace::CheckService.call(current_user.trace_token, @project, "1", branch_name)
if code == 200 if code == 200
current_user.trace_user.increment!(:tasks_count, 1) @project.increment!(:trace_tasks_count, 1)
render_ok render_ok
else else
render_error("检测失败 Error:#{error}") render_error("检测失败 Error:#{error}")
@ -27,7 +27,7 @@ class Traces::ProjectsController < Traces::BaseController
page = params[:page].to_i.zero? ? 1 : params[:page].to_i page = params[:page].to_i.zero? ? 1 : params[:page].to_i
code, data, error = Trace::CheckResultService.call(current_user.trace_token, @project, nil, page, limit) code, data, error = Trace::CheckResultService.call(current_user.trace_token, @project, nil, page, limit)
if code == 200 if code == 200
current_user.trace_user.update_column(:tasks_count, data[0]["totalsize"]) if data.size > 0 @project.update_column(:trace_tasks_count, data[0]["totalsize"]) if data.size > 0
render :json => {data: data} render :json => {data: data}
else else
render_error("获取检测记录失败 Error:#{error}") render_error("获取检测记录失败 Error:#{error}")

1
app/models/trace_user.rb
View File

@ -14,7 +14,6 @@
# expired_at :datetime # expired_at :datetime
# created_at :datetime not null # created_at :datetime not null
# updated_at :datetime not null # updated_at :datetime not null
# tasks_count :integer default("0")
# #
# Indexes # Indexes
# #

5
db/migrate/20220511022334_add_tasks_count_to_trace_users.rb
View File

@ -1,5 +0,0 @@
class AddTasksCountToTraceUsers < ActiveRecord::Migration[5.2]
def change
add_column :trace_users, :tasks_count, :integer, default: 0
end
end

5
db/migrate/20220511031711_add_trace_tasks_count_to_projects.rb Normal file
View File

@ -0,0 +1,5 @@
class AddTraceTasksCountToProjects < ActiveRecord::Migration[5.2]
def change
add_column :projects, :trace_tasks_count, :integer, default: 0
end
end