aboutsummaryrefslogtreecommitdiff
path: root/weed/storage/needle/async_request.go
blob: ea02c55c55ab5d7fb259bb1f35738f39f5d31a43 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package needle

type AsyncRequest struct {
	N              *Needle
	IsWriteRequest bool
	ActualSize     int64
	offset         uint64
	size           uint64
	doneChan       chan interface{}
	isUnchanged    bool
	err            error
}

func NewAsyncRequest(n *Needle, isWriteRequest bool) *AsyncRequest {
	return &AsyncRequest{
		offset:         0,
		size:           0,
		ActualSize:     0,
		doneChan:       make(chan interface{}),
		N:              n,
		isUnchanged:    false,
		IsWriteRequest: isWriteRequest,
		err:            nil,
	}
}

func (r *AsyncRequest) WaitComplete() (uint64, uint64, bool, error) {
	<-r.doneChan
	return r.offset, r.size, r.isUnchanged, r.err
}

func (r *AsyncRequest) Complete(offset uint64, size uint64, isUnchanged bool, err error) {
	r.offset = offset
	r.size = size
	r.isUnchanged = isUnchanged
	r.err = err
	close(r.doneChan)
}

func (r *AsyncRequest) UpdateResult(offset uint64, size uint64, isUnchanged bool, err error) {
	r.offset = offset
	r.size = size
	r.isUnchanged = isUnchanged
	r.err = err
}

func (r *AsyncRequest) Submit() {
	close(r.doneChan)
}

func (r *AsyncRequest) IsSucceed() bool {
	return r.err == nil
}