84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package driver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/vmware/govmomi"
|
|
"github.com/vmware/govmomi/find"
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/session"
|
|
"github.com/vmware/govmomi/vapi/rest"
|
|
"github.com/vmware/govmomi/vim25"
|
|
"github.com/vmware/govmomi/vim25/soap"
|
|
)
|
|
|
|
type Driver struct {
|
|
// context that controls the authenticated sessions used to run the VM commands
|
|
ctx context.Context
|
|
client *govmomi.Client
|
|
restClient *rest.Client
|
|
finder *find.Finder
|
|
datacenter *object.Datacenter
|
|
}
|
|
|
|
type ConnectConfig struct {
|
|
VCenterServer string
|
|
Username string
|
|
Password string
|
|
InsecureConnection bool
|
|
Datacenter string
|
|
}
|
|
|
|
func NewDriver(config *ConnectConfig) (*Driver, error) {
|
|
ctx := context.TODO()
|
|
|
|
vcenterUrl, err := url.Parse(fmt.Sprintf("https://%v/sdk", config.VCenterServer))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
credentials := url.UserPassword(config.Username, config.Password)
|
|
vcenterUrl.User = credentials
|
|
|
|
soapClient := soap.NewClient(vcenterUrl, config.InsecureConnection)
|
|
vimClient, err := vim25.NewClient(ctx, soapClient)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vimClient.RoundTripper = session.KeepAlive(vimClient.RoundTripper, 10*time.Minute)
|
|
client := &govmomi.Client{
|
|
Client: vimClient,
|
|
SessionManager: session.NewManager(vimClient),
|
|
}
|
|
|
|
err = client.SessionManager.Login(ctx, credentials)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
restClient := rest.NewClient(vimClient)
|
|
err = restClient.Login(ctx, credentials)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
finder := find.NewFinder(client.Client, false)
|
|
datacenter, err := finder.DatacenterOrDefault(ctx, config.Datacenter)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
finder.SetDatacenter(datacenter)
|
|
|
|
d := Driver{
|
|
ctx: ctx,
|
|
client: client,
|
|
restClient: restClient,
|
|
datacenter: datacenter,
|
|
finder: finder,
|
|
}
|
|
return &d, nil
|
|
}
|