blob: 5741e58765d690ef10d7556bbcaeaf60887368a0 (
plain) (
blame)
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
|
package org.openslx.taskmanager.network;
import java.nio.charset.StandardCharsets;
import org.apache.log4j.Logger;
import org.openslx.taskmanager.api.TaskStatus;
import org.openslx.taskmanager.main.Taskmanager;
import com.google.gson.Gson;
public class RequestParser
{
private static final Logger log = Logger.getLogger( RequestParser.class );
/**
* Our very own gson instance (for serializing replies)
*/
private final Gson sendGson = new Gson();
private final Taskmanager taskManager;
public RequestParser( Taskmanager tm )
{
this.taskManager = tm;
}
/**
* Handle the given unparsed request.
*
* @param payload Packet data received from network, already converted to a string
*/
public byte[] handle( String payload )
{
String[] parts = payload.split( " *, *", 3 );
// Message format is "<message id>, <command>, <command payload/argument>"
if ( parts.length != 3 ) {
log.debug( "Could not split message" );
return null;
}
// Look at parts[1], if it's "status" it's a request for the task
// with the ID given in parts[2]
if ( parts[1].equals( "status" ) ) {
TaskStatus status = taskManager.getTaskStatus( parts[2] );
return serialize( parts[0], status );
}
// Now check if parts[1] is "release"
if ( parts[1].equals( "release" ) ) {
taskManager.releaseTask( parts[2] );
return null;
}
// Anything else in parts[0] will be treated as a fresh task invocation, so let's
// pass it on to the task manager.
TaskStatus status = taskManager.submitTask( parts[1], parts[2] );
return serialize( parts[0], status );
}
private byte[] serialize( String messageId, TaskStatus status )
{
String data;
try {
synchronized ( sendGson ) {
data = sendGson.toJson( status );
}
} catch ( Throwable e ) {
log.warn( "Could not serialize reply with TaskStatus " + status.getStatusObjectClassName() );
log.warn( e.toString() );
return null;
}
return ( messageId + ',' + data ).getBytes( StandardCharsets.UTF_8 );
}
}
|