@ -14,6 +14,7 @@
@@ -14,6 +14,7 @@
* See the License for the specific language governing permissions and
* limitations under the License .
* /
package org.springframework.boot.samples.websocket.snake ;
import java.util.ArrayDeque ;
@ -23,117 +24,116 @@ import java.util.Deque;
@@ -23,117 +24,116 @@ import java.util.Deque;
import org.springframework.web.socket.TextMessage ;
import org.springframework.web.socket.WebSocketSession ;
public class Snake {
private static final int DEFAULT_LENGTH = 5 ;
private final int id ;
private final WebSocketSession session ;
private Direction direction ;
private int length = DEFAULT_LENGTH ;
private Location head ;
private final Deque < Location > tail = new ArrayDeque < Location > ( ) ;
private final String hexColor ;
public Snake ( int id , WebSocketSession session ) {
this . id = id ;
this . session = session ;
this . hexColor = SnakeUtils . getRandomHexColor ( ) ;
resetState ( ) ;
}
private void resetState ( ) {
this . direction = Direction . NONE ;
this . head = SnakeUtils . getRandomLocation ( ) ;
this . tail . clear ( ) ;
this . length = DEFAULT_LENGTH ;
}
private synchronized void kill ( ) throws Exception {
resetState ( ) ;
sendMessage ( "{'type': 'dead'}" ) ;
}
private synchronized void reward ( ) throws Exception {
length + + ;
sendMessage ( "{'type': 'kill'}" ) ;
}
protected void sendMessage ( String msg ) throws Exception {
session . sendMessage ( new TextMessage ( msg ) ) ;
}
public synchronized void update ( Collection < Snake > snakes ) throws Exception {
Location nextLocation = head . getAdjacentLocation ( direction ) ;
if ( nextLocation . x > = SnakeUtils . PLAYFIELD_WIDTH ) {
nextLocation . x = 0 ;
}
if ( nextLocation . y > = SnakeUtils . PLAYFIELD_HEIGHT ) {
nextLocation . y = 0 ;
}
if ( nextLocation . x < 0 ) {
nextLocation . x = SnakeUtils . PLAYFIELD_WIDTH ;
}
if ( nextLocation . y < 0 ) {
nextLocation . y = SnakeUtils . PLAYFIELD_HEIGHT ;
}
if ( direction ! = Direction . NONE ) {
tail . addFirst ( head ) ;
if ( tail . size ( ) > length ) {
tail . removeLast ( ) ;
}
head = nextLocation ;
}
handleCollisions ( snakes ) ;
}
private void handleCollisions ( Collection < Snake > snakes ) throws Exception {
for ( Snake snake : snakes ) {
boolean headCollision = id ! = snake . id & & snake . getHead ( ) . equals ( head ) ;
boolean tailCollision = snake . getTail ( ) . contains ( head ) ;
if ( headCollision | | tailCollision ) {
kill ( ) ;
if ( id ! = snake . id ) {
snake . reward ( ) ;
}
}
}
}
public synchronized Location getHead ( ) {
return head ;
}
public synchronized Collection < Location > getTail ( ) {
return tail ;
}
public synchronized void setDirection ( Direction direction ) {
this . direction = direction ;
}
public synchronized String getLocationsJson ( ) {
StringBuilder sb = new StringBuilder ( ) ;
sb . append ( String . format ( "{x: %d, y: %d}" ,
Integer . valueOf ( head . x ) , Integer . valueOf ( head . y ) ) ) ;
for ( Location location : tail ) {
sb . append ( ',' ) ;
sb . append ( String . format ( "{x: %d, y: %d}" ,
Integer . valueOf ( location . x ) , Integer . valueOf ( location . y ) ) ) ;
}
return String . format ( "{'id':%d,'body':[%s]}" ,
Integer . valueOf ( id ) , sb . toString ( ) ) ;
}
public int getId ( ) {
return id ;
}
public String getHexColor ( ) {
return hexColor ;
}
private static final int DEFAULT_LENGTH = 5 ;
private final int id ;
private final WebSocketSession session ;
private Direction direction ;
private int length = DEFAULT_LENGTH ;
private Location head ;
private final Deque < Location > tail = new ArrayDeque < Location > ( ) ;
private final String hexColor ;
public Snake ( int id , WebSocketSession session ) {
this . id = id ;
this . session = session ;
this . hexColor = SnakeUtils . getRandomHexColor ( ) ;
resetState ( ) ;
}
private void resetState ( ) {
this . direction = Direction . NONE ;
this . head = SnakeUtils . getRandomLocation ( ) ;
this . tail . clear ( ) ;
this . length = DEFAULT_LENGTH ;
}
private synchronized void kill ( ) throws Exception {
resetState ( ) ;
sendMessage ( "{'type': 'dead'}" ) ;
}
private synchronized void reward ( ) throws Exception {
this . length + + ;
sendMessage ( "{'type': 'kill'}" ) ;
}
protected void sendMessage ( String msg ) throws Exception {
this . session . sendMessage ( new TextMessage ( msg ) ) ;
}
public synchronized void update ( Collection < Snake > snakes ) throws Exception {
Location nextLocation = this . head . getAdjacentLocation ( this . direction ) ;
if ( nextLocation . x > = SnakeUtils . PLAYFIELD_WIDTH ) {
nextLocation . x = 0 ;
}
if ( nextLocation . y > = SnakeUtils . PLAYFIELD_HEIGHT ) {
nextLocation . y = 0 ;
}
if ( nextLocation . x < 0 ) {
nextLocation . x = SnakeUtils . PLAYFIELD_WIDTH ;
}
if ( nextLocation . y < 0 ) {
nextLocation . y = SnakeUtils . PLAYFIELD_HEIGHT ;
}
if ( this . direction ! = Direction . NONE ) {
this . tail . addFirst ( this . head ) ;
if ( this . tail . size ( ) > this . length ) {
this . tail . removeLast ( ) ;
}
this . head = nextLocation ;
}
handleCollisions ( snakes ) ;
}
private void handleCollisions ( Collection < Snake > snakes ) throws Exception {
for ( Snake snake : snakes ) {
boolean headCollision = this . id ! = snake . id
& & snake . getHead ( ) . equals ( this . head ) ;
boolean tailCollision = snake . getTail ( ) . contains ( this . head ) ;
if ( headCollision | | tailCollision ) {
kill ( ) ;
if ( this . id ! = snake . id ) {
snake . reward ( ) ;
}
}
}
}
public synchronized Location getHead ( ) {
return this . head ;
}
public synchronized Collection < Location > getTail ( ) {
return this . tail ;
}
public synchronized void setDirection ( Direction direction ) {
this . direction = direction ;
}
public synchronized String getLocationsJson ( ) {
StringBuilder sb = new StringBuilder ( ) ;
sb . append ( String . format ( "{x: %d, y: %d}" , Integer . valueOf ( this . head . x ) ,
Integer . valueOf ( this . head . y ) ) ) ;
for ( Location location : this . tail ) {
sb . append ( ',' ) ;
sb . append ( String . format ( "{x: %d, y: %d}" , Integer . valueOf ( location . x ) ,
Integer . valueOf ( location . y ) ) ) ;
}
return String . format ( "{'id':%d,'body':[%s]}" , Integer . valueOf ( this . id ) ,
sb . toString ( ) ) ;
}
public int getId ( ) {
return this . id ;
}
public String getHexColor ( ) {
return this . hexColor ;
}
}
Phillip Webb
12 years ago