Elegant way to eliminate multiple condition checks in function

huangapple
go评论105阅读模式
英文:

Elegant way to eliminate multiple condition checks in function

问题

有多个函数中有多个条件检查。

  1. type VA struct {
  2. 	A string
  3. }
  5. func (va *VA) CheckA(s string) error {
  6. 	if s != va.A {
  7. 		return errors.New("无效的字符串")
  8. 	}
  9. 	return nil
  10. }
  12. type VB struct {
  13. 	B int
  14. }
  16. func (vb *VB) CheckB(i int) error {
  17. 	if i == vb.B {
  18. 		return errors.New("无效的整数")
  19. 	}
  20. 	return nil
  21. }
  23. func FuncA(s string, i int) error {
  24. 	a := &VA{A: "testa"}
  25. 	errA := a.CheckA(s)
  26. 	if errA != nil {
  27. 		return errA
  28. 	}
  30. 	b := &VB{B: 3}
  31. 	errB := b.CheckB(i)
  32. 	if errB != nil {
  33. 		return errB
  34. 	}
  36. 	// 更多逻辑...
  38. 	return nil
  39. }
  41. func FuncB(sb string, v int32) error {
  42. 	a := &VA{A: "testb"}
  43. 	errA := a.CheckA(sb)
  44. 	if errA != nil {
  45. 		return errA
  46. 	}
  48. 	// 更多逻辑...
  50. 	return nil
  51. }
  53. func FuncC(sc string, vv int) error {
  54. 	b := &VB{B: 3}
  55. 	errB := b.CheckB(vv)
  56. 	if errB != nil {
  57. 		return errB
  58. 	}
  60. 	// 更多逻辑...
  62. 	return nil
  63. }

我们在函数FuncA中进行了CheckA和CheckB，在函数FuncB中进行了CheckA。然而，在函数FuncC中只进行了CheckB。有一个问题是，当CheckA的返回值发生变化时，FuncA和FuncB都需要进行修改。

我们想要重构上述代码。在Golang中是否有一种优雅的方式来实现这一点？

我们尝试将CheckA和CheckB合并到一个名为ValidateFunc的函数中，如下所示：

  1. type VALIDATE int
  3. const (
  4. 	_ VALIDATE = 1 << iota
  5. 	VALIDATEA
  6. 	VALIDATEB
  7. )
  9. func ValidateFunc(vs, s string, vi, i int, validate VALIDATE) error {
  10. 	if validate&VALIDATEA == VALIDATEA {
  11. 		a := &VA{A: vs}
  12. 		errA := a.CheckA(s)
  13. 		if errA != nil {
  14. 			return errA
  15. 		}
  16. 	}
  17. 	if validate&VALIDATEB == VALIDATEB {
  18. 		b := &VB{B: vi}
  19. 		errB := b.CheckB(i)
  20. 		if errB != nil {
  21. 			return errB
  22. 		}
  23. 	}
  24. 	return nil
  25. }
  27. func FuncA(s string, i int) error {
  28. 	err := ValidateFunc("testa", s, 3, i, VALIDATEA|VALIDATEB)
  29. 	if err != nil {
  30. 		return err
  31. 	}
  33. 	// 更多逻辑...
  35. 	return nil
  36. }
英文:

There are multiple condition checks in multiple functions

  1. type VA struct {
  2. 	A string
  3. }
  5. func (va *VA) CheckA(s string) error {
  6. 	if s != va.A {
  7. 		return errors.New(&quot;invalid str &quot;)
  8. 	}
  9. 	return nil
  10. }
  12. type VB struct {
  13. 	B int
  14. }
  16. func (vb *VB) CheckB(i int) error {
  17. 	if i == vb.B {
  18. 		return errors.New(&quot;invalid int&quot;)
  19. 	}
  20. 	return nil
  21. }
  23. func FuncA(s string, i int) error {
  24. 	a := &amp;VA{A: &quot;testa&quot;}
  25. 	errA := a.CheckA(s)
  26. 	if errA != nil {
  27. 		return errA
  28. 	}
  30. 	b := &amp;VB{B: 3}
  31. 	errB := b.CheckB(i)
  32. 	if errB != nil {
  33. 		return errB
  34. 	}
  36. 	// more logic ...
  38. 	return nil
  39. }
  41. func FuncB(sb string, v int32) error {
  42. 	a := &amp;VA{A: &quot;testb&quot;}
  43. 	errA := a.CheckA(sb)
  44. 	if errA != nil {
  45. 		return errA
  46. 	}
  48. 	// more logic ...
  50. 	return nil
  51. }
  53. func FuncC(sc string, vv int) error {
  54. 	b := &amp;VB{B: 3}
  55. 	errB := b.CheckB(vv)
  56. 	if errB != nil {
  57. 		return errB
  58. 	}
  60. 	// more logic ...
  62. 	return nil
  63. }

We do CheckA and CheckB in function FuncA and do CheckA in function FuncB. However, only do CheckB in function FuncC. There is one pitfall that when the return value of CheckA is changed, both FuncA and FuncB would be changed.

We want to refactor the above codes. Is there any elegant way to do that in Golang?

What we have tried, combine CheckA and CheckB in one function ValidateFunc like below

  1. type VALIDATE int
  3. const (
  4. 	_ VALIDATE = 1 &lt;&lt; iota
  5. 	VALIDATEA
  6. 	VALIDATEB
  7. )
  9. func ValidateFunc(vs, s string, vi, i int, validate VALIDATE) error {
  10. 	if validate&amp;VALIDATEA == VALIDATEA {
  11. 		a := &amp;VA{A: vs}
  12. 		errA := a.CheckA(s)
  13. 		if errA != nil {
  14. 			return errA
  15. 		}
  16. 	}
  17. 	if validate&amp;VALIDATEB == VALIDATEB {
  18. 		b := &amp;VB{B: vi}
  19. 		errB := b.CheckB(i)
  20. 		if errB != nil {
  21. 			return errB
  22. 		}
  23. 	}
  24. 	return nil
  25. }
  27. func FuncA(s string, i int) error {
  28. 	err := ValidateFunc(&quot;testa&quot;, s, 3, i, VALIDATEA|VALIDATEB)
  29. 	if err != nil {
  30. 		return err
  31. 	}
  33. 	// more logic ...
  35. 	return nil
  36. }

答案1

得分: 1

参考选项模式，代码似乎比以前更简单。

  1. type Validator struct {
  2. 	A VA
  3. 	B VB
  4. }
  6. type Validate func(v *Validator) error
  8. func WithVA(s string) Validate {
  9. 	return func(v *Validator) error {
  10. 		if err := v.A.CheckA(s); err != nil {
  11. 			return err
  12. 		}
  13. 		return nil
  14. 	}
  15. }
  17. func WithVB(i int) Validate {
  18. 	return func(v *Validator) error {
  19. 		if err := v.B.CheckB(i); err != nil {
  20. 			return err
  21. 		}
  22. 		return nil
  23. 	}
  24. }
  26. func DoValidate(vs string, vi int, vals ...func(v *Validator) error) error {
  27. 	v := &Validator{A: VA{A: vs}, B: VB{B: vi}}
  29. 	for _, val := range vals {
  30. 		if err := val(v); err != nil {
  31. 			return err
  32. 		}
  33. 	}
  35. 	return nil
  36. }
  38. func FuncA(s string, i int) error {
  39. 	err := DoValidate("testa", 3, WithVA(s), WithVB(i))
  40. 	if err != nil {
  41. 		return err
  42. 	}
  44. 	// more logic ...
  46. 	return nil
  47. }
英文:

Refer to Option pattern, it seems the codes will be simpler than before.

  1. type Validator struct {
  2. 	A VA
  3. 	B VB
  4. }
  6. type Validate func(v *Validator) error
  8. func WithVA(s string) Validate {
  9. 	return func(v *Validator) error {
  10. 		if err := v.A.CheckA(s); err != nil {
  11. 			return err
  12. 		}
  13. 		return nil
  14. 	}
  15. }
  17. func WithVB(i int) Validate {
  18. 	return func(v *Validator) error {
  19. 		if err := v.B.CheckB(i); err != nil {
  20. 			return err
  21. 		}
  22. 		return nil
  23. 	}
  24. }
  26. func DoValidate(vs string, vi int, vals ...func(v *Validator) error) error {
  27. 	v := &amp;Validator{A: VA{A: vs}, B: VB{B: vi}}
  29. 	for _, val := range vals {
  30. 		if err := val(v); err != nil {
  31. 			return err
  32. 		}
  33. 	}
  35. 	return nil
  36. }
  38. func FuncA(s string, i int) error {
  39. 	err := DoValidate(&quot;testa&quot;, 3, WithVA(s), WithVB(i))
  40. 	if err != nil {
  41. 		return err
  42. 	}
  44. 	// more logic ...
  46. 	return nil
  47. }

答案2

得分: -1

也许将VA和VB合并成一个新的结构体，然后验证它？

英文:

Perhaps combine VA & VB into a new struct
Then validate that one?

huangapple
  • 本文由 发表于 2022年1月25日 16:54:16
  • 转载请务必保留本文链接：https://go.coder-hub.com/70845815.html
  • go
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定