91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
/*
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package pvecloud
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
|
|
cloudprovider "k8s.io/cloud-provider"
|
|
|
|
pve "git.giftfish.de/ston1th/pve-go"
|
|
v1 "k8s.io/api/core/v1"
|
|
)
|
|
|
|
type instancesV2 struct {
|
|
client *pve.Client
|
|
}
|
|
|
|
func newInstancesV2(client *pve.Client) *instancesV2 {
|
|
return &instancesV2{client}
|
|
}
|
|
|
|
func (i *instancesV2) InstanceExists(ctx context.Context, node *v1.Node) (exists bool, err error) {
|
|
server, err := i.getServer(ctx, node)
|
|
if err != nil && err != cloudprovider.InstanceNotFound {
|
|
return
|
|
}
|
|
err = nil
|
|
exists = server != nil
|
|
return
|
|
}
|
|
|
|
func (i *instancesV2) InstanceShutdown(ctx context.Context, node *v1.Node) (isOff bool, err error) {
|
|
server, err := i.getServer(ctx, node)
|
|
if err != nil && err != cloudprovider.InstanceNotFound {
|
|
return
|
|
}
|
|
err = nil
|
|
isOff = server != nil && server.Status == pve.ServerStatusStopped
|
|
return
|
|
}
|
|
|
|
func (i *instancesV2) getServer(ctx context.Context, node *v1.Node) (server *pve.Server, err error) {
|
|
if node.Spec.ProviderID != "" {
|
|
server, err = getServerByProviderID(ctx, i.client, node.Spec.ProviderID)
|
|
return
|
|
}
|
|
server, err = getServerByName(ctx, i.client, node.Name)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return getServerByProviderID(ctx, i.client, server.K8sID())
|
|
}
|
|
|
|
func (i *instancesV2) nodeAddresses(ctx context.Context, srv *pve.Server) []v1.NodeAddress {
|
|
var internalIP string
|
|
ipc := srv.IPConfig.Get(0)
|
|
if ipc != nil {
|
|
ip, _, err := net.ParseCIDR(ipc.IPv4CIDR)
|
|
if err == nil {
|
|
internalIP = ip.String()
|
|
}
|
|
}
|
|
addrs := []v1.NodeAddress{
|
|
{Type: v1.NodeHostName, Address: srv.Name},
|
|
{Type: v1.NodeInternalIP, Address: internalIP},
|
|
}
|
|
return addrs
|
|
}
|
|
|
|
func (i *instancesV2) InstanceMetadata(ctx context.Context, node *v1.Node) (*cloudprovider.InstanceMetadata, error) {
|
|
server, err := i.getServer(ctx, node)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &cloudprovider.InstanceMetadata{
|
|
ProviderID: server.K8sID(),
|
|
InstanceType: InstanceType,
|
|
NodeAddresses: i.nodeAddresses(ctx, server),
|
|
}, nil
|
|
}
|