【粉丝福利社】分布式系统性能优化:方法与实践
2025/12/26 23:05:58
将以下脚本保存为:/gitlab/custom_hooks/pre-receive.d/check-merge-rules,并确保其可执行 (chmod +x)。
#!/usr/bin/env bash# 全局 pre-receive 钩子:仅校验MR合并的分支路径规则# 注意:分支的推送权限(是否允许直接push)应由GitLab界面上的“分支保护”功能控制。# --- 规则配置 ---readonlyMAIN_BRANCH="main"readonlyDEVELOP_BRANCH="develop"# 定义允许合并到 main 的源分支模式readonlyALLOWED_MERGE_TO_MAIN=("develop""hotfix/*")# 定义允许合并到 develop 的源分支模式readonlyALLOWED_MERGE_TO_DEVELOP=("feature/*""bugfix/*""sync-hotfix-*")# --- 函数:拒绝推送并给出提示 ---reject_with_message(){echo"❌ 合并请求路径校验失败"echo"📌 目标分支:$1"echo"📌 源分支:$2"echo"📋 规则:$3"echo"💡 请创建符合上述规则的合并请求。"exit1}# --- 主逻辑 ---whileread-r oldrev newrev refname;do# 提取目标分支名target_branch="${refname#refs/heads/}"# 仅对推送到 main 或 develop 的合并操作进行检查if[["$target_branch"!="$MAIN_BRANCH"&&"$target_branch"!="$DEVELOP_BRANCH"]];thencontinue# 推送到其他分支,不予干涉fi# 判断是否为“快进合并”推送(这是GitLab执行MR合并的典型方式)ifgitmerge-base --is-ancestor"$oldrev""$newrev"2>/dev/null;then# 是快进推送,尝试分析是否为合并提交# 获取本次推送引入的最后一个提交(最可能是合并提交)latest_commit=$(gitrev-list -n1"$oldrev".."$newrev")# 检查这个提交是否是合并提交(有两个或以上父提交)parent_count=$(gitlog --pretty=%P -n1"$latest_commit"|wc-w)if[[$parent_count-lt2]];then# 如果不是合并提交,说明是直接推送,这应由分支保护功能阻止# 此处我们只记录日志并放行,最终应由GitLab的“允许推送”设置来拦截echo"信息: 检测到向${target_branch}的直接推送,分支保护规则将决定是否允许。"continuefi# 是合并提交,提取其第二个父提交(即被合并的源分支的最后提交)source_commit=$(gitlog --pretty=%P -n1"$latest_commit"|cut-d' '-f2)# 查找包含此提交的源分支名(取第一个找到的远程分支)source_branch=$(gitbranch -r --contains"$source_commit"2>/dev/null|head-1|sed's|^origin/||')if[[-z"$source_branch"]];then# 如果通过父提交找不到远程分支,尝试一种备用方案:# 检查本次推送引入的提交信息中,是否包含类似 "Merge branch 'feature/xxx'" 的GitLab默认合并信息merge_message=$(gitlog --format=%B -n1"$latest_commit")if[["$merge_message"=~[Mm]erge\branch\\'([^\']+)]];thensource_branch="${BASH_REMATCH[1]}"echo"信息: 通过提交信息解析出源分支为: '$source_branch'"elif[["$merge_message"=~[Mm]erge\request\\!\d+]];then# 如果提交信息是类似 “Merge request !123” 格式,说明是GitLab MR合并# 这种情况下,我们无法准确获知源分支名,但可以判断这是一个合规的MR合并操作# 出于安全,我们对此类“无法识别源分支的MR合并”进行放行,但记录日志。# 你也可以选择在此处添加更严格的判断,例如要求必须匹配特定模式。echo"警告: 检测到标准的GitLab MR合并提交,但无法解析具体源分支名。予以放行。"continueelse# 如果既找不到分支,提交信息也不是标准合并,则拒绝reject_with_message"$target_branch""(未知)""无法识别合并来源,请确保从有效的远程分支创建MR。"fifi# 根据目标分支应用对应规则进行匹配case"$target_branch"in"$MAIN_BRANCH")allowed=falseforpatternin"${ALLOWED_MERGE_TO_MAIN[@]}";doif[["$source_branch"==$pattern]];then# 注意这里的 $pattern 不加引号,以启用通配符匹配allowed=truebreakfidoneif!$allowed;thenreject_with_message"$MAIN_BRANCH""$source_branch""仅允许从${ALLOWED_MERGE_TO_MAIN[*]}合并。"fi;;"$DEVELOP_BRANCH")allowed=falseforpatternin"${ALLOWED_MERGE_TO_DEVELOP[@]}";doif[["$source_branch"==$pattern]];thenallowed=truebreakfidoneif!$allowed;thenreject_with_message"$DEVELOP_BRANCH""$source_branch""仅允许从${ALLOWED_MERGE_TO_DEVELOP[*]}合并。"fi;;esacecho"信息: 合并路径检查通过 [${source_branch}->${target_branch}]。"else# 非快进推送(如强制推送),拒绝以保护历史echo"❌ 拒绝向${target_branch}的强制推送或非快进推送。"echo"💡 请通过创建合并请求来合入代码。"exit1fidoneexit0此脚本与GitLab Web界面设置形成了明确分工,如下表所示:
| 控制维度 | 负责组件 | 具体设置/行为 |
|---|---|---|
| 能否直接推送 | GitLab 分支保护 | 在设置 > 仓库 > 受保护的分支中,将main和develop的“允许推送”设置为“无一人”。 |
| 谁能合并MR | GitLab 分支保护 | 在同上位置,设置“允许合并”为“维护者”或“开发者+维护者”。 |
| MR的源分支是否符合路径规则 | 本pre-receive钩子 | 如上脚本,校验feature/* -> develop,hotfix/* -> main等规则。 |
| Hotfix同步回Develop | 本pre-receive钩子 | 脚本ALLOWED_MERGE_TO_DEVELOP规则已包含sync-hotfix-*模式,允许以此类分支为源创建到develop的MR。 |
保存并配置脚本:
sudovim/gitlab/custom_hooks/pre-receive.d/check-merge-rules# 粘贴脚本内容sudochmod+x /gitlab/custom_hooks/pre-receive.d/check-merge-rulessudochowngit:git /gitlab/custom_hooks/pre-receive.d/check-merge-rules在GitLab Web界面完成关键设置:
main分支:无一人维护者(根据你的团队权限调整)develop分支:无一人开发者 + 维护者测试验证:
feature/test创建到develop的MR,合并时应成功。feature/test创建到main的MR,合并时钩子应拒绝,并显示清晰的规则错误。git push origin develop,GitLab分支保护应直接拒绝(与钩子无关)。hotfix/xxx合并到main(应成功)。main新建分支sync-hotfix-xxx。sync-hotfix-xxx->develop的MR,合并时应成功。sync-hotfix-*作为合规源分支,这要求团队在同步时遵循“从main创建同步分支,再向develop提MR”的安全流程。通过以上配置,你将得到一个既严格又清晰的分支合并管控系统。如果需要调整分支命名模式(例如将bugfix/*改为fix/*),只需修改脚本顶部的配置数组即可。