mirror of
https://github.com/kubesphere/kubekey.git
synced 2025-12-26 01:22:51 +00:00
54 lines
1.6 KiB
Go
54 lines
1.6 KiB
Go
package modules
|
|
|
|
import (
|
|
"context"
|
|
|
|
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
|
|
|
|
"github.com/kubesphere/kubekey/v4/pkg/connector"
|
|
"github.com/kubesphere/kubekey/v4/pkg/variable"
|
|
)
|
|
|
|
/*
|
|
Module Setup
|
|
|
|
This module is used to set up the connection to a remote host and gather facts about it.
|
|
It performs the following operations:
|
|
1. Establishes a connection to the specified host using the appropriate connector
|
|
2. If the connector supports fact gathering (implements GatherFacts interface):
|
|
- Retrieves host information
|
|
- Merges the remote host information into the variables map
|
|
3. Returns success if all operations complete successfully
|
|
|
|
Usage:
|
|
- host: The target host to connect to
|
|
- variable: Map of variables to be used for connection and fact gathering
|
|
*/
|
|
|
|
// ModuleSetup establishes a connection to a remote host and gathers facts about it.
|
|
// It returns StdoutSuccess if successful, or an error message if any step fails.
|
|
func ModuleSetup(ctx context.Context, options ExecOptions) (string, string, error) {
|
|
// get connector
|
|
conn, err := options.getConnector(ctx)
|
|
if err != nil {
|
|
return StdoutFailed, StderrGetConnector, nil
|
|
}
|
|
defer conn.Close(ctx)
|
|
|
|
if gf, ok := conn.(connector.GatherFacts); ok {
|
|
remoteInfo, err := gf.HostInfo(ctx)
|
|
if err != nil {
|
|
return StdoutFailed, "failed to get host info", err
|
|
}
|
|
if err := options.Variable.Merge(variable.MergeRemoteVariable(remoteInfo, options.Host)); err != nil {
|
|
return StdoutFailed, "failed to merge setup variable", err
|
|
}
|
|
}
|
|
|
|
return StdoutSuccess, "", nil
|
|
}
|
|
|
|
func init() {
|
|
utilruntime.Must(RegisterModule("setup", ModuleSetup))
|
|
}
|