方法功能中的Golang无效接收器类型

Question

我正在尝试创建一个简单的包来将SSH命令发送到服务器.

我有以下代码:

type Connection *ssh.Client

func Connect(addr, user, password string) (conn Connection, err error) {
    sshConfig := &ssh.ClientConfig{
        User: user,
        Auth: []ssh.AuthMethod{
            ssh.Password(password),
        },
        HostKeyCallback: ssh.HostKeyCallback(func(hostname string, remote net.Addr, key ssh.PublicKey) error { return nil }),
    }

    conn, err = ssh.Dial("tcp", addr, sshConfig)
    return
}

func (conn Connection) SendCommand() ([]byte, error) {
    session, err := (*ssh.Client)(conn).NewSession()
    // ...
}

我的问题是关于两行func (conn Connection) SendCommand() ([]byte, error)和session, err := (*ssh.Client)(conn).NewSession().

我无法弄清楚如何使用*ssh.Client我的覆盖Connection类型可用的方法.

我知道我需要进行一些转换,并且使用ssh.Client(*conn).NewSession()可以工作,但它复制的值*ssh.Client似乎不是正确的方法.

*ssh.Client在使用自定义type Connection *ssh.Client类型时,应该如何访问可用的方法？

Answer 1

您不能使用指针TypeSpec声明新类型.同时声明一个新类型专门用于删除整个方法集,因此您将不会拥有任何原始方法*ssh.Client.

你想要的是通过嵌入*ssh.Client你自己的结构类型来使用组合:

type Connection struct {
    *ssh.Client
}

func Connect(addr, user, password string) (*Connection, error) {
    sshConfig := &ssh.ClientConfig{
        User: user,
        Auth: []ssh.AuthMethod{
            ssh.Password(password),
        },
        HostKeyCallback: ssh.HostKeyCallback(func(hostname string, remote net.Addr, key ssh.PublicKey) error { return nil }),
    }

    conn, err = ssh.Dial("tcp", addr, sshConfig)
    if err != nil {
        return nil, err
    }

    return &Connection{conn}, nil
}

func (conn *Connection) SendCommand() ([]byte, error) {
    session, err := conn.NewSession()
    // ...
}