欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Golang 使用接口实现泛型的方法示例

程序员文章站 2023-09-06 19:11:27
在c/c++中我们可以使用泛型的方法使代码得以重复使用,最常见例如stl functions:vector vint or vector

在c/c++中我们可以使用泛型的方法使代码得以重复使用,最常见例如stl functions:vector<int> vint or vector<float> vfloat等。这篇文章将使用interface{...}接口使golang实现泛型。

interface{...}是实现泛型的基础。如一个数组元素类型是interface{...}的话,那么实现了该接口的实体都可以被放置入数组中。注意其中并不一定必须是空接口(简单类型我们可以通过把他转化为自定义类型后实现接口)。为什么interface中要声明方法:因为当我们需要对数组内数据进行操作时(如比较大小),我们需要为这个操作声明一个自定义的方法。换言之,只有实现了这个方法的实体才允许被加入进数组中。

基础demo

在下面演示的demo中,我们将实现一个最简单的vector,并实现插入时排序的功能。

type comper interface{
  lessthan (comper) bool
}
type sdata struct{
  data []comper
}

func (t *sdata) push (item comper){
  t.data = append(t.data, item)
  for k,v:=range t.data{
    if item.lessthan(v) {  //调用接口定义的方法
      //排序操作
      break
    }
  }
}

如此便实现了一个最简单的demo,使用sdata的数组元素必须先实现lessthan方法:

type myint int

func (t myint) lessthan (x comper) bool {
  return t<x.(myint)
}
func main() {
  mydata := sdata{make([]comper, 0)}
  for i:=10;i>0;i--{
    mydata.push((myint(i)))
  }
  fmt.println(mydata)
}

但这个demo的缺点也有许多,一是简单类型元素无法使用sdata进行排序,二是不支持并发,在并发的情况下会产生不可预料的结果。

通过reflect支持简单类型的demo

为要支持简单类型,我们只能使用空接口作为数组元素类型。这时候我们的程序逻辑应该是这样:如果这是一个简单类型,那么我们直接调用内置的"<"与">"进行比较;如果这不是一个简单类型,那么我们仍旧调用lessthan方法:

type comper interface{
  lessthan (comper) bool
}
type sdata struct{
  data []interface{}
}

func (t *sdata) push (item interface{}){
  for _,v:=range t.data{
    if reflect.typeof(item).implements(reflect.typeof(new(comper)).elem()) {
      citem:=item.(comper)
      cv:=v.(comper)
      if citem.lessthan(cv) {
        //要执行的操作
        break
      }
    }else{
      x,v:=reflect.valueof(item),reflect.valueof(v)
      switch x.kind() {
      case reflect.int:
      case reflect.int8:
      case reflect.int16:
        /*...*/
        //x, y:=x.int(), y.int()
        /*...*/
        break
      case reflect.uint:
        /*...*/
      }
    }
  }
}

利用reflect判断item的类型:

reflect.typeof(item).implements(reflect.typeof(new(comper)).elem()),即item类型是否实现了comper接口类型。typeof(new(comper))是一个指针ptr,elem()将指针转为值。如果该函数返回值为true,则可将item和v从interface{}强制转为comper接口,调用lessthan(...);当然你也可以使用类型断言,那种方式更简单也更常用,我在这儿只是尝试一下使用反射的方法:if v,ok:=item.(comper); ok{...}

不能直接对value类型进行大小比较:

value类型不能通过">"与"<"直接比较大小,即使我们知道他是简单类型。作者还没有找到简单的方法能直接转化值为简单类型并比较,因此采用了枚举的方法。若有更简便的方法,也请告知。

如果使用实例指针实现接口:

这是一个比较难以发现的问题,涉及到golang的类型系统。也就是说,如果我们实现lessthen的方法是这样func (t*myint) lessthan (x comper) bool,那么很有可能你的断言item类型就要失败了。我们可以看一下此时item的类型:

fmt.println(reflect.typeof(t.data[0])) //main.xxx

这不是我们期待的,因为我们知道只有*t类型的方法集才是s和*s,而t类型的方法集只有s。很明显,main.xxx的方法集里不包括lessthan方法,只有*main.xxx才包括。所以正确的使用方法是,在最初赋值的时候就赋值给指针类型:

mi := myint(i)
mydata.push(&mi)

多接口分层demo

空接口其实只是一个特殊用例,我们将其推广后即可发现,我们可以定义多个接口,声明多种方法,实体实现了若干种方法便有权限调用若干函数:

例如我们可以赋予读取权限,写入权限与删除权限,来对应不同需求:

type reader interface {
  read () interface{}
}
type writer interface {
  write (writer)
}
type readwriter interface {
  reader
  writer
}
type remover interface {
  remove ()
}

type sdata struct {
  data []interface{}
}

func (t *sdata)get(i int)interface{}{
  if len(t.data) == 0{return nil}
  if reflect.typeof(t.data[0]).implements(reflect.typeof(new(reader)).elem()) == true{
    return t.data[i].(reader).read()
  }
}

func (t *sdata)modify(i int, w writer){
  // if reflect.typeof(t.data[0]).implements(reflect.typeof(new(readwriter)).elem()) == true
  if _,ok:=t.data[0].(readwriter);ok{
    t.data[i].(writer).write(w)
  }
}
//......

自定义myint类型并实现reader,writer接口:

type readint int
func (t readint) read() interface{}{
  return int(t)
}
//---------------------------------------------
type myint int
func (t myint) read() interface{}{
  return int(t)
}
func (t *myint) write(w writer){
  *t = *w.(*myint)
  return
}

func main() {
  mydata := sdata{make([]interface{}, 1)}
  var u,v myint = 5,6
  mydata.data[0] = &u
  fmt.println("myint is ", mydata.get(0))
  mydata.modify(0,&v)
  fmt.println("myint is ", mydata.get(0))

  var ru readint = 100
  readdata := sdata{make([]interface{}, 1)}
  readdata.data[0] = &ru
  fmt.println("readint is ", readdata.get(0))
  //var rv readint = 101
  readdata.modify(0,&v) //事实上,如果传递rv则编译根本不会通过。
  fmt.println("readint is ", readdata.get(0))
}

运行结果:
myint is  5
myint is  6
readint is  100
readint is  100

说明:如果因为认为上述代码传递&rv根本不会通过编译而不去作类型检查,这是不可取的。因为对于空接口interface{}而言,无所谓实体的类型,只在乎是否实现方法,因此传递&v是合情合理的。另外,因为该demo是一个简易版本,所以判断权限部分仅仅根据判断第0个元素的权限。事实上,判断权限应该在初始化时完成并将其存储在结构体变量中。

最后关于并发的问题,套用读写锁即可。过于简单不再通过demo验证。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。