ston1th/pve-go
0
0
Fork 0
Code Issues 3 Pull requests Releases Wiki Activity

added clout init image docs

Browse source
This commit is contained in:
ston1th 2020-11-01 20:09:16 +01:00
commit eb9546c9af
4 changed files with 84 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

38
snippet.go
View file

@ -3,7 +3,6 @@
package pve
import (
"bytes"
"context"
"errors"
"fmt"
@ -18,16 +17,6 @@ var (
ErrNoSnippetName = errors.New("missing snippet name")
)
type CopyBuffer struct {
*bytes.Buffer
}
func (c *CopyBuffer) Copy() *bytes.Buffer {
buf := make([]byte, c.Len())
copy(buf, c.Bytes())
return bytes.NewBuffer(buf)
}
type SnippetClient struct {
client *Client
}
@ -45,38 +34,32 @@ func (c *SnippetClient) Create(ctx context.Context, node, storage, name string,
err = ErrNoSnippetName
return
}
var b bytes.Buffer
w := multipart.NewWriter(&b)
err = w.WriteField("content", "snippets")
ub, b := newUnreadBuffer()
m := multipart.NewWriter(b)
err = m.WriteField("content", "snippets")
if err != nil {
return
}
fw, err := w.CreateFormFile("filename", name)
fw, err := m.CreateFormFile("filename", name)
if err != nil {
return
}
if _, err = fw.Write(data); err != nil {
return
}
w.Close()
cb := CopyBuffer{&b}
ct := w.FormDataContentType()
m.Close()
ub.Save()
ct := m.FormDataContentType()
var taskid string
var req *http.Request
var resp *http.Response
for i := 0; i < 3; i++ {
buf := cb.Copy()
req, err = c.client.NewRequest(ctx, "POST", fmt.Sprintf("/nodes/%s/storage/%s/upload", node, storage), buf)
req, err = c.client.NewRequest(ctx, "POST", fmt.Sprintf("/nodes/%s/storage/%s/upload", node, storage), b)
req.Header.Set("Content-Type", ct)
resp, err = c.client.Do(req, &taskid)
if err != nil {
if err != nil || resp.StatusCode != http.StatusOK {
ub.Reset(&b)
time.Sleep(time.Millisecond * 500)
// TODO log error
continue
}
if resp.StatusCode != http.StatusOK {
time.Sleep(time.Millisecond * 500)
// TODO log error
continue
}
break
@ -90,7 +73,6 @@ func SnippetVolume(storage, name string) string {
}
func (c *SnippetClient) Delete(ctx context.Context, node, storage, name string) (t *Task, err error) {
//volume := fmt.Sprintf("/%s:snippets/%s", storage, name)
volume := SnippetVolume(storage, name)
req, err := c.client.NewRequest(ctx, "DELETE", fmt.Sprintf("/nodes/%s/storage/%s/content/%s", node, storage, volume), nil)
if err != nil {