Function WrapAgentConn

codersdk/workspacesdk/agentconn.go:52–63

WrapAgentConn returns an AgentConn that delegates every operation to conn and applies closeFunc exactly once when the logical session is closed. If conn is nil, any provided closeFunc is invoked immediately so logical session cleanup is not silently dropped.

(conn AgentConn, closeFunc func() error)

// If conn is nil, any provided closeFunc is invoked immediately so logical
// session cleanup is not silently dropped.
func WrapAgentConn(conn AgentConn, closeFunc func() error) AgentConn {
	if conn == nil {
		if closeFunc != nil {
			_ = closeFunc()
		}
		return nil
	}
	if closeFunc == nil {
		closeFunc = func() error { return nil }
	}
	return &wrappedAgentConn{AgentConn: conn, closeFunc: closeFunc}
}

type wrappedAgentConn struct {
	AgentConn