WebsocketDemoActivity.java
5.18 KB
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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package com.huaheng.mobilewms.websocket;
import android.annotation.SuppressLint;
import android.content.Intent;
import android.os.Bundle;
import android.util.Log;
import android.view.View;
import android.widget.Button;
import android.widget.EditText;
import android.widget.TextView;
import com.huaheng.mobilewms.R;
import com.huaheng.mobilewms.WMSApplication;
import com.huaheng.mobilewms.activity.model.CommonActivity;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import io.reactivex.schedulers.Schedulers;
import rx.android.schedulers.AndroidSchedulers;
import ua.naiksoftware.stomp.Stomp;
import ua.naiksoftware.stomp.StompClient;
import ua.naiksoftware.stomp.dto.StompCommand;
import ua.naiksoftware.stomp.dto.StompHeader;
import ua.naiksoftware.stomp.dto.StompMessage;
public class WebsocketDemoActivity extends CommonActivity implements View.OnClickListener {
private Button btConnect;
private Button btDisconnect;
private Button btSend;
private Button btService;
private EditText edWebsockUri;
private StompClient mStompClient;
private EditText edTextSend;
private TextView textEcho;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_websocket_demo2);
btConnect = findViewById(R.id.btConnect);
btDisconnect = findViewById(R.id.btDisconnect);
btSend = findViewById(R.id.btSend);
btService = findViewById(R.id.btService);
edWebsockUri = findViewById(R.id.websockUri);
edTextSend = findViewById(R.id.textSend);
textEcho = findViewById(R.id.textEcho);
btConnect.setOnClickListener(this);
btDisconnect.setOnClickListener(this);
btSend.setOnClickListener(this);
btService.setOnClickListener(this);
}
@Override
public void onClick(View v) {
switch (v.getId()) {
case R.id.btConnect:
createStompClient(edWebsockUri.getText().toString());
break;
case R.id.btDisconnect:
if(mStompClient != null && mStompClient.isConnected()) {
mStompClient.disconnect();
}
break;
case R.id.btSend:
sendMessage(WebsocketConstants.UP_STRING_MESSAGE_DEST, edTextSend.getText().toString());
break;
case R.id.btService:
Intent intent = new Intent(this, WebsocketService.class);
startService(intent);
break;
default:
break;
}
}
private void sendMessage(String uri, String msg){
mStompClient.send(uri, msg).subscribe();
}
/**
* 回复收到消息
* @param msg
*/
private void confirmMessage(StompMessage msg){
for(StompHeader header: msg.getStompHeaders()){
if(header.getKey().equals(WebsocketConstants.HEADER_MEG_ID)){
String msg_id = header.getValue();
List<StompHeader> headers = new ArrayList<>();
// headers.add(new StompHeader(WebsocketConstants.HEADER_MEG_ID, msg_id));
headers.add(new StompHeader(StompHeader.DESTINATION, WebsocketConstants.UP_CONFIRM_MESSAGE_DEST));
mStompClient.send(new StompMessage(StompCommand.SEND, headers, null)).subscribe();
}
}
}
@SuppressLint("CheckResult")
private void createStompClient(String uri) {
HashMap<String, String> headers = new HashMap<>();
headers.put("Cookie", WMSApplication.getOkhttpCookie());
mStompClient = Stomp.over(Stomp.ConnectionProvider.OKHTTP, uri, headers);
mStompClient.connect();
mStompClient.withClientHeartbeat(WebsocketConstants.HEART_BEAT_CLIENT).withServerHeartbeat(WebsocketConstants.HEART_BEAT_SERVER);
mStompClient.lifecycle().subscribe(lifecycleEvent -> {
switch (lifecycleEvent.getType()) {
case OPENED:
Log.d("TAG", "Stomp connection opened");
mStompClient.topic(WebsocketConstants.SUBSCRIBE_TOPIC_MESSAGE)
.subscribeOn(Schedulers.io())
// .observeOn(AndroidSchedulers.mainThread())
.subscribe(topicMessage -> {
// showShort(topicMessage.getPayload());
});
mStompClient.topic(WebsocketConstants.SUBSCRIBE_USER_TOPIC_MESSAGE)
.subscribeOn(Schedulers.io())
// .observeOn(AndroidSchedulers.mainThread())
.subscribe(topicMessage -> {
// showShort(topicMessage.getPayload());
confirmMessage(topicMessage);
});
break;
case ERROR:
Log.e("TAG", "Error", lifecycleEvent.getException());
// startActivity(LoginActivity.class);
break;
case CLOSED:
Log.d("TAG", "Stomp connection closed");
break;
}
});
}
}