mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-26 19:13:34 +00:00
79dc05cb66
* chore: fix landing logo shadow * feat: implement service logs command with colored output and strict ordering * revert Makefile * fix after rebase * move logs command under services with root shortcut * simplify server options for streaming logs and CLI flags * refactor ContainerLogs grpc server * update --tail flag * minor docker.proto * refactor client ServiceLogs and ContainerLogs * move ProxyMachinesContext from api to client pkg * minor refactor proto Stream * add api/logs * implement LogMerger * fix LogMerger to correctly use semaphore * increate inflish entries to 100 per stream * send heartbeats * refactor ContainerLogs to synchronise Send of entries and heartbeats to the steram * minor logmerege * remove ContainerName form ServiceLogEntryMetadata * refactor ContainerLogs into docker.Service * detect stalled container logs streams * update logmerger tests * fix comment in test * refactor LogMerger with options * uc logs: format one or multiple services * make LogMerger emit heartbeats, emit entries <= watermark, rewrite tests * update uc logs with new LogMerger * go mod tidy * fix after merge --------- Co-authored-by: Evgenii Orlov <evgenii.orlov@semrush.com>
99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"slices"
|
|
|
|
"github.com/psviderski/uncloud/internal/machine/api/pb"
|
|
"github.com/psviderski/uncloud/pkg/api"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
func (cli *Client) InspectMachine(ctx context.Context, nameOrID string) (*pb.MachineMember, error) {
|
|
machines, err := cli.ListMachines(ctx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, m := range machines {
|
|
if m.Machine.Id == nameOrID || m.Machine.Name == nameOrID {
|
|
return m, nil
|
|
}
|
|
}
|
|
|
|
return nil, api.ErrNotFound
|
|
}
|
|
|
|
// ListMachines returns a list of all machines registered in the cluster that match the filter.
|
|
func (cli *Client) ListMachines(ctx context.Context, filter *api.MachineFilter) (api.MachineMembersList, error) {
|
|
resp, err := cli.ClusterClient.ListMachines(ctx, &emptypb.Empty{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
machines := resp.Machines
|
|
|
|
if filter != nil {
|
|
var matchedMachines api.MachineMembersList
|
|
for _, m := range machines {
|
|
if MachineMatchesFilter(m, filter) {
|
|
matchedMachines = append(matchedMachines, m)
|
|
}
|
|
}
|
|
machines = matchedMachines
|
|
}
|
|
|
|
return machines, nil
|
|
}
|
|
|
|
func MachineMatchesFilter(machine *pb.MachineMember, filter *api.MachineFilter) bool {
|
|
if filter == nil {
|
|
return true
|
|
}
|
|
|
|
if filter.Available && machine.State == pb.MachineMember_DOWN {
|
|
return false
|
|
}
|
|
|
|
if len(filter.NamesOrIDs) > 0 {
|
|
if !slices.ContainsFunc(filter.NamesOrIDs, func(nameOrID string) bool {
|
|
return machine.Machine.Id == nameOrID || machine.Machine.Name == nameOrID
|
|
}) {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// UpdateMachine updates machine configuration in the cluster.
|
|
func (cli *Client) UpdateMachine(ctx context.Context, req *pb.UpdateMachineRequest) (*pb.MachineInfo, error) {
|
|
resp, err := cli.ClusterClient.UpdateMachine(ctx, req)
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok && s.Code() == codes.NotFound {
|
|
return nil, api.ErrNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
return resp.Machine, nil
|
|
}
|
|
|
|
// RenameMachine renames an existing machine in the cluster.
|
|
func (cli *Client) RenameMachine(ctx context.Context, nameOrID, newName string) (*pb.MachineInfo, error) {
|
|
// First, resolve the machine to get its ID
|
|
machine, err := cli.InspectMachine(ctx, nameOrID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Update the machine with the new name
|
|
req := &pb.UpdateMachineRequest{
|
|
MachineId: machine.Machine.Id,
|
|
Name: &newName,
|
|
}
|
|
|
|
return cli.UpdateMachine(ctx, req)
|
|
}
|