mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-26 11:03:34 +00:00
chore: refactor duplication
This commit is contained in:
+34
-60
@@ -36,37 +36,8 @@ func (cli *Client) PullImage(
|
||||
return nil, err
|
||||
}
|
||||
|
||||
decoder := json.NewDecoder(respBody)
|
||||
ch := make(chan PullPushImageMessage)
|
||||
|
||||
go func() {
|
||||
defer respBody.Close()
|
||||
defer close(ch)
|
||||
|
||||
for {
|
||||
var jm jsonmessage.JSONMessage
|
||||
if err = decoder.Decode(&jm); err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
break
|
||||
}
|
||||
ch <- PullPushImageMessage{Err: fmt.Errorf("decode image pull message: %w", err)}
|
||||
break
|
||||
}
|
||||
|
||||
msg := PullPushImageMessage{Message: jm}
|
||||
if jm.Error != nil {
|
||||
msg.Err = errors.New(jm.Error.Message)
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
ch <- PullPushImageMessage{Err: ctx.Err()}
|
||||
return
|
||||
default:
|
||||
ch <- msg
|
||||
}
|
||||
}
|
||||
}()
|
||||
go processPullPushImageResp(ctx, respBody, ch)
|
||||
|
||||
return ch, nil
|
||||
}
|
||||
@@ -87,41 +58,44 @@ func (cli *Client) PushImage(
|
||||
return nil, err
|
||||
}
|
||||
|
||||
decoder := json.NewDecoder(respBody)
|
||||
ch := make(chan PullPushImageMessage)
|
||||
|
||||
go func() {
|
||||
defer respBody.Close()
|
||||
defer close(ch)
|
||||
|
||||
for {
|
||||
var jm jsonmessage.JSONMessage
|
||||
if err = decoder.Decode(&jm); err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
break
|
||||
}
|
||||
ch <- PullPushImageMessage{Err: fmt.Errorf("decode image push message: %w", err)}
|
||||
break
|
||||
}
|
||||
|
||||
msg := PullPushImageMessage{Message: jm}
|
||||
if jm.Error != nil {
|
||||
msg.Err = errors.New(jm.Error.Message)
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
ch <- PullPushImageMessage{Err: ctx.Err()}
|
||||
return
|
||||
default:
|
||||
ch <- msg
|
||||
}
|
||||
}
|
||||
}()
|
||||
go processPullPushImageResp(ctx, respBody, ch)
|
||||
|
||||
return ch, nil
|
||||
}
|
||||
|
||||
// processPullPushImageResp decodes JSON messages from the image pull/push response body and
|
||||
// sends them to the provided channel.
|
||||
func processPullPushImageResp(ctx context.Context, respBody io.ReadCloser, ch chan<- PullPushImageMessage) {
|
||||
defer respBody.Close()
|
||||
defer close(ch)
|
||||
|
||||
decoder := json.NewDecoder(respBody)
|
||||
for {
|
||||
var jm jsonmessage.JSONMessage
|
||||
if err := decoder.Decode(&jm); err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
return
|
||||
}
|
||||
ch <- PullPushImageMessage{Err: fmt.Errorf("decode image pull/push message: %w", err)}
|
||||
return
|
||||
}
|
||||
|
||||
msg := PullPushImageMessage{Message: jm}
|
||||
if jm.Error != nil {
|
||||
msg.Err = errors.New(jm.Error.Message)
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
ch <- PullPushImageMessage{Err: ctx.Err()}
|
||||
return
|
||||
default:
|
||||
ch <- msg
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// RetrieveLocalDockerRegistryAuth retrieves the authentication token for the specified image from the local Docker
|
||||
// config file. It returns the encoded authentication token if it contains any credentials, or an empty string if
|
||||
// no credentials are found.
|
||||
|
||||
Reference in New Issue
Block a user