summaryrefslogtreecommitdiff
path: root/src/main/java/com/orbekk/protobuf/Rpc.java
blob: c208380511b4cde901e5e083efeaffc9d14b6805 (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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/**
 * Copyright 2012 Kjetil Ørbekk <kjetil.orbekk@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.orbekk.protobuf;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;

import com.google.protobuf.RpcCallback;
import com.google.protobuf.RpcController;

public class Rpc implements RpcController {
    private String errorText = "";
    private boolean hasFailed;
    private boolean canceled;
    private List<RpcCallback<Object>> cancelNotificationListeners = null;
    
    public Rpc() {
    }
    
    public Rpc(Rpc other) {
        copyFrom(other);
    }

    public void copyFrom(Rpc other) {
        errorText = other.errorText;
        hasFailed = other.hasFailed;
        canceled = other.canceled;
        if (other.cancelNotificationListeners != null) {
            for (RpcCallback<Object> listener :
                    other.cancelNotificationListeners) {
                notifyOnCancel(listener);
            }
        }
    }
    
    public void writeTo(Data.Response.Builder response) {
        response.setHasFailed(hasFailed);
        response.setCanceled(canceled);
        response.setErrorText(errorText);
    }
    
    public void readFrom(Data.Response response) {
        hasFailed = response.getHasFailed();
        canceled = response.getCanceled();
        errorText = response.getErrorText();
    }
    
    @Override
    public String errorText() {
        return errorText;
    }

    public boolean isOk() {
        return !hasFailed && !canceled;
    }
    
    @Override
    public boolean failed() {
        return hasFailed;
    }

    @Override
    public boolean isCanceled() {
        return canceled;
    }

    @Override
    public void notifyOnCancel(RpcCallback<Object> listener) {
        if (cancelNotificationListeners == null) {
            cancelNotificationListeners =
                    new ArrayList<RpcCallback<Object>>();
        }
        cancelNotificationListeners.add(listener);
    }

    @Override
    public void reset() {
        copyFrom(new Rpc());
    }

    @Override
    public void setFailed(String message) {
        hasFailed = true;
        errorText = message;
    }
    
    public void cancel() {
        canceled = true;
        if (cancelNotificationListeners != null) {
            for (RpcCallback<Object> listener :
                    cancelNotificationListeners) {
                listener.run(null);
            }
        }
    }

    @Override
    public void startCancel() {
    }
}