mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-31 00:11:36 +01:00
59d7d9344b
Use the `machined` socket for `shutdown` and `poweroff` aliases. This ensures that worker nodes does not have to wait on apid to start. Signed-off-by: Noel Georgi <git@frezbo.dev>
40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package poweroff
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
"github.com/siderolabs/talos/pkg/grpc/middleware/authz"
|
|
"github.com/siderolabs/talos/pkg/machinery/client"
|
|
"github.com/siderolabs/talos/pkg/machinery/constants"
|
|
"github.com/siderolabs/talos/pkg/machinery/role"
|
|
)
|
|
|
|
// Main is the entrypoint into /sbin/poweroff.
|
|
func Main() {
|
|
ctx := context.Background()
|
|
|
|
md := metadata.Pairs()
|
|
authz.SetMetadata(md, role.MakeSet(role.Admin))
|
|
adminCtx := metadata.NewOutgoingContext(ctx, md)
|
|
|
|
client, err := client.New(adminCtx, client.WithUnixSocket(constants.MachineSocketPath), client.WithGRPCDialOptions(grpc.WithTransportCredentials(insecure.NewCredentials())))
|
|
if err != nil {
|
|
log.Fatalf(fmt.Errorf("error while creating machinery client: %w", err).Error())
|
|
}
|
|
|
|
err = client.Shutdown(adminCtx)
|
|
if err != nil {
|
|
log.Fatalf(fmt.Errorf("error while sending shutdown command: %w", err).Error())
|
|
}
|
|
}
|