philipp/uncloud
1
0
Fork 0
mirror of https://github.com/psviderski/uncloud.git synced 2026-08-26 11:03:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: refactor gRPC docker client

Browse Source
This commit is contained in:
Pasha Sviderski
2025-09-26 21:24:56 +10:00
parent bc6f2fd49d
commit e0045c7570
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/machine/docker/client.go
+6 -10
View File
@@ -16,6 +16,7 @@ import (
"github.com/docker/docker/pkg/jsonmessage"
regtypes "github.com/google/go-containerregistry/pkg/v1/types"
ocispec "github.com/opencontainers/image-spec/specs-go/v1"
"github.com/psviderski/uncloud/internal/docker"
"github.com/psviderski/uncloud/internal/machine/api/pb"
"github.com/psviderski/uncloud/pkg/api"
"google.golang.org/grpc"
@@ -205,12 +206,7 @@ type PullOptions struct {
Platform string
}
type PullImageMessage struct {
Message jsonmessage.JSONMessage
Err error
}
func (c *Client) PullImage(ctx context.Context, image string, opts PullOptions) (<-chan PullImageMessage, error) {
func (c *Client) PullImage(ctx context.Context, image string, opts PullOptions) (<-chan docker.PullImageMessage, error) {
optsBytes, err := json.Marshal(opts)
if err != nil {
return nil, fmt.Errorf("marshal options: %w", err)
@@ -221,7 +217,7 @@ func (c *Client) PullImage(ctx context.Context, image string, opts PullOptions)
return nil, err
}
ch := make(chan PullImageMessage)
ch := make(chan docker.PullImageMessage)
go func() {
defer close(ch)
@@ -232,16 +228,16 @@ func (c *Client) PullImage(ctx context.Context, image string, opts PullOptions)
return
}
if err != nil {
ch <- PullImageMessage{Err: err}
ch <- docker.PullImageMessage{Err: err}
return
}
var jm jsonmessage.JSONMessage
if err = json.Unmarshal(msg.Message, &jm); err != nil {
ch <- PullImageMessage{Err: fmt.Errorf("unmarshal JSON message: %w", err)}
ch <- docker.PullImageMessage{Err: fmt.Errorf("unmarshal JSON message: %w", err)}
return
}
ch <- PullImageMessage{Message: jm}
ch <- docker.PullImageMessage{Message: jm}
}
}()