webforwarding.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684
  1. package service
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. v1 "github.com/go-nunu/nunu-layout-advanced/api/v1"
  7. "github.com/go-nunu/nunu-layout-advanced/internal/model"
  8. "github.com/go-nunu/nunu-layout-advanced/internal/repository"
  9. "github.com/go-nunu/nunu-layout-advanced/pkg/rabbitmq"
  10. "golang.org/x/sync/errgroup"
  11. "net"
  12. "sort"
  13. "strconv"
  14. "strings"
  15. )
  16. type WebForwardingService interface {
  17. GetWebForwarding(ctx context.Context, req v1.GetForwardingRequest) (v1.WebForwardingDataRequest, error)
  18. GetWebForwardingWafWebAllIps(ctx context.Context, req v1.GetForwardingRequest) ([]v1.WebForwardingDataRequest, error)
  19. AddWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) error
  20. EditWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) error
  21. DeleteWebForwarding(ctx context.Context, Ids []int) error
  22. }
  23. func NewWebForwardingService(
  24. service *Service,
  25. required RequiredService,
  26. webForwardingRepository repository.WebForwardingRepository,
  27. crawler CrawlerService,
  28. parser ParserService,
  29. wafformatter WafFormatterService,
  30. aoDun AoDunService,
  31. mq *rabbitmq.RabbitMQ,
  32. gatewayGroupIpRep repository.GateWayGroupIpRepository,
  33. gatewayGroupRep repository.GatewayGroupRepository,
  34. ) WebForwardingService {
  35. return &webForwardingService{
  36. Service: service,
  37. webForwardingRepository: webForwardingRepository,
  38. required: required,
  39. parser: parser,
  40. crawler: crawler,
  41. wafformatter: wafformatter,
  42. aoDun: aoDun,
  43. mq: mq,
  44. gatewayGroupIpRep: gatewayGroupIpRep,
  45. gatewayGroupRep: gatewayGroupRep,
  46. }
  47. }
  48. type webForwardingService struct {
  49. *Service
  50. webForwardingRepository repository.WebForwardingRepository
  51. required RequiredService
  52. parser ParserService
  53. crawler CrawlerService
  54. wafformatter WafFormatterService
  55. aoDun AoDunService
  56. mq *rabbitmq.RabbitMQ
  57. gatewayGroupIpRep repository.GateWayGroupIpRepository
  58. gatewayGroupRep repository.GatewayGroupRepository
  59. }
  60. func (s *webForwardingService) require(ctx context.Context,req v1.GlobalRequire) (v1.GlobalRequire, error) {
  61. var err error
  62. var res v1.GlobalRequire
  63. g, gCtx := errgroup.WithContext(ctx)
  64. g.Go(func() error {
  65. result, e := s.wafformatter.require(gCtx, req, "web")
  66. if e != nil {
  67. return e
  68. }
  69. res = result
  70. return nil
  71. })
  72. g.Go(func() error {
  73. e := s.wafformatter.validateWafDomainCount(gCtx, req)
  74. if e != nil {
  75. return e
  76. }
  77. return nil
  78. })
  79. if err = g.Wait(); err != nil {
  80. return v1.GlobalRequire{}, err
  81. }
  82. return res, nil
  83. }
  84. func (s *webForwardingService) GetWebForwarding(ctx context.Context, req v1.GetForwardingRequest) (v1.WebForwardingDataRequest, error) {
  85. var webForwarding model.WebForwarding
  86. var backend model.WebForwardingRule
  87. g, gCtx := errgroup.WithContext(ctx)
  88. g.Go(func() error {
  89. res, e := s.webForwardingRepository.GetWebForwarding(gCtx, int64(req.Id))
  90. if e != nil {
  91. // 直接返回错误,errgroup 会捕获它
  92. return fmt.Errorf("GetWebForwarding failed: %w", e)
  93. }
  94. if res != nil {
  95. webForwarding = *res
  96. }
  97. return nil
  98. })
  99. g.Go(func() error {
  100. res, e := s.webForwardingRepository.GetWebForwardingIpsByID(ctx, req.Id)
  101. if e != nil {
  102. return fmt.Errorf("GetWebForwardingByID failed: %w", e)
  103. }
  104. if res != nil {
  105. backend = *res
  106. }
  107. return nil
  108. })
  109. if err := g.Wait(); err != nil {
  110. return v1.WebForwardingDataRequest{}, err
  111. }
  112. return v1.WebForwardingDataRequest{
  113. Id: webForwarding.Id,
  114. WafWebId: webForwarding.WafWebId,
  115. Tag: webForwarding.Tag,
  116. Port: webForwarding.Port,
  117. Domain: webForwarding.Domain,
  118. CustomHost: webForwarding.CustomHost,
  119. WafWebLimitId: webForwarding.WebLimitRuleId,
  120. WafGatewayGroupId: webForwarding.WafGatewayGroupId,
  121. CcCount: webForwarding.CcCount,
  122. CcDuration: webForwarding.CcDuration,
  123. CcBlockCount: webForwarding.CcBlockCount,
  124. CcBlockDuration: webForwarding.CcBlockDuration,
  125. Cc4xxCount: webForwarding.Cc4xxCount,
  126. Cc4xxDuration: webForwarding.Cc4xxDuration,
  127. Cc4xxBlockCount: webForwarding.Cc4xxBlockCount,
  128. Cc4xxBlockDuration: webForwarding.Cc4xxBlockDuration,
  129. Cc5xxCount: webForwarding.Cc5xxCount,
  130. Cc5xxDuration: webForwarding.Cc5xxDuration,
  131. Cc5xxBlockCount: webForwarding.Cc5xxBlockCount,
  132. Cc5xxBlockDuration: webForwarding.Cc5xxBlockDuration,
  133. IsHttps: webForwarding.IsHttps,
  134. Comment: webForwarding.Comment,
  135. BackendList: backend.BackendList,
  136. AllowIpList: backend.AllowIpList,
  137. DenyIpList: backend.DenyIpList,
  138. AccessRule: backend.AccessRule,
  139. HttpsKey: webForwarding.HttpsKey,
  140. HttpsCert: webForwarding.HttpsCert,
  141. }, nil
  142. }
  143. // buildWafFormData 辅助函数,用于构建通用的 formData
  144. func (s *webForwardingService) buildWafFormData(req *v1.WebForwardingDataSend, require v1.GlobalRequire) map[string]interface{} {
  145. // 将BackendList序列化为JSON字符串
  146. backendJSON, err := json.MarshalIndent(req.BackendList, "", " ")
  147. var backendStr interface{}
  148. if err != nil {
  149. // 如果序列化失败,使用空数组
  150. backendStr = "[]"
  151. } else {
  152. // 成功序列化后,使用JSON字符串
  153. backendStr = string(backendJSON)
  154. }
  155. return map[string]interface{}{
  156. "waf_web_id": req.WafWebId,
  157. "tag": require.Tag,
  158. "port": req.Port,
  159. "domain": req.Domain,
  160. "custom_host": req.CustomHost,
  161. "waf_gateway_group_id": require.WafGatewayGroupId,
  162. "waf_web_limit_id": require.LimitRuleId,
  163. "cc_count": req.CcCount,
  164. "cc_duration": req.CcDuration,
  165. "cc_block_count": req.CcBlockCount,
  166. "cc_block_duration": req.CcBlockDuration,
  167. "cc_4xx_count": req.Cc4xxCount,
  168. "cc_4xx_duration": req.Cc4xxDuration,
  169. "cc_4xx_block_count": req.Cc4xxBlockCount,
  170. "cc_4xx_block_duration": req.Cc4xxBlockDuration,
  171. "cc_5xx_count": req.Cc5xxCount,
  172. "cc_5xx_duration": req.Cc5xxDuration,
  173. "cc_5xx_block_count": req.Cc5xxBlockCount,
  174. "cc_5xx_block_duration": req.Cc5xxBlockDuration,
  175. "backend": backendStr,
  176. "allow_ip_list": req.AllowIpList,
  177. "deny_ip_list": req.DenyIpList,
  178. "access_rule": req.AccessRule,
  179. "is_https": req.IsHttps,
  180. "comment": req.Comment,
  181. "https_cert": req.HttpsCert,
  182. "https_key": req.HttpsKey,
  183. }
  184. }
  185. // buildWebForwardingModel 辅助函数,用于构建通用的 WebForwarding 模型
  186. // ruleId 是从 WAF 系统获取的 ID
  187. func (s *webForwardingService) buildWebForwardingModel(req *v1.WebForwardingDataRequest,ruleId int, require v1.GlobalRequire) *model.WebForwarding {
  188. return &model.WebForwarding{
  189. HostId: require.HostId,
  190. WafWebId: ruleId,
  191. Tag: require.Tag,
  192. Port: req.Port,
  193. Domain: req.Domain,
  194. CustomHost: req.CustomHost,
  195. WafGatewayGroupId: require.WafGatewayGroupId,
  196. WebLimitRuleId: require.LimitRuleId,
  197. CcCount: req.CcCount,
  198. CcDuration: req.CcDuration,
  199. CcBlockCount: req.CcBlockCount,
  200. CcBlockDuration: req.CcBlockDuration,
  201. Cc4xxCount: req.Cc4xxCount,
  202. Cc4xxDuration: req.Cc4xxDuration,
  203. Cc4xxBlockCount: req.Cc4xxBlockCount,
  204. Cc4xxBlockDuration: req.Cc4xxBlockDuration,
  205. Cc5xxCount: req.Cc5xxCount,
  206. Cc5xxDuration: req.Cc5xxDuration,
  207. Cc5xxBlockCount: req.Cc5xxBlockCount,
  208. Cc5xxBlockDuration: req.Cc5xxBlockDuration,
  209. IsHttps: req.IsHttps,
  210. Comment: req.Comment,
  211. HttpsCert: req.HttpsCert,
  212. HttpsKey: req.HttpsKey,
  213. }
  214. }
  215. func (s *webForwardingService) buildWebRuleModel(reqData *v1.WebForwardingDataRequest, require v1.GlobalRequire, localDbId int) *model.WebForwardingRule {
  216. return &model.WebForwardingRule{
  217. Uid: require.Uid,
  218. HostId: require.HostId,
  219. WebId: localDbId, // 关联到本地数据库的主记录 ID
  220. BackendList: reqData.BackendList,
  221. AllowIpList: reqData.AllowIpList,
  222. DenyIpList: reqData.DenyIpList,
  223. AccessRule: reqData.AccessRule,
  224. }
  225. }
  226. func (s *webForwardingService) prepareWafData(ctx context.Context, req *v1.WebForwardingRequest) (v1.GlobalRequire, map[string]interface{}, error) {
  227. // 1. 获取必要的全局信息
  228. require, err := s.require(ctx, v1.GlobalRequire{
  229. HostId: req.HostId,
  230. Uid: req.Uid,
  231. Comment: req.WebForwardingData.Comment,
  232. Domain: req.WebForwardingData.Domain,
  233. })
  234. if err != nil {
  235. return v1.GlobalRequire{}, nil, err
  236. }
  237. if require.WafGatewayGroupId == 0 || require.LimitRuleId == 0 {
  238. return v1.GlobalRequire{}, nil, fmt.Errorf("请先配置实例")
  239. }
  240. // 2. 将字符串切片拼接成字符串,用于 WAF API
  241. allowIpListStr := strings.Join(req.WebForwardingData.AllowIpList, "\n")
  242. denyIpListStr := strings.Join(req.WebForwardingData.DenyIpList, "\n")
  243. PortInt, err := strconv.Atoi(req.WebForwardingData.Port)
  244. if err != nil {
  245. return v1.GlobalRequire{}, nil, err
  246. }
  247. // 3. 创建用于构建 WAF 表单的数据结构
  248. formDataBase := v1.WebForwardingDataSend{
  249. Tag: require.Tag,
  250. WafWebId: req.WebForwardingData.WafWebId,
  251. WafGatewayGroupId: require.WafGatewayGroupId,
  252. WafWebLimitId: require.LimitRuleId,
  253. Port: PortInt,
  254. Domain: req.WebForwardingData.Domain,
  255. CustomHost: req.WebForwardingData.CustomHost,
  256. CcCount: req.WebForwardingData.CcCount,
  257. CcDuration: req.WebForwardingData.CcDuration,
  258. CcBlockCount: req.WebForwardingData.CcBlockCount,
  259. CcBlockDuration: req.WebForwardingData.CcBlockDuration,
  260. Cc4xxCount: req.WebForwardingData.Cc4xxCount,
  261. Cc4xxDuration: req.WebForwardingData.Cc4xxDuration,
  262. Cc4xxBlockCount: req.WebForwardingData.Cc4xxBlockCount,
  263. Cc4xxBlockDuration: req.WebForwardingData.Cc4xxBlockDuration,
  264. Cc5xxCount: req.WebForwardingData.Cc5xxCount,
  265. Cc5xxDuration: req.WebForwardingData.Cc5xxDuration,
  266. Cc5xxBlockCount: req.WebForwardingData.Cc5xxBlockCount,
  267. Cc5xxBlockDuration: req.WebForwardingData.Cc5xxBlockDuration,
  268. IsHttps: req.WebForwardingData.IsHttps,
  269. BackendList: req.WebForwardingData.BackendList,
  270. AllowIpList: allowIpListStr,
  271. DenyIpList: denyIpListStr,
  272. AccessRule: req.WebForwardingData.AccessRule,
  273. Comment: req.WebForwardingData.Comment,
  274. HttpsCert: req.WebForwardingData.HttpsCert,
  275. HttpsKey: req.WebForwardingData.HttpsKey,
  276. }
  277. // 4. 构建 WAF 表单数据映射
  278. formData := s.buildWafFormData(&formDataBase, require)
  279. return require, formData, nil
  280. }
  281. func (s *webForwardingService) AddWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) error {
  282. require, formData, err := s.prepareWafData(ctx, req)
  283. if err != nil {
  284. return err
  285. }
  286. err = s.wafformatter.validateWafPortCount(ctx, require.HostId)
  287. if err != nil {
  288. return err
  289. }
  290. wafWebId, err := s.wafformatter.sendFormData(ctx, "admin/info/waf_web/new", "admin/new/waf_web", formData)
  291. if err != nil {
  292. return err
  293. }
  294. gatewayIps, firstIp, err := s.wafformatter.GetIp(ctx, require.WafGatewayGroupId)
  295. if err != nil {
  296. return err
  297. }
  298. if req.WebForwardingData.Domain != "" {
  299. // 异步任务:将域名添加到白名单
  300. doMain, err := s.wafformatter.ConvertToWildcardDomain(ctx, req.WebForwardingData.Domain)
  301. if err != nil {
  302. return err
  303. }
  304. go s.wafformatter.PublishDomainWhitelistTask(doMain,firstIp, "add")
  305. }
  306. // IP过白
  307. var ips []string
  308. if req.WebForwardingData.BackendList != nil {
  309. for _, v := range req.WebForwardingData.BackendList {
  310. ip, _, err := net.SplitHostPort(v.Addr)
  311. if err != nil {
  312. return err
  313. }
  314. ips = append(ips,ip)
  315. }
  316. go s.wafformatter.PublishIpWhitelistTask(ips, "add","")
  317. }
  318. var accessRuleIps []string
  319. if len(req.WebForwardingData.AllowIpList) > 0 {
  320. for _, v := range gatewayIps {
  321. for _, ip := range req.WebForwardingData.AllowIpList {
  322. if net.ParseIP(ip) != nil{
  323. accessRuleIps = append(accessRuleIps, ip)
  324. }
  325. }
  326. go s.wafformatter.PublishIpWhitelistTask(accessRuleIps, "add",v)
  327. }
  328. }
  329. webModel := s.buildWebForwardingModel(&req.WebForwardingData, wafWebId, require)
  330. id, err := s.webForwardingRepository.AddWebForwarding(ctx, webModel)
  331. if err != nil {
  332. return err
  333. }
  334. webRuleModel := s.buildWebRuleModel(&req.WebForwardingData, require, id)
  335. if _, err = s.webForwardingRepository.AddWebForwardingIps(ctx, *webRuleModel); err != nil {
  336. return err
  337. }
  338. return nil
  339. }
  340. func (s *webForwardingService) EditWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) error {
  341. WafWebId, err := s.webForwardingRepository.GetWebForwardingWafWebIdById(ctx, req.WebForwardingData.Id)
  342. if err != nil {
  343. return err
  344. }
  345. req.WebForwardingData.WafWebId = WafWebId
  346. require, formData, err := s.prepareWafData(ctx, req)
  347. if err != nil {
  348. return err
  349. }
  350. _, err = s.wafformatter.sendFormData(ctx, "admin/info/waf_web/edit?&__goadmin_edit_pk="+strconv.Itoa(req.WebForwardingData.WafWebId), "admin/edit/waf_web", formData)
  351. if err != nil {
  352. return err
  353. }
  354. // 将域名添加到白名单
  355. webData, err := s.webForwardingRepository.GetWebForwarding(ctx, int64(req.WebForwardingData.Id))
  356. if err != nil {
  357. return err
  358. }
  359. gatewayIps, firstIp, err := s.wafformatter.GetIp(ctx, webData.WafGatewayGroupId)
  360. if err != nil {
  361. return err
  362. }
  363. // 异步任务:将域名添加到白名单
  364. if webData.Domain != req.WebForwardingData.Domain {
  365. doMain, err := s.wafformatter.ConvertToWildcardDomain(ctx, req.WebForwardingData.Domain)
  366. if err != nil {
  367. return err
  368. }
  369. oldDomain, err := s.wafformatter.ConvertToWildcardDomain(ctx, webData.Domain)
  370. if err != nil {
  371. return err
  372. }
  373. go s.wafformatter.PublishDomainWhitelistTask(oldDomain, firstIp, "del")
  374. go s.wafformatter.PublishDomainWhitelistTask(doMain, firstIp, "add")
  375. }
  376. // IP过白
  377. ipData, err := s.webForwardingRepository.GetWebForwardingIpsByID(ctx, req.WebForwardingData.Id)
  378. if err != nil {
  379. return err
  380. }
  381. var oldIps []string
  382. var oldAllowIps []string
  383. var newIps []string
  384. var newAllowIps []string
  385. for _, v := range ipData.BackendList {
  386. ip, _, err := net.SplitHostPort(v.Addr)
  387. if err != nil {
  388. return err
  389. }
  390. oldIps = append(oldIps, ip)
  391. }
  392. for _, v := range req.WebForwardingData.BackendList {
  393. ip, _, err := net.SplitHostPort(v.Addr)
  394. if err != nil {
  395. return err
  396. }
  397. newIps = append(newIps, ip)
  398. }
  399. addedIps, removedIps := s.wafformatter.findIpDifferences(oldIps, newIps)
  400. if len(addedIps) > 0 {
  401. go s.wafformatter.PublishIpWhitelistTask(addedIps, "add","")
  402. }
  403. if len(removedIps) > 0 {
  404. go s.wafformatter.PublishIpWhitelistTask(removedIps, "del","")
  405. }
  406. if len(ipData.AllowIpList) > 0 {
  407. for _, v := range ipData.AllowIpList {
  408. if net.ParseIP(v) != nil{
  409. oldAllowIps = append(oldAllowIps, v)
  410. }
  411. }
  412. }
  413. if len(req.WebForwardingData.AllowIpList) > 0 {
  414. for _, v := range req.WebForwardingData.AllowIpList {
  415. if net.ParseIP(v) != nil{
  416. newAllowIps = append(newAllowIps, v)
  417. }
  418. }
  419. }
  420. addedAllowIps, removedAllowIps := s.wafformatter.findIpDifferences(oldAllowIps, newAllowIps)
  421. for _, v := range gatewayIps {
  422. if len(addedAllowIps) > 0 {
  423. go s.wafformatter.PublishIpWhitelistTask(addedAllowIps, "add",v)
  424. }
  425. if len(removedAllowIps) > 0 {
  426. go s.wafformatter.PublishIpWhitelistTask(removedAllowIps, "del",v)
  427. }
  428. }
  429. webModel := s.buildWebForwardingModel(&req.WebForwardingData, req.WebForwardingData.WafWebId, require)
  430. webModel.Id = req.WebForwardingData.Id
  431. if err = s.webForwardingRepository.EditWebForwarding(ctx, webModel); err != nil {
  432. return err
  433. }
  434. webRuleModel := s.buildWebRuleModel(&req.WebForwardingData, require, req.WebForwardingData.Id)
  435. if err = s.webForwardingRepository.EditWebForwardingIps(ctx, *webRuleModel); err != nil {
  436. return err
  437. }
  438. return nil
  439. }
  440. func (s *webForwardingService) DeleteWebForwarding(ctx context.Context, Ids []int) error {
  441. for _, Id := range Ids {
  442. wafWebId, err := s.webForwardingRepository.GetWebForwardingWafWebIdById(ctx, Id)
  443. if err != nil {
  444. return err
  445. }
  446. _, err = s.crawler.DeleteRule(ctx, wafWebId, "admin/delete/waf_web?page=1&__pageSize=10&__sort=waf_web_id&__sort_type=desc")
  447. if err != nil {
  448. return err
  449. }
  450. webData, err := s.webForwardingRepository.GetWebForwarding(ctx, int64(Id))
  451. if err != nil {
  452. return err
  453. }
  454. _, firstIp, err := s.wafformatter.GetIp(ctx, webData.WafGatewayGroupId)
  455. if err != nil {
  456. return err
  457. }
  458. // 异步任务:将域名添加到白名单
  459. if webData.Domain != "" {
  460. doMain, err := s.wafformatter.ConvertToWildcardDomain(ctx, webData.Domain)
  461. if err != nil {
  462. return err
  463. }
  464. go s.wafformatter.PublishDomainWhitelistTask(doMain,firstIp, "del")
  465. }
  466. // IP过白
  467. ipData, err := s.webForwardingRepository.GetWebForwardingIpsByID(ctx, Id)
  468. if err != nil {
  469. return err
  470. }
  471. var ips []string
  472. if len(ipData.BackendList) > 0 {
  473. for _, v := range ipData.BackendList {
  474. ip, _, err := net.SplitHostPort(v.Addr)
  475. if err != nil {
  476. return err
  477. }
  478. ips = append(ips, ip)
  479. }
  480. }
  481. if len(ipData.AllowIpList) > 0 {
  482. ips = append(ips, ipData.AllowIpList...)
  483. }
  484. if len(ips) > 0 {
  485. go s.wafformatter.PublishIpWhitelistTask(ips, "del","")
  486. }
  487. if err = s.webForwardingRepository.DeleteWebForwarding(ctx, int64(Id)); err != nil {
  488. return err
  489. }
  490. if err = s.webForwardingRepository.DeleteWebForwardingIpsById(ctx, Id); err != nil {
  491. return err
  492. }
  493. }
  494. return nil
  495. }
  496. func (s *webForwardingService) GetWebForwardingWafWebAllIps(ctx context.Context, req v1.GetForwardingRequest) ([]v1.WebForwardingDataRequest, error) {
  497. type CombinedResult struct {
  498. Id int
  499. Forwarding *model.WebForwarding
  500. BackendRule *model.WebForwardingRule
  501. Err error // 如果此ID的处理出错,则携带错误
  502. }
  503. g, gCtx := errgroup.WithContext(ctx)
  504. ids, err := s.webForwardingRepository.GetWebForwardingWafWebAllIds(gCtx, req.HostId)
  505. if err != nil {
  506. return nil, fmt.Errorf("GetWebForwardingWafWebAllIds failed: %w", err)
  507. }
  508. if len(ids) == 0 {
  509. return nil, nil // 没有ID,直接返回空切片
  510. }
  511. // 创建一个通道来接收每个ID的处理结果
  512. // 通道缓冲区大小设为ID数量,这样发送者不会因为接收者慢而阻塞(在所有goroutine都启动后)
  513. resultsChan := make(chan CombinedResult, len(ids))
  514. for _, idVal := range ids {
  515. currentID := idVal // 捕获循环变量
  516. g.Go(func() error {
  517. var wf *model.WebForwarding
  518. var bk *model.WebForwardingRule
  519. var localErr error
  520. // 1. 获取 WebForwarding 信息
  521. wf, localErr = s.webForwardingRepository.GetWebForwarding(gCtx, int64(currentID))
  522. if localErr != nil {
  523. // 发送错误到通道,并由 errgroup 捕获
  524. // errgroup 会处理第一个非nil错误,并取消其他 goroutine
  525. resultsChan <- CombinedResult{Id: currentID, Err: fmt.Errorf("GetWebForwarding for id %d failed: %w", currentID, localErr)}
  526. return localErr // 返回错误给 errgroup
  527. }
  528. if wf == nil { // 正常情况下,如果没错误,wf不应为nil,但防御性检查
  529. localErr = fmt.Errorf("GetWebForwarding for id %d returned nil data without error", currentID)
  530. resultsChan <- CombinedResult{Id: currentID, Err: localErr}
  531. return localErr
  532. }
  533. // 2. 获取 Backend IP 信息
  534. // 注意:这里我们允许 GetWebForwardingIpsByID 可能返回 nil 数据(例如没有规则)而不是错误
  535. // 如果它也可能返回错误,则处理方式与上面类似
  536. bk, localErr = s.webForwardingRepository.GetWebForwardingIpsByID(gCtx, currentID)
  537. if localErr != nil {
  538. // 如果获取IP信息失败是一个致命错误,则也应返回错误
  539. // 如果允许部分成功(比如有WebForwarding但没有IP信息),则可以不将此视为errgroup的错误
  540. // 这里假设它也是一个需要errgroup捕获的错误
  541. resultsChan <- CombinedResult{Id: currentID, Forwarding: wf, Err: fmt.Errorf("GetWebForwardingIpsByID for id %d failed: %w", currentID, localErr)}
  542. return localErr // 返回错误给 errgroup
  543. }
  544. // bk 可能是 nil 如果没有错误且没有规则,这取决于业务逻辑
  545. // 发送成功的结果到通道
  546. resultsChan <- CombinedResult{Id: currentID, Forwarding: wf, BackendRule: bk}
  547. return nil // 此goroutine成功
  548. })
  549. }
  550. // 等待所有goroutine完成
  551. groupErr := g.Wait()
  552. // 关闭通道,表示所有发送者都已完成
  553. // 这一步很重要,这样下面的 range 循环才能正常结束
  554. close(resultsChan)
  555. // 如果 errgroup 捕获到任何错误,优先返回该错误
  556. if groupErr != nil {
  557. // 虽然errgroup已经出错了,但通道中可能已经有一些结果(来自出错前成功或出错的goroutine)
  558. // 我们需要排空通道以避免goroutine泄漏(如果它们在发送时阻塞)
  559. // 但由于我们优先返回groupErr,这些结果将被丢弃。
  560. // 在这种设计下,通常任何一个子任务失败都会导致整个操作失败。
  561. return nil, groupErr
  562. }
  563. // 如果没有错误,收集所有成功的结果
  564. finalResults := make([]v1.WebForwardingDataRequest, 0, len(ids))
  565. for res := range resultsChan {
  566. // 再次检查通道中的错误,尽管 errgroup 应该已经捕获了
  567. // 但这是一种更细致的错误处理,以防万一有goroutine在errgroup.Wait()前发送了错误但未被errgroup捕获
  568. // (理论上,如果goroutine返回了错误,errgroup会处理)
  569. // 主要目的是处理 res.forwarding 为 nil 的情况 (如果上面允许不返回错误)
  570. if res.Err != nil {
  571. // 如果到这里还有错误,说明逻辑可能有问题,或者我们决定忽略某些类型的子错误
  572. // 在此示例中,因为 g.Wait() 没有错误,所以这里的 res.err 应该是nil
  573. // 如果不是,那么可能是goroutine在return nil前发送了带有错误的res。
  574. // 严格来说,如果errgroup没有错误,这里res.err也应该是nil
  575. // 但以防万一,我们可以记录日志
  576. return nil, fmt.Errorf("received error from goroutine for ID %d: %w", res.Id, res.Err)
  577. }
  578. if res.Forwarding == nil {
  579. return nil, fmt.Errorf("received nil forwarding from goroutine for ID %d", res.Id)
  580. }
  581. dataReq := v1.WebForwardingDataRequest{
  582. Id: res.Forwarding.Id,
  583. Port: res.Forwarding.Port,
  584. Domain: res.Forwarding.Domain,
  585. CustomHost: res.Forwarding.CustomHost,
  586. CcCount: res.Forwarding.CcCount,
  587. CcDuration: res.Forwarding.CcDuration,
  588. CcBlockCount: res.Forwarding.CcBlockCount,
  589. CcBlockDuration: res.Forwarding.CcBlockDuration,
  590. Cc4xxCount: res.Forwarding.Cc4xxCount,
  591. Cc4xxDuration: res.Forwarding.Cc4xxDuration,
  592. Cc4xxBlockCount: res.Forwarding.Cc4xxBlockCount,
  593. Cc4xxBlockDuration: res.Forwarding.Cc4xxBlockDuration,
  594. Cc5xxCount: res.Forwarding.Cc5xxCount,
  595. Cc5xxDuration: res.Forwarding.Cc5xxDuration,
  596. Cc5xxBlockCount: res.Forwarding.Cc5xxBlockCount,
  597. Cc5xxBlockDuration: res.Forwarding.Cc5xxBlockDuration,
  598. IsHttps: res.Forwarding.IsHttps,
  599. Comment: res.Forwarding.Comment,
  600. HttpsKey: res.Forwarding.HttpsKey,
  601. HttpsCert: res.Forwarding.HttpsCert,
  602. }
  603. if res.BackendRule != nil { // 只有当 BackendRule 存在时才填充相关字段
  604. dataReq.BackendList = res.BackendRule.BackendList
  605. dataReq.AllowIpList = res.BackendRule.AllowIpList
  606. dataReq.DenyIpList = res.BackendRule.DenyIpList
  607. dataReq.AccessRule = res.BackendRule.AccessRule
  608. }
  609. finalResults = append(finalResults, dataReq)
  610. }
  611. sort.Slice(finalResults, func(i, j int) bool {
  612. return finalResults[i].Id > finalResults[j].Id
  613. })
  614. return finalResults, nil
  615. }