众所周知,Golang中不支持类似C++/Java中的标记式泛型,所以对于常用算法,比如冒泡排序算法,有些同学容易写出逻辑上重复的代码,即整型是第一套代码,字符串型是第二套代码,用户自定义类型是第三套代码。
重复是万恶之源,我们当然不能容忍,所以要消除重复,使得代码保持在最佳的状态。本文通过一个实际使用的简单算法的演进过程,初次体验了Golang的泛型编程,消除了重复代码,非常自然。

切片算法支持整型

  1. type Slice []int
  2. func NewSlice() Slice {
  3. return make(Slice, 0)
  4. }
  5. func (this* Slice) Add(elem int) error {
  6. for _, v := range *this {
  7. if v == elem {
  8. fmt.Printf("Slice:Add elem: %v already exist\n", elem)
  9. return ERR_ELEM_EXIST
  10. }
  11. }
  12. *this = append(*this, elem)
  13. fmt.Printf("Slice:Add elem: %v succ\n", elem)
  14. return nil
  15. }
  16. func (this* Slice) Remove(elem int) error {
  17. found := false
  18. for i, v := range *this {
  19. if v == elem {
  20. if i == len(*this) - 1 {
  21. *this = (*this)[:i]
  22. } else {
  23. *this = append((*this)[:i], (*this)[i+1:]...)
  24. }
  25. found = true
  26. break
  27. }
  28. }
  29. if !found {
  30. fmt.Printf("Slice:Remove elem: %v not exist\n", elem)
  31. return ERR_ELEM_NT_EXIST
  32. }
  33. fmt.Printf("Slice:Remove elem: %v succ\n", elem)
  34. return nil
  35. }

切片算法支持字符串

  1. type Slice []interface{}
  2. func NewSlice() Slice {
  3. return make(Slice, 0)
  4. }
  5. func (this* Slice) Add(elem interface{}) error {
  6. for _, v := range *this {
  7. if v == elem {
  8. fmt.Printf("Slice:Add elem: %v already exist\n", elem)
  9. return ERR_ELEM_EXIST
  10. }
  11. }
  12. *this = append(*this, elem)
  13. fmt.Printf("Slice:Add elem: %v succ\n", elem)
  14. return nil
  15. }
  16. func (this* Slice) Remove(elem interface{}) error {
  17. found := false
  18. for i, v := range *this {
  19. if v == elem {
  20. if i == len(*this) - 1 {
  21. *this = (*this)[:i]
  22. } else {
  23. *this = append((*this)[:i], (*this)[i+1:]...)
  24. }
  25. found = true
  26. break
  27. }
  28. }
  29. if !found {
  30. fmt.Printf("Slice:Remove elem: %v not exist\n", elem)
  31. return ERR_ELEM_NT_EXIST
  32. }
  33. fmt.Printf("Slice:Remove elem: %v succ\n", elem)
  34. return nil
  35. }

切片算法支持用户自定义的类型

自定义一个类型:

  1. type Student struct {
  2. id string
  3. name string
  4. }

Student类型有两个数据成员,即id和name。id是学号,全局我唯一;name是中文名字的拼音,可重复。

用户自定义类型和基本类型(int或string)不同的是两个元素是否相等的判断方式不一样:

  1. 基本类型(int或string)直接通过”==“运算符来判断;

  2. 用户自定义类型万千种种,数组切片算法中不可能知道,所以需要通过interface提供的方法进行两个元素是否相等的判断。

我们接着定义一个interface:

  1. type Comparable interface {
  2. IsEqual(obj interface{}) bool
  3. }

只要用户自定义的类型实现了接口Comparable,就可以调用它的方法IsEqual进行两个元素是否相等的判断了,于是我们实现了Student类型的IsEqual方法:

  1. func (this Student) IsEqual(obj interface{}) bool {
  2. if student, ok := obj.(Student); ok {
  3. return this.GetId() == student.GetId()
  4. }
  5. panic("unexpected type")
  6. }
  7. func (this Student) GetId() string {
  8. return this.id
  9. }

用户自定义的GetId方法是必要的,因为Id不一定就是数据成员,可能是由多个数据成员拼接而成。
我们将数组切片算法的易变部分”v == elem”抽出来封装成方法:

  1. func isEqual(a, b interface{}) bool {
  2. return a == b
  3. }

于是数组切片的Add方法和Remove方法就变成:

  1. func (this* Slice) Add(elem interface{}) error {
  2. for _, v := range *this {
  3. if isEqual(v, elem) {
  4. fmt.Printf("Slice:Add elem: %v already exist\n", elem)
  5. return ERR_ELEM_EXIST
  6. }
  7. }
  8. *this = append(*this, elem)
  9. fmt.Printf("Slice:Add elem: %v succ\n", elem)
  10. return nil
  11. }
  12. func (this* Slice) Remove(elem interface{}) error {
  13. found := false
  14. for i, v := range *this {
  15. if isEqual(v, elem) {
  16. if i == len(*this) - 1 {
  17. *this = (*this)[:i]
  18. } else {
  19. *this = append((*this)[:i], (*this)[i+1:]...)
  20. }
  21. found = true
  22. break
  23. }
  24. }
  25. if !found {
  26. fmt.Printf("Slice:Remove elem: %v not exist\n", elem)
  27. return ERR_ELEM_NT_EXIST
  28. }
  29. fmt.Printf("Slice:Remove elem: %v succ\n", elem)
  30. return nil
  31. }

于是数组切片算法对于支持用户自定义类型的改动仅仅局限于isEqual函数了,我们通过接口查询来完成代码修改:

  1. func isEqual(a, b interface{}) bool {
  2. if comparable, ok := a.(Comparable); ok {
  3. return comparable.IsEqual(b)
  4. } else {
  5. return a == b
  6. }
  7. }

golang 泛型编程 - 图1