Go泛型：如何处理自引用接口约束的类型链式调用问题？

问题场景 #

你有Circle和Square两个自定义类型，它们的Bigger()和Smaller()方法均返回同类型的新实例，希望通过泛型统一处理这类类型的链式调用（如.Bigger().Smaller()），但之前的泛型约束写法无法通过编译，核心问题是Go编译器无法识别类型参数的方法自引用关系。

解决方案 #

Go 1.18+支持泛型的自引用接口约束，我们可以通过两种方式实现需求：

方案一：泛型自引用接口（保留具体类型） #

定义一个泛型接口，明确要求类型参数实现返回自身的方法，这样编译器就能确认链式调用的合法性：

package main

import "fmt"

type Circle struct{ r float64 }

func NewCircle(r float64) *Circle  { return &Circle{r: r} }
func (c *Circle) Radius() float64  { return c.r }
func (c *Circle) Bigger() *Circle  { return &Circle{r: c.r + 1} }
func (c *Circle) Smaller() *Circle { return &Circle{r: c.r - 1} }

type Square struct{ s float64 }

func NewSquare(s float64) *Square   { return &Square{s: s} }
func (s *Square) Side() float64     { return s.s }
func (s1 *Square) Bigger() *Square  { return &Square{s: s1.s + 1} }
func (s1 *Square) Smaller() *Square { return &Square{s: s1.s - 1} }

// 泛型自引用接口：要求类型T实现返回自身的Bigger和Smaller方法
type Shape[T any] interface {
    Bigger() T
    Smaller() T
}

// process函数接受实现了Shape[T]的类型T，返回同类型实例
func process[T Shape[T]](s T) T {
    return s.Bigger().Smaller()
}

func main() {
    circle := process(NewCircle(3))
    fmt.Println(circle.Radius()) // 输出3

    square := process(NewSquare(6))
    fmt.Println(square.Side()) // 输出6
}

方案二：非泛型自引用接口（适合无需保留具体类型的场景） #

如果不需要返回具体类型，可直接定义包含联合类型和自引用方法的接口，使用时通过类型断言获取具体类型：

package main

import "fmt"

type Circle struct{ r float64 }

func NewCircle(r float64) *Circle  { return &Circle{r: r} }
func (c *Circle) Radius() float64  { return c.r }
func (c *Circle) Bigger() *Circle  { return &Circle{r: c.r + 1} }
func (c *Circle) Smaller() *Circle { return &Circle{r: c.r - 1} }

type Square struct{ s float64 }

func NewSquare(s float64) *Square   { return &Square{s: s} }
func (s *Square) Side() float64     { return s.s }
func (s1 *Square) Bigger() *Square  { return &Square{s: s1.s + 1} }
func (s1 *Square) Smaller() *Square { return &Square{s: s1.s - 1} }

// 非泛型自引用接口：包含具体类型联合，方法返回接口本身
type Shape interface {
    *Circle | *Square
    Bigger() Shape
    Smaller() Shape
}

func process(s Shape) Shape {
    return s.Bigger().Smaller()
}

func main() {
    // 需通过类型断言转换为具体类型
    circle := process(NewCircle(3)).(*Circle)
    fmt.Println(circle.Radius()) // 输出3

    square := process(NewSquare(6)).(*Square)
    fmt.Println(square.Side()) // 输出6
}

原代码问题分析 #

你之前的写法失败原因在于：

1. Shape[ST]接口仅约束了方法返回ST，但未约束ST本身实现Shape[ST]，编译器无法确认ST具有Smaller()方法；
2. 对类型参数直接使用类型断言不被Go允许，因为类型参数的约束是具体类型联合，无法直接断言为接口类型。

内容的提问来源于stack exchange，提问作者pcronin