Skip to content

Commit

Permalink
Fix metrics & MergeFiles()
Browse files Browse the repository at this point in the history
  • Loading branch information
benbjohnson committed Mar 19, 2024
1 parent e90a41e commit a24fee7
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 39 deletions.
39 changes: 0 additions & 39 deletions flaps/flaps_machines.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,15 +21,6 @@ func (f *Client) sendRequestMachines(ctx context.Context, method, endpoint strin
}

func (f *Client) Launch(ctx context.Context, builder fly.LaunchMachineInput) (out *fly.Machine, err error) {
//metrics.Started(ctx, "machine_launch")
//sendUpdateMetrics := metrics.StartTiming(ctx, "machine_launch/duration")
//defer func() {
// metrics.Status(ctx, "machine_launch", err == nil)
// if err == nil {
// sendUpdateMetrics()
// }
//}()

ctx = contextWithAction(ctx, machineLaunch)

out = new(fly.Machine)
Expand All @@ -46,15 +37,6 @@ func (f *Client) Update(ctx context.Context, builder fly.LaunchMachineInput, non
headers[NonceHeader] = []string{nonce}
}

//metrics.Started(ctx, "machine_update")
//sendUpdateMetrics := metrics.StartTiming(ctx, "machine_update/duration")
//defer func() {
// metrics.Status(ctx, "machine_update", err == nil)
// if err == nil {
// sendUpdateMetrics()
// }
//}()

ctx = contextWithAction(ctx, machineUpdate)
ctx = contextWithMachineID(ctx, builder.ID)

Expand All @@ -76,11 +58,6 @@ func (f *Client) Start(ctx context.Context, machineID string, nonce string) (out

out = new(fly.MachineStartResponse)

//metrics.Started(ctx, "machine_start")
//defer func() {
// metrics.Status(ctx, "machine_start", err == nil)
//}()

ctx = contextWithAction(ctx, machineStart)
ctx = contextWithMachineID(ctx, machineID)

Expand Down Expand Up @@ -402,14 +379,6 @@ func (f *Client) GetProcesses(ctx context.Context, machineID string) (fly.Machin
}

func (f *Client) Cordon(ctx context.Context, machineID string, nonce string) (err error) {
//metrics.Started(ctx, "machine_cordon")
//sendUpdateMetrics := metrics.StartTiming(ctx, "machine_cordon/duration")
//defer func() {
// metrics.Status(ctx, "machine_cordon", err == nil)
// if err == nil {
// sendUpdateMetrics()
// }
//}()
headers := make(map[string][]string)
if nonce != "" {
headers[NonceHeader] = []string{nonce}
Expand All @@ -426,14 +395,6 @@ func (f *Client) Cordon(ctx context.Context, machineID string, nonce string) (er
}

func (f *Client) Uncordon(ctx context.Context, machineID string, nonce string) (err error) {
//metrics.Started(ctx, "machine_uncordon")
//sendUpdateMetrics := metrics.StartTiming(ctx, "machine_uncordon/duration")
//defer func() {
// metrics.Status(ctx, "machine_uncordon", err == nil)
// if err == nil {
// sendUpdateMetrics()
// }
//}()
headers := make(map[string][]string)
if nonce != "" {
headers[NonceHeader] = []string{nonce}
Expand Down
22 changes: 22 additions & 0 deletions fly.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
package fly

import "slices"

// MergeFiles merges the files parsed from the command line or fly.toml into the machine configuration.
func MergeFiles(machineConf *MachineConfig, files []*File) {
for _, f := range files {
idx := slices.IndexFunc(machineConf.Files, func(i *File) bool {
return i.GuestPath == f.GuestPath
})

switch {
case idx == -1:
machineConf.Files = append(machineConf.Files, f)
continue
case f.RawValue == nil && f.SecretName == nil:
machineConf.Files = slices.Delete(machineConf.Files, idx, idx+1)
default:
machineConf.Files = slices.Replace(machineConf.Files, idx, idx+1, f)
}
}
}

0 comments on commit a24fee7

Please sign in to comment.