├── README.md
├── pom.xml
└── src
    └── main
        ├── java
            └── com
            │   └── example
            │       └── demo
            │           ├── MyDemoApplication.java
            │           ├── config
            │               └── WebSocketConfig.java
            │           ├── controller
            │               ├── ChatController.java
            │               ├── PageController.java
            │               └── UserController.java
            │           ├── dao
            │               ├── RelationMapper.java
            │               └── UserMapper.java
            │           ├── message
            │               ├── BaseMessage.java
            │               └── ChatMessage.java
            │           ├── model
            │               ├── Relation.java
            │               └── User.java
            │           ├── security
            │               ├── AnyUserDetailsService.java
            │               ├── UserPrincipal.java
            │               └── WebSecurityConfig.java
            │           ├── service
            │               ├── RelationService.java
            │               └── UserService.java
            │           ├── utils
            │               └── Constants.java
            │           └── vo
            │               └── BaseUser.java
        └── resources
            ├── application.properties
            ├── generatorConfig.xml
            ├── mapper
                ├── RelationMapper.xml
                └── UserMapper.xml
            ├── mybatis-config.xml
            └── templates
                ├── chat.html
                ├── login.html
                └── register.html


/README.md:
--------------------------------------------------------------------------------
1 | # simple-chat-room 
2 | 


--------------------------------------------------------------------------------
/pom.xml:
--------------------------------------------------------------------------------
  1 | <?xml version="1.0" encoding="UTF-8"?>
  2 | <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  3 | 	xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  4 | 	<modelVersion>4.0.0</modelVersion>
  5 | 
  6 | 	<groupId>com.example</groupId>
  7 | 	<artifactId>myDemo</artifactId>
  8 | 	<version>0.0.1-SNAPSHOT</version>
  9 | 	<packaging>jar</packaging>
 10 | 
 11 | 	<name>myDemo</name>
 12 | 	<description>Demo project for Spring Boot</description>
 13 | 
 14 | 	<parent>
 15 | 		<groupId>org.springframework.boot</groupId>
 16 | 		<artifactId>spring-boot-starter-parent</artifactId>
 17 | 		<version>1.5.9.RELEASE</version>
 18 | 		<relativePath/> <!-- lookup parent from repository -->
 19 | 	</parent>
 20 | 
 21 | 	<properties>
 22 | 		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
 23 | 		<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
 24 | 		<java.version>1.8</java.version>
 25 | 		<thymeleaf.version>3.0.0.RELEASE</thymeleaf.version>
 26 |     	<thymeleaf-layout-dialect.version>2.0.0</thymeleaf-layout-dialect.version>
 27 | 	</properties>
 28 | 
 29 | 	<dependencies>
 30 | 		<dependency>
 31 | 			<groupId>org.springframework.boot</groupId>
 32 | 			<artifactId>spring-boot-starter-web</artifactId>
 33 | 		</dependency>
 34 | 		<dependency>
 35 | 			<groupId>org.springframework.boot</groupId>
 36 | 			<artifactId>spring-boot-starter-websocket</artifactId>
 37 | 		</dependency>
 38 | 
 39 | 		<dependency>
 40 | 			<groupId>org.springframework.boot</groupId>
 41 | 			<artifactId>spring-boot-starter-test</artifactId>
 42 | 			<scope>test</scope>
 43 | 		</dependency>
 44 | 		<dependency>
 45 | 	      <groupId>org.springframework.boot</groupId>
 46 | 	      <artifactId>spring-boot-starter-security</artifactId>
 47 | 	    </dependency>
 48 | 	
 49 | 	    <dependency>
 50 | 	      <groupId>org.springframework.boot</groupId>
 51 | 	      <artifactId>spring-boot-starter-thymeleaf</artifactId>
 52 | 	    </dependency>
 53 | 	    <dependency>
 54 |             <groupId>org.thymeleaf.extras</groupId>
 55 |             <artifactId>thymeleaf-extras-springsecurity4</artifactId>
 56 |         </dependency>
 57 | 
 58 |         <dependency>
 59 |             <groupId>mysql</groupId>
 60 |             <artifactId>mysql-connector-java</artifactId>
 61 |             </dependency>
 62 |         <dependency>
 63 | 	        <groupId>org.mybatis.spring.boot</groupId>
 64 | 	        <artifactId>mybatis-spring-boot-starter</artifactId>
 65 | 	        <version>1.1.1</version>
 66 | 	    </dependency>  
 67 | 	    <dependency>
 68 |             <groupId>tk.mybatis</groupId>
 69 |             <artifactId>mapper-spring-boot-starter</artifactId>
 70 |             <version>1.1.4</version>
 71 |         </dependency>
 72 |       	<dependency>
 73 |       		<groupId>org.projectlombok</groupId>
 74 |       		<artifactId>lombok</artifactId>
 75 |       		<version>1.16.16</version><!--$NO-MVN-MAN-VER$-->
 76 |     	</dependency>   
 77 |     	<dependency>
 78 |      		<groupId>com.alibaba</groupId>
 79 |       		<artifactId>fastjson</artifactId>
 80 |       		<version>1.2.30</version>
 81 |     	</dependency>
 82 |     	
 83 |     	<dependency>
 84 | 	      <groupId>org.webjars</groupId>
 85 | 	      <artifactId>webjars-locator</artifactId>
 86 | 	    </dependency>
 87 | 	    <dependency>
 88 | 	      <groupId>org.webjars</groupId>
 89 | 	      <artifactId>sockjs-client</artifactId>
 90 | 	      <version>1.0.2</version>
 91 | 	    </dependency>
 92 | 	    <dependency>
 93 | 	      <groupId>org.webjars</groupId>
 94 | 	      <artifactId>stomp-websocket</artifactId>
 95 | 	      <version>2.3.3</version>
 96 | 	    </dependency>
 97 | 	    <dependency>
 98 | 	      <groupId>org.webjars</groupId>
 99 | 	      <artifactId>bootstrap</artifactId>
100 | 	      <version>3.3.7</version>
101 | 	    </dependency>
102 | 	    <dependency>
103 | 	      <groupId>org.webjars</groupId>
104 | 	      <artifactId>jquery</artifactId>
105 | 	      <version>3.1.0</version>
106 | 	    </dependency>
107 | 	</dependencies>
108 | 
109 | 	<build>
110 | 		<plugins>
111 | 			<plugin>
112 | 				<groupId>org.springframework.boot</groupId>
113 | 				<artifactId>spring-boot-maven-plugin</artifactId>
114 | 			</plugin>
115 | 			<plugin>  
116 |                 <groupId>org.mybatis.generator</groupId>  
117 |                 <artifactId>mybatis-generator-maven-plugin</artifactId>  
118 |                 <version>1.3.2</version>  
119 |                 <configuration>  
120 |                     <verbose>true</verbose>  
121 |                     <overwrite>true</overwrite>  
122 |                 </configuration>  
123 |             </plugin>  
124 | 		</plugins>
125 | 	</build>
126 | 
127 | 
128 | </project>
129 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/MyDemoApplication.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo;
 2 | 
 3 | import org.mybatis.spring.annotation.MapperScan;
 4 | import org.springframework.boot.SpringApplication;
 5 | import org.springframework.boot.autoconfigure.SpringBootApplication;
 6 | 
 7 | @SpringBootApplication
 8 | @MapperScan("com.example.demo.dao")
 9 | public class MyDemoApplication {
10 | 
11 | 	public static void main(String[] args) {
12 | 		SpringApplication.run(MyDemoApplication.class, args);
13 | 	}
14 | 
15 | }
16 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/config/WebSocketConfig.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.config;
 2 | 
 3 | import org.springframework.context.annotation.Configuration;
 4 | import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 5 | import org.springframework.web.socket.CloseStatus;
 6 | import org.springframework.web.socket.WebSocketHandler;
 7 | import org.springframework.web.socket.WebSocketSession;
 8 | import org.springframework.web.socket.config.annotation.AbstractWebSocketMessageBrokerConfigurer;
 9 | import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
10 | import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
11 | import org.springframework.web.socket.config.annotation.WebSocketTransportRegistration;
12 | import org.springframework.web.socket.handler.WebSocketHandlerDecorator;
13 | import org.springframework.web.socket.handler.WebSocketHandlerDecoratorFactory;
14 | 
15 | @Configuration
16 | @EnableWebSocketMessageBroker
17 | public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer{
18 | 
19 | 	@Override
20 | 	public void registerStompEndpoints(StompEndpointRegistry registry) {
21 | 		//添加服务端点，接收客户端的连接
22 | 		registry.addEndpoint("/any-socket")
23 | 				//开启SockJS支持
24 | 				.withSockJS();
25 | 	}
26 | 	
27 | 	@Override
28 | 	public void configureMessageBroker(MessageBrokerRegistry registry) {
29 | 		//服务端接收地址的前缀
30 | 		registry.setApplicationDestinationPrefixes("/app");
31 | 		//客户端订阅地址的前缀信息
32 | 		registry.enableSimpleBroker("/topic","/user");
33 | 	}
34 | 	
35 | 	@Override
36 | 	public void configureWebSocketTransport(WebSocketTransportRegistration registration) {
37 | 		registration.addDecoratorFactory(new WebSocketHandlerDecoratorFactory() {
38 | 			
39 | 			@Override
40 | 			public WebSocketHandler decorate(WebSocketHandler handler) {
41 | 				return new WebSocketHandlerDecorator(handler){
42 | 					/* (non-Javadoc)
43 | 					 * 客户端与服务端建立连接后的服务端的操作
44 | 					 */
45 | 					@Override
46 | 					public void afterConnectionEstablished(WebSocketSession session) throws Exception {
47 | 						String username = session.getPrincipal().getName();
48 | 						System.out.println("online:"+username);
49 | 						super.afterConnectionEstablished(session);
50 | 					}
51 | 					
52 | 					/* (non-Javadoc)
53 | 					 * 客户端与服务端断开连接后服务端的操作
54 | 					 */
55 | 					@Override
56 | 					public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus)
57 | 							throws Exception {
58 | 						String username = session.getPrincipal().getName();
59 | 						System.out.println("offline:"+username);
60 | 						super.afterConnectionClosed(session, closeStatus);
61 | 					}
62 | 				};
63 | 			}
64 | 		});
65 | 	}
66 | 
67 | }
68 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/controller/ChatController.java:
--------------------------------------------------------------------------------
  1 | package com.example.demo.controller;
  2 | 
  3 | import java.security.Principal;
  4 | import java.text.SimpleDateFormat;
  5 | import java.util.Date;
  6 | 
  7 | 
  8 | import org.springframework.beans.factory.annotation.Autowired;
  9 | import org.springframework.messaging.handler.annotation.MessageMapping;
 10 | import org.springframework.messaging.handler.annotation.SendTo;
 11 | import org.springframework.messaging.simp.SimpMessagingTemplate;
 12 | import org.springframework.stereotype.Controller;
 13 | 
 14 | import com.alibaba.fastjson.JSON;
 15 | import com.example.demo.message.BaseMessage;
 16 | import com.example.demo.message.ChatMessage;
 17 | import com.example.demo.utils.Constants;
 18 | 
 19 | @Controller
 20 | public class ChatController {
 21 | 	
 22 | 	private SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
 23 | 	
 24 | 	@Autowired
 25 | 	private SimpMessagingTemplate template;
 26 | 	
 27 | 	/**
 28 | 	 * 群聊
 29 | 	 * @param principal 当前用户
 30 | 	 * @param message 接收到的客户端的消息
 31 | 	 * @return 包装后的消息
 32 | 	 */
 33 | 	@MessageMapping("/all")//接收发送到  /{服务端接收地址的前缀}/all 地址的消息
 34 | 	@SendTo("/topic/all")//将return的结果发送到  /topic/all 地址
 35 | 	public String all(Principal principal,String message){
 36 | 		BaseMessage baseMessage = new BaseMessage();
 37 | 		baseMessage.setType(Constants.TO_ALL);
 38 | 		baseMessage.setContent(message);
 39 | 		baseMessage.setSender(principal.getName());
 40 | 		baseMessage.setSendTime(format.format(new Date()));
 41 | 		return JSON.toJSONString(baseMessage);
 42 | 	}
 43 | 	
 44 | 	/**
 45 | 	 * 群发图片
 46 | 	 * @param principal
 47 | 	 * @param message
 48 | 	 * @return
 49 | 	 */
 50 | 	@MessageMapping("/pic/all")//接收发送到  /{服务端接收地址的前缀}/all 地址的消息
 51 | 	@SendTo("/topic/pic/all")//将return的结果发送到  /topic/all 地址
 52 | 	public String picAll(Principal principal,String message){
 53 | 		return message;
 54 | 	}
 55 | 	
 56 | 	@MessageMapping("/pic/chat")//接收发送到  /{服务端接收地址的前缀}/chat 地址的消息
 57 | 	public void pic2One(Principal principal,String message){
 58 | 		ChatMessage chatMessage = JSON.parseObject(message,ChatMessage.class);
 59 | 		BaseMessage baseMessage = new BaseMessage();
 60 | 		baseMessage.setType(Constants.TO_ONE);
 61 | 		baseMessage.setSender(principal.getName());
 62 | 		System.out.println(principal.getName());
 63 | 		baseMessage.setContent(chatMessage.getContent());
 64 | 		baseMessage.setSendTime(format.format(new Date()));
 65 | 		
 66 | 		//转发包装后的消息至/{user}/topic/chat地址
 67 | 		template.convertAndSendToUser(chatMessage.getReceiver(), "/topic/pic/chat", JSON.toJSONString(baseMessage));
 68 | 	}
 69 | 	
 70 | 	/**
 71 | 	 * 群发文件
 72 | 	 * @param principal
 73 | 	 * @param message
 74 | 	 * @return
 75 | 	 */
 76 | 	@MessageMapping("/file/all")//接收发送到  /{服务端接收地址的前缀}/all 地址的消息
 77 | 	@SendTo("/topic/file/all")//将return的结果发送到  /topic/all 地址
 78 | 	public String fileAll(Principal principal,String message){
 79 | 		return message;
 80 | 	}
 81 | 	
 82 | 	@MessageMapping("/file/chat")//接收发送到  /{服务端接收地址的前缀}/chat 地址的消息
 83 | 	public void file2One(Principal principal,String message){
 84 | 		ChatMessage chatMessage = JSON.parseObject(message,ChatMessage.class);
 85 | 		BaseMessage baseMessage = new BaseMessage();
 86 | 		baseMessage.setType(Constants.TO_ONE);
 87 | 		baseMessage.setSender(principal.getName());
 88 | 		System.out.println(principal.getName());
 89 | 		baseMessage.setContent(chatMessage.getContent());
 90 | 		baseMessage.setSendTime(format.format(new Date()));
 91 | 		
 92 | 		//转发包装后的消息至/{user}/topic/chat地址
 93 | 		template.convertAndSendToUser(chatMessage.getReceiver(), "/topic/file/chat", JSON.toJSONString(baseMessage));
 94 | 	}
 95 | 	
 96 | 	/**
 97 | 	 * 私聊
 98 | 	 * @param principal
 99 | 	 * @param message
100 | 	 */
101 | 	@MessageMapping("/chat")//接收发送到  /{服务端接收地址的前缀}/chat 地址的消息
102 | 	public void chat(Principal principal,String message){
103 | 		ChatMessage chatMessage = JSON.parseObject(message,ChatMessage.class);
104 | 		BaseMessage baseMessage = new BaseMessage();
105 | 		baseMessage.setType(Constants.TO_ONE);
106 | 		baseMessage.setSender(principal.getName());
107 | 		System.out.println(principal.getName());
108 | 		baseMessage.setContent(chatMessage.getContent());
109 | 		baseMessage.setSendTime(format.format(new Date()));
110 | 		
111 | 		//转发包装后的消息至/{user}/topic/chat地址
112 | 		template.convertAndSendToUser(chatMessage.getReceiver(), "/topic/chat", JSON.toJSONString(baseMessage));
113 | 	}
114 | }
115 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/controller/PageController.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.controller;
 2 | 
 3 | import java.util.List;
 4 | 
 5 | import org.springframework.beans.factory.annotation.Autowired;
 6 | import org.springframework.security.core.annotation.AuthenticationPrincipal;
 7 | import org.springframework.stereotype.Controller;
 8 | import org.springframework.ui.Model;
 9 | import org.springframework.web.bind.annotation.GetMapping;
10 | 
11 | import com.example.demo.security.UserPrincipal;
12 | import com.example.demo.service.RelationService;
13 | import com.example.demo.vo.BaseUser;
14 | 
15 | @Controller
16 | public class PageController {
17 | 	
18 | 	@Autowired
19 | 	private RelationService relationService;
20 | 	
21 | 	/**
22 | 	 * @return 跳转url地址别名或者物理跳转地址
23 | 	 */
24 | 	@GetMapping(value="/login")
25 | 	public String login(){
26 | 		return "login";
27 | 	}
28 | 	@GetMapping(value="/register")
29 | 	public String register(){
30 | 		return "register";
31 | 	}
32 | 	
33 | 	@GetMapping(value="/Pic")
34 | 	public String Pic(){
35 | 		return "Pic";
36 | 	}
37 | 	
38 | 	/**
39 | 	 * @param userPrincipal @AuthenticationPrincipal可获取当前登录用户
40 | 	 * @param model 用于向页面传递数据
41 | 	 * @return
42 | 	 */
43 | 	@GetMapping(value="/chat")
44 | 	public String chat(@AuthenticationPrincipal UserPrincipal userPrincipal, Model model){
45 | 		model.addAttribute("user", userPrincipal);
46 | 		List<BaseUser> friends = relationService.getFriendsByUsername(userPrincipal.getUsername());
47 | 		model.addAttribute("friends", friends);
48 | 		return "chat";
49 | 	}
50 | }
51 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/controller/UserController.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.controller;
 2 | 
 3 | import java.text.SimpleDateFormat;
 4 | import java.util.Date;
 5 | 
 6 | import javax.servlet.http.HttpServletRequest;
 7 | 
 8 | import org.springframework.beans.factory.annotation.Autowired;
 9 | import org.springframework.security.core.annotation.AuthenticationPrincipal;
10 | import org.springframework.stereotype.Controller;
11 | import org.springframework.web.bind.annotation.PostMapping;
12 | import org.springframework.web.bind.annotation.RequestParam;
13 | import org.springframework.web.bind.annotation.ResponseBody;
14 | 
15 | import com.example.demo.model.User;
16 | import com.example.demo.security.UserPrincipal;
17 | import com.example.demo.service.RelationService;
18 | import com.example.demo.service.UserService;
19 | 
20 | @Controller
21 | public class UserController {
22 | 	@Autowired
23 | 	UserService userService;
24 | 	@Autowired
25 | 	RelationService relationService;
26 | 	
27 | 	@PostMapping(value="/register")
28 | 	@ResponseBody
29 | 	public boolean register(HttpServletRequest request){
30 | 		String username = request.getParameter("username");
31 | 		String password = request.getParameter("password");
32 | 		String nickname = request.getParameter("nickname");
33 | 		User user = new User(username,password,nickname);
34 | 		if(null != userService.selectByUsername(user.getUsername())){
35 | 			return false;
36 | 		}
37 | 		SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
38 | 		user.setRegisterData(dateFormat.format(new Date()));
39 | 		return userService.save(user);
40 | 	}
41 | 	
42 | 	@PostMapping(value="/addfriend")
43 | 	@ResponseBody
44 | 	public String addfriend(@AuthenticationPrincipal UserPrincipal userPrincipal,@RequestParam String friend){
45 | 		return relationService.addFriend(userPrincipal.getUsername(),friend);
46 | 	}	
47 | 	
48 | }
49 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/dao/RelationMapper.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.dao;
 2 | 
 3 | import java.util.List;
 4 | 
 5 | import org.apache.ibatis.annotations.Param;
 6 | 
 7 | import com.example.demo.model.User;
 8 | 
 9 | import tk.mybatis.mapper.common.Mapper;
10 | import tk.mybatis.mapper.common.MySqlMapper;
11 | 
12 | public interface RelationMapper extends Mapper<User>, MySqlMapper<User>{
13 | 	
14 | 	List<String> getFriendsByUsername(String username);
15 | 	
16 | 	void addFriend(@Param("username")String username,@Param("friend")String friend);
17 | 	
18 | }
19 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/dao/UserMapper.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.dao;
 2 | 
 3 | import com.example.demo.model.User;
 4 | 
 5 | import tk.mybatis.mapper.common.Mapper;
 6 | import tk.mybatis.mapper.common.MySqlMapper;
 7 | 
 8 | public interface UserMapper extends Mapper<User>, MySqlMapper<User>{
 9 | 
10 |     int insert(User record);
11 | 
12 | 	User selectByUsername(String username);
13 | }


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/message/BaseMessage.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.message;
 2 | 
 3 | import lombok.Data;
 4 | 
 5 | @Data
 6 | public class BaseMessage {
 7 | 	public String getSender() {
 8 | 		return sender;
 9 | 	}
10 | 	public void setSender(String sender) {
11 | 		this.sender = sender;
12 | 	}
13 | 	public String getSendTime() {
14 | 		return sendTime;
15 | 	}
16 | 	public void setSendTime(String sendTime) {
17 | 		this.sendTime = sendTime;
18 | 	}
19 | 	public String getContent() {
20 | 		return content;
21 | 	}
22 | 	public void setContent(String content) {
23 | 		this.content = content;
24 | 	}
25 | 	private String sender;
26 | 	private String sendTime;
27 | 	private String content;
28 | 	private String type;
29 | 	public String getType() {
30 | 		return type;
31 | 	}
32 | 	public void setType(String type) {
33 | 		this.type = type;
34 | 	}
35 | }
36 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/message/ChatMessage.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.message;
 2 | 
 3 | import lombok.Data;
 4 | 
 5 | @Data
 6 | public class ChatMessage{
 7 | 	private String receiver;
 8 | 	private String content;
 9 | 	private String type;
10 | 
11 | 	public String getContent() {
12 | 		return content;
13 | 	}
14 | 
15 | 	public void setContent(String content) {
16 | 		this.content = content;
17 | 	}
18 | 
19 | 	public String getType() {
20 | 		return type;
21 | 	}
22 | 
23 | 	public void setType(String type) {
24 | 		this.type = type;
25 | 	}
26 | 
27 | 	public String getReceiver() {
28 | 		return receiver;
29 | 	}
30 | 
31 | 	public void setReceiver(String receiver) {
32 | 		this.receiver = receiver;
33 | 	}
34 | 	
35 | //	public static void main(String[] args) {
36 | //		ChatMessage chatMessage = new ChatMessage();
37 | //		chatMessage.setContent("chat");
38 | //		chatMessage.setSender("qwe");
39 | //		chatMessage.setReceiver("zxc");
40 | //		chatMessage.setSendTime("2018-1-31");
41 | //		System.out.println(JSON.toJSON(chatMessage));
42 | //	}
43 | }
44 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/model/Relation.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.model;
 2 | 
 3 | import javax.persistence.Column;
 4 | import javax.persistence.Table;
 5 | 
 6 | @Table(name="liusong.relation")
 7 | public class Relation {
 8 | 	@Column(name="id")
 9 | 	private Integer id;
10 | 	@Column(name="username")
11 | 	private String username;
12 | 	@Column(name="friend")
13 | 	private String friend;
14 | 	public Integer getId() {
15 | 		return id;
16 | 	}
17 | 	public void setId(Integer id) {
18 | 		this.id = id;
19 | 	}
20 | 	public String getUsername() {
21 | 		return username;
22 | 	}
23 | 	public void setUsername(String username) {
24 | 		this.username = username;
25 | 	}
26 | 	public String getFriend() {
27 | 		return friend;
28 | 	}
29 | 	public void setFriend(String friend) {
30 | 		this.friend = friend;
31 | 	}
32 | }
33 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/model/User.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.model;
 2 | 
 3 | import javax.persistence.Column;
 4 | import javax.persistence.Table;
 5 | 
 6 | @Table(name = "liusong.user")
 7 | public class User {
 8 | 	@Column(name="id")
 9 |     private Integer id;
10 | 	@Column(name="nick_name")
11 |     private String nickName;
12 | 	@Column(name="password")
13 |     private String password;
14 | 	@Column(name="register_data")
15 |     private String registerData;
16 | 	@Column(name="username")
17 |     private String username;
18 | 
19 | 	public User() {
20 | 		super();
21 | 	}
22 | 
23 | 	public User(String nickName, String password, String username) {
24 | 		super();
25 | 		this.nickName = nickName;
26 | 		this.password = password;
27 | 		this.username = username;
28 | 	}
29 | 
30 | 	public Integer getId() {
31 |         return id;
32 |     }
33 | 
34 |     public void setId(Integer id) {
35 |         this.id = id;
36 |     }
37 | 
38 |     public String getNickName() {
39 |         return nickName;
40 |     }
41 | 
42 |     public void setNickName(String nickName) {
43 |         this.nickName = nickName == null ? null : nickName.trim();
44 |     }
45 | 
46 |     public String getPassword() {
47 |         return password;
48 |     }
49 | 
50 |     public void setPassword(String password) {
51 |         this.password = password == null ? null : password.trim();
52 |     }
53 | 
54 |     public String getRegisterData() {
55 |         return registerData;
56 |     }
57 | 
58 |     public void setRegisterData(String registerData) {
59 |         this.registerData = registerData == null ? null : registerData.trim();
60 |     }
61 | 
62 |     public String getUsername() {
63 |         return username;
64 |     }
65 | 
66 |     public void setUsername(String username) {
67 |         this.username = username == null ? null : username.trim();
68 |     }
69 | 
70 | }


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/security/AnyUserDetailsService.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.security;
 2 | 
 3 | import java.util.ArrayList;
 4 | import java.util.List;
 5 | 
 6 | import org.springframework.beans.factory.annotation.Autowired;
 7 | import org.springframework.security.core.GrantedAuthority;
 8 | import org.springframework.security.core.authority.SimpleGrantedAuthority;
 9 | import org.springframework.security.core.userdetails.UserDetails;
10 | import org.springframework.security.core.userdetails.UserDetailsService;
11 | import org.springframework.security.core.userdetails.UsernameNotFoundException;
12 | import org.springframework.stereotype.Service;
13 | 
14 | import com.example.demo.model.User;
15 | import com.example.demo.service.UserService;
16 | 
17 | @Service
18 | public class AnyUserDetailsService implements UserDetailsService{
19 | 	
20 | 	private final static String ROLE_TAG = "ROLE_USER";
21 | 
22 | 	@Autowired
23 | 	private UserService userService;
24 | 	
25 | 	@Override
26 | 	public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
27 | 		User user = userService.selectByUsername(username);
28 | 		if(user==null){
29 | 			throw new UsernameNotFoundException("用户不存在！");
30 | 		}
31 | 		List<GrantedAuthority> authorities = new ArrayList<>();
32 | 		authorities.add(new SimpleGrantedAuthority(ROLE_TAG));
33 | 		UserPrincipal userPrincipal = new UserPrincipal(username,user.getPassword(),authorities);
34 | 		userPrincipal.setNickName(user.getNickName());
35 | 		return userPrincipal;
36 | 	}
37 | 
38 | }
39 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/security/UserPrincipal.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.security;
 2 | 
 3 | import java.util.Collection;
 4 | 
 5 | import org.springframework.security.core.GrantedAuthority;
 6 | import org.springframework.security.core.userdetails.User;
 7 | 
 8 | public class UserPrincipal extends User{
 9 | 
10 | 	private static final long serialVersionUID = -3962444933610725267L;
11 | 	
12 | 	private String nickName;
13 | 
14 | 	public String getNickName() {
15 | 		return nickName;
16 | 	}
17 | 
18 | 	public void setNickName(String nickName) {
19 | 		this.nickName = nickName;
20 | 	}
21 | 
22 | 	public UserPrincipal(String username, String password, Collection<? extends GrantedAuthority> authorities) {
23 | 		super(username, password, authorities);
24 | 	}
25 | 
26 | 	public UserPrincipal(String username, String password, boolean enabled, boolean accountNonExpired,
27 | 			boolean credentialsNonExpired, boolean accountNonLocked,
28 | 			Collection<? extends GrantedAuthority> authorities) {
29 | 		super(username, password, enabled, accountNonExpired, credentialsNonExpired, accountNonLocked, authorities);
30 | 	}
31 | 
32 | }
33 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/security/WebSecurityConfig.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.security;
 2 | 
 3 | import org.springframework.beans.factory.annotation.Autowired;
 4 | import org.springframework.context.annotation.Configuration;
 5 | import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;
 6 | import org.springframework.security.config.annotation.web.builders.HttpSecurity;
 7 | import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity;
 8 | import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;
 9 | 
10 | @Configuration
11 | @EnableWebSecurity
12 | public class WebSecurityConfig extends WebSecurityConfigurerAdapter{
13 | 	
14 | 	@Autowired
15 | 	private AnyUserDetailsService anyUserDetailsService;
16 | 	
17 | 	@Autowired
18 | 	public void configureGlobal(AuthenticationManagerBuilder auth) throws Exception {
19 | 		auth.userDetailsService(this.anyUserDetailsService);
20 | 	}
21 | 	
22 | 	@Override
23 | 	protected void configure(HttpSecurity http) throws Exception {
24 | 		http.authorizeRequests().antMatchers("/","/register","/api/common/**").permitAll()
25 | 			.anyRequest().authenticated().and().formLogin().loginPage("/login").defaultSuccessUrl("/chat", true)
26 | 			.permitAll().and().logout().permitAll().and().csrf().disable();;
27 | 	}
28 | }
29 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/service/RelationService.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.service;
 2 | 
 3 | import java.util.ArrayList;
 4 | import java.util.List;
 5 | 
 6 | import org.springframework.beans.factory.annotation.Autowired;
 7 | import org.springframework.stereotype.Service;
 8 | 
 9 | import com.example.demo.dao.RelationMapper;
10 | import com.example.demo.model.User;
11 | import com.example.demo.vo.BaseUser;
12 | 
13 | @Service
14 | public class RelationService {
15 | 	
16 | 	@Autowired
17 | 	private UserService userService;
18 | 	
19 | 	@Autowired
20 | 	private RelationMapper relationMapper;
21 | 	
22 | 	public List<BaseUser> getFriendsByUsername(String username){
23 | 		List<String> list = relationMapper.getFriendsByUsername(username);
24 | 		List<BaseUser> friends = new ArrayList<>();
25 | 		for(String l:list){
26 | 			BaseUser bu = new BaseUser(l);
27 | 			User user = userService.selectByUsername(l);
28 | 			bu.setNickName(user.getNickName());
29 | 			friends.add(bu);
30 | 		}
31 | 		return friends;
32 | 	}
33 | 	
34 | 	public String addFriend(String username, String friend){
35 | 		if(null == userService.selectByUsername(friend)){
36 | 			System.out.println("用户不存在");
37 | 			return "用户不存在！";
38 | 		}
39 | 		if(username.equals(friend)){
40 | 			System.out.println("不能添加本人!");
41 | 			return "不能添加本人！";
42 | 		}
43 | 		if(relationMapper.getFriendsByUsername(username)!=null&&relationMapper.getFriendsByUsername(username).contains(friend)){
44 | 			System.out.println("已是好友");
45 | 			return "已是好友！";
46 | 		}
47 | 		relationMapper.addFriend(username,friend);
48 | 		relationMapper.addFriend(friend, username);
49 | 		return "添加成功！";
50 | 	}
51 | 	
52 | }
53 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/service/UserService.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.service;
 2 | 
 3 | import org.springframework.beans.factory.annotation.Autowired;
 4 | import org.springframework.stereotype.Service;
 5 | 
 6 | import com.example.demo.dao.UserMapper;
 7 | import com.example.demo.model.User;
 8 | 
 9 | @Service
10 | public class UserService {
11 | 	@Autowired
12 | 	UserMapper userMapper;
13 | 	
14 | 	public boolean save(User user){
15 | 		userMapper.insert(user);
16 | 		return true;
17 | 	}
18 | 	
19 | 	public User selectByUsername(String username){
20 | 		return userMapper.selectByUsername(username);
21 | 	}
22 | 	
23 | 	
24 | }
25 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/utils/Constants.java:
--------------------------------------------------------------------------------
1 | package com.example.demo.utils;
2 | 
3 | public class Constants {
4 | 	public final static String TO_ALL = "to_all";
5 | 	public final static String TO_ONE = "to_one";
6 | }
7 | 


--------------------------------------------------------------------------------
/src/main/java/com/example/demo/vo/BaseUser.java:
--------------------------------------------------------------------------------
 1 | package com.example.demo.vo;
 2 | 
 3 | public class BaseUser {
 4 | 	private String name;
 5 | 	private String nickName;
 6 | 
 7 | 	public String getNickName() {
 8 | 		return nickName;
 9 | 	}
10 | 
11 | 	public void setNickName(String nickName) {
12 | 		this.nickName = nickName;
13 | 	}
14 | 
15 | 	public BaseUser(String name) {
16 | 		this.name = name;
17 | 	}
18 | 
19 | 	public String getName() {
20 | 		return name;
21 | 	}
22 | 
23 | 	public void setName(String name) {
24 | 		this.name = name;
25 | 	}
26 | 	
27 | }
28 | 


--------------------------------------------------------------------------------
/src/main/resources/application.properties:
--------------------------------------------------------------------------------
 1 | spring.datasource.driver-class-name=com.mysql.jdbc.Driver
 2 | spring.datasource.url=jdbc:mysql://localhost:3306/liusong?useUnicode=true&characterEncoding=utf-8
 3 | spring.datasource.username=root
 4 | spring.datasource.password=123456
 5 | logging.level.org.springframework.security=info
 6 | spring.thymeleaf.cache=false
 7 | 
 8 | mybatis.type-aliases-package=com.example.demo.model
 9 | mybatis.mapper-locations=classpath:mapper/*.xml
10 | mybatis.config=classpath:mybatis-config.xml


--------------------------------------------------------------------------------
/src/main/resources/generatorConfig.xml:
--------------------------------------------------------------------------------
 1 | <?xml version="1.0" encoding="UTF-8"?>
 2 | <!DOCTYPE generatorConfiguration    
 3 |         PUBLIC "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"    
 4 |         "http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd"> 
 5 |         
 6 | <generatorConfiguration>
 7 | 
 8 | 	<!-- 指定数据连接驱动jar地址 -->  
 9 | 	<classPathEntry location="C:/Users/liusong1.GEMDALE/.m2/repository/mysql/mysql-connector-java/6.0.5/mysql-connector-java-6.0.5.jar"/>
10 | 	
11 | 	<!-- 一个数据库一个context -->  
12 | 	<context id="MysqlTables" targetRuntime="MyBatis3">
13 | 		<!-- 注释 -->  
14 | 		<commentGenerator>
15 | 			<property name="suppressAllComments" value="true"/><!-- 是否取消注释 -->  
16 | 		</commentGenerator>
17 | 		
18 | 		<!-- jdbc连接 -->  
19 | 		<jdbcConnection driverClass="com.mysql.jdbc.Driver" connectionURL="jdbc:mysql://localhost:3306/liusong" userId="root" password="123456"/>
20 | 		
21 | 		<!-- 类型转换 --> 
22 | 		<javaTypeResolver>
23 | 			<!-- 是否使用bigDecimal， false可自动转化以下类型（Long, Integer, Short, etc.） --> 
24 | 			<property name="forceBigDecimals" value="false"/>
25 | 		</javaTypeResolver>
26 | 		
27 | 		<!-- 生成实体类地址 -->  
28 | 		<javaModelGenerator targetPackage="com.example.demo.model" targetProject="myDemo">
29 | 			<!-- 是否在当前路径下新加一层schema,eg：fase路径com.oop.eksp.user.model， true:com.oop.eksp.user.model.[schemaName] -->
30 | 			<property name="enableSubPackages" value="true"/>
31 | 			<!-- 是否针对string类型的字段在set的时候进行trim调用 -->  
32 | 			<property name="trimStrings" value="true"/>
33 | 		</javaModelGenerator>
34 | 		
35 | 		<!-- 生成mapxml文件 -->  
36 | 		<sqlMapGenerator targetPackage="com.example.demo.dao" targetProject="myDemo">
37 | 			<!-- 是否在当前路径下新加一层schema,eg：fase路径com.oop.eksp.user.model， true:com.oop.eksp.user.model.[schemaName] -->
38 | 			<property name="enableSubPackages" value="false"/>
39 | 		</sqlMapGenerator>
40 | 		
41 | 		<!-- 生成mapxml对应client，也就是接口dao -->
42 | 		<javaClientGenerator targetPackage="com.example.demo.dao" type="XMLMAPPER" targetProject="myDemo"></javaClientGenerator>
43 | 		
44 | 		<!-- 配置表信息 --> 
45 | 		<table tableName="user" domainObjectName="User" schema=""></table>
46 | 		<table tableName="relation" domainObjectName="Relation" schema=""></table>
47 | 	</context>
48 | </generatorConfiguration>


--------------------------------------------------------------------------------
/src/main/resources/mapper/RelationMapper.xml:
--------------------------------------------------------------------------------
 1 | <?xml version="1.0" encoding="UTF-8"?>
 2 | <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 3 | <mapper namespace="com.example.demo.dao.RelationMapper">
 4 |   <resultMap id="BaseResultMap" type="com.example.demo.model.Relation">
 5 |     <id column="id" jdbcType="INTEGER" property="id" />
 6 |     <result column="username" jdbcType="VARCHAR" property="username" />
 7 |     <result column="friend" jdbcType="VARCHAR" property="friend" />
 8 |   </resultMap>
 9 |   
10 |   <sql id="Base_Column_List">
11 |     id, username, friend
12 |   </sql>
13 | 	
14 |   <select id="getFriendsByUsername" parameterType="java.lang.String" resultType="java.lang.String">
15 |     SELECT friend FROM relation WHERE username = #{username,jdbcType=VARCHAR}
16 |   </select>
17 |   
18 |   <insert id="addFriend">
19 |   	INSERT INTO relation (username,friend) VALUES (#{username},#{friend})
20 |   </insert>
21 |   
22 | </mapper>


--------------------------------------------------------------------------------
/src/main/resources/mapper/UserMapper.xml:
--------------------------------------------------------------------------------
 1 | <?xml version="1.0" encoding="UTF-8"?>
 2 | <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 3 | <mapper namespace="com.example.demo.dao.UserMapper">
 4 | 
 5 |   <select id="selectByUsername" parameterType="java.lang.String" resultType="com.example.demo.model.User">
 6 |     select *
 7 |     from user
 8 |     where username = #{username,jdbcType=VARCHAR}
 9 |   </select>
10 | 
11 |   <insert id="insert" parameterType="com.example.demo.model.User">
12 |     insert into user (nick_name, password, 
13 |       register_data, username)
14 |     values (#{nickName,jdbcType=VARCHAR}, #{password,jdbcType=VARCHAR}, 
15 |       #{registerData,jdbcType=VARCHAR}, #{username,jdbcType=VARCHAR})
16 |   </insert>
17 | 
18 | </mapper>


--------------------------------------------------------------------------------
/src/main/resources/mybatis-config.xml:
--------------------------------------------------------------------------------
 1 | <?xml version="1.0" encoding="UTF-8" ?>
 2 | <!DOCTYPE configuration 
 3 | 	PUBLIC "-//mybatis.org//DTD Config 3.0//EN" "http://mybatis.org/dtd/mybatis-3-config.dtd">
 4 | <configuration>
 5 | 	<settings>
 6 | 		<setting name="cacheEnabled" 				value="true" />
 7 | 		<setting name="defaultExecutorType" 		value="REUSE" />
 8 | 		<setting name="defaultStatementTimeout" 	value="25000" />
 9 | 		<setting name="lazyLoadingEnabled" 			value="true" />
10 | 		<setting name="mapUnderscoreToCamelCase" 	value="true" />
11 | 		<setting name="multipleResultSetsEnabled" 	value="true" />
12 | 		<setting name="useColumnLabel" 				value="true" />
13 | 	</settings>
14 | </configuration>


--------------------------------------------------------------------------------
/src/main/resources/templates/chat.html:
--------------------------------------------------------------------------------
  1 | <!DOCTYPE html>
  2 | <html xmlns="http://www.w3.org/1999/xhtml" xmlns:th="http://www.thymeleaf.org">
  3 | <head>
  4 | <meta charset="UTF-8"/>
  5 | <title>chat</title>
  6 |     <link href="/webjars/bootstrap/css/bootstrap.min.css" rel="stylesheet">
  7 |     <script src="/webjars/jquery/jquery.min.js"></script>
  8 |     <script src="/webjars/sockjs-client/sockjs.min.js"></script>
  9 |     <script src="/webjars/stomp-websocket/stomp.min.js"></script>
 10 | </head>
 11 | <body>
 12 | 	<div class="row" style="margin: 0 0 10px 20px">
 13 | 		<form class="form-inline" onSubmit="return false;">
 14 |   			<div class="form-group">
 15 |    	 			<input type="text" class="form-control" id="friendInput" placeholder="好友帐号"/>
 16 |   			</div>
 17 |   			<button class="btn btn-default" id="addBtn">添加好友</button>
 18 |   			<button class="btn btn-default disabled" id="groupBtn">群聊模式</button>
 19 | 		</form>
 20 | 	</div>
 21 | 	
 22 | 	<div class="row center-block">
 23 | 
 24 |         <div style="margin: 0 10px; width: 350px; display: inline-block">
 25 | 
 26 |             <div class="panel panel-danger" >
 27 |                 <div class="panel-heading">
 28 |                     
 29 |                     <div style="display:inline-block; width: 200px; height: 64px;">
 30 |                         <p style="margin: 0; padding: 6px 0; font-size: 20px" th:text="${user.username}"></p>
 31 |                         <p style="margin-bottom: 0; overflow: hidden; height: 16px">Online</p>
 32 |                     </div>
 33 |                 </div>
 34 |                 <div class="panel-body" style="height: 390px; overflow-y: scroll; overflow: auto">
 35 | 
 36 |                     <button class="friend" th:each="friend : ${friends}">
 37 |                         <div class="media-body" style="text-align: left">
 38 |                             <h4 class="media-heading f-name" th:text="${friend.name}"></h4>
 39 |                             <p class="hidden f-name" th:text="${friend.name}">anoy</p>
 40 |                         </div>
 41 |                     </button>
 42 | 
 43 |                 </div>
 44 |             </div>
 45 |         </div>
 46 | 
 47 |         <div class="col-md-8">
 48 |             <div class="panel panel-primary">
 49 |                 <div class="panel-heading">
 50 |                     <h3 class="panel-title"><span class="glyphicon glyphicon-console"></span> [当前] - [<span id="target">所有人</span>]</h3>
 51 |                 </div>
 52 |                 <div class="panel-body" style="height: 380px; overflow-y: scroll; overflow: auto" id="messageWindow">
 53 | 
 54 | 
 55 |                 </div>
 56 | 
 57 |             </div>
 58 | 			
 59 | 			<input id="image"style="width: 0.1px; height: 0.1px; opacity: 0; overflow: hidden; position: absolute" type="file" accept="image/*"/>	
 60 | 			<label for="image">选择图片</label>
 61 | 			
 62 | 			<input id="file"style="width: 0.1px; height: 0.1px; opacity: 0; overflow: hidden; position: absolute" type="file"/>	
 63 | 			<label for="file">选择文件</label>
 64 |             <div class="row">
 65 |                 <div class="col-xs-9 col-md-10">
 66 |                     <input type="text" class="form-control" placeholder="Message" id="message"/>
 67 |                 </div>
 68 |                 <div class="col-xs-3 col-md-2">
 69 |                 	
 70 |                     <button class="btn btn-primary btn-block" id="send"><span class="glyphicon glyphicon-send"></span> 发送</button>
 71 |                 </div>
 72 |             </div>
 73 |         </div>
 74 | 
 75 |     </div>
 76 | 	
 77 | 	
 78 | 	
 79 | 	<script type="text/javascript">
 80 | 	var target = "to_all";
 81 | 
 82 |     var stompClient = null;
 83 | 
 84 |     function connect() {
 85 |         var socket = new SockJS('/any-socket');
 86 |         stompClient = Stomp.over(socket);
 87 |         stompClient.connect({}, function (frame) {
 88 |             console.log('Connected: ' + frame);
 89 |             stompClient.subscribe('/topic/all', function (message) {
 90 |                 showMessage(JSON.parse(message.body));
 91 |             });
 92 |             stompClient.subscribe('/user/topic/chat', function (message) {
 93 |                 showMessage(JSON.parse(message.body));
 94 |             });
 95 |             stompClient.subscribe('/topic/pic/all', function (message) {
 96 |                 showImg(message);
 97 |             });
 98 |             stompClient.subscribe('/topic/pic/chat', function (message) {
 99 |                 showImg(message);
100 |             });
101 |             stompClient.subscribe('/topic/file/all', function (message) {
102 |                 showfile(message);
103 |             });
104 |             stompClient.subscribe('/topic/file/chat', function (message) {
105 |                 showfile(message);
106 |             });
107 |         });
108 |     }
109 | 
110 |     function showMessage(message) {
111 |         $("#messageWindow").append("<div class='message-div row'>"+
112 |                                    " <div style='margin-left: 55px'>"+
113 |                                    "<p class='user-username'>" + message.sender + "说：</p> <p class='user-time'>"+message.sendTime+"</p>"+
114 |                                    "<div class='message-box-left'>" + message.content + "</div></div></div><br></br>"
115 |         );
116 |     }
117 |     
118 |     var imgsrc = "";
119 |     
120 |     function showImg(message) {
121 |     	var img = document.createElement("img");
122 |     	var body = message.body;
123 |     	if(body.indexOf(":") != -1){
124 |     		imgsrc += body.substring(0,60000);
125 |     	}else{
126 |     		imgsrc += body;
127 |     		img.src="data:"+imgsrc;
128 |         	img.width=100;
129 |         	document.getElementById("messageWindow").appendChild(img);
130 |         	$("#messageWindow").append("<br/>");
131 |         	imgsrc = "";
132 |     	}
133 |     	
134 |     }
135 |     
136 |    	var filesrc = "";
137 |     function showfile(message) {
138 |     	var body = message.body;
139 |     	if(body.indexOf("_") != -1){
140 |     		filesrc += body.substring(0,60000);
141 |     	}else{
142 |     		filesrc += body;
143 |     		var blob = new Blob([filesrc]);
144 |     		var a = document.createElement("a");
145 |     		a.href = window.URL.createObjectURL(blob);
146 |     		a.download = "a.txt";
147 |     		a.textContent = "Download";
148 | 
149 |     		document.getElementById("messageWindow").appendChild(a);
150 |         	/* $("#messageWindow").append("<div>"+filesrc+"</div><br/>"); */
151 |         	filesrc = "";
152 |     	}
153 |     	
154 |     } 
155 | 
156 |     $(function () {
157 |         connect();
158 | 
159 |         $("#send").click(function () {
160 |         	var imageElement = document.getElementById("image");  
161 |             var imageList = imageElement.files;  
162 |             var image=imageList[0];
163 |             
164 |             var fileElement = document.getElementById("file");  
165 |             var fileList = fileElement.files;  
166 |             var file=fileList[0]; 
167 |  
168 |             if (target == "to_all"){
169 |             	if($("#message").val()){
170 |             		stompClient.send("/app/all", {}, $("#message").val());
171 |             	}
172 |                 
173 |             	if(file){
174 |                     var reader = new FileReader();  
175 |                     //以二进制形式读取文件  
176 |                     reader.readAsText(file,"UTF-8");  
177 |                     //文件读取完毕后该函数响应  
178 |                     reader.onload = function loaded(evt) {
179 |                         var blob = evt.target.result;
180 |                         var str = blob;
181 |                         /* var sr = blob.split(":");
182 |                         var str = sr[1];
183 |                         var lock = false; */
184 |                         while(str.length>60001){
185 |                         	var sub=str.substring(0,60000)+"_";
186 |                         	str = str.substring(60000);
187 |                         	stompClient.send("/app/file/all", {}, sub); 
188 |                         }
189 |                         stompClient.send("/app/file/all", {}, str); 
190 |                         
191 |                     } 
192 |                 }  
193 |             	
194 |                 if(image){
195 |                     var reader = new FileReader();  
196 |                     //以二进制形式读取文件  
197 |                     reader.readAsDataURL(image);  
198 |                     //文件读取完毕后该函数响应  
199 |                     reader.onload = function loaded(evt) {
200 |                         var blob = evt.target.result;
201 |                         var sr = blob.split(":");
202 |                         var str = sr[1];
203 |                         while(str.length>60001){
204 |                         	var sub=str.substring(0,60000)+":";
205 |                         	str = str.substring(60000);
206 |                         	stompClient.send("/app/pic/all", {},/* img */sub); 
207 |                         }
208 |                         stompClient.send("/app/pic/all", {},/* img */str); 
209 |                         
210 |                     } 
211 |                 }  
212 | 
213 |             }else{
214 |                 var content = "{'type':'to_one','content':'" + $("#message").val() + "','receiver':'"+target+"'}";
215 |                 stompClient.send("/app/chat", {}, content);
216 |             }
217 |             $("#message").val("");
218 |         });
219 | 
220 |         $(".friend").click(function () {
221 |             target = $(this).find("p.f-name").text();
222 |             var username = $(this).find("h4.f-name").text();
223 |             $("#target").text(username);
224 |             $("#groupBtn").attr("class","btn btn-default");
225 |         });
226 |         
227 |         $("#groupBtn").click(function(){
228 |         	$(this).attr("class","btn btn-default disabled");
229 |         	target = "to_all";
230 |         	$("#target").text("所有人");
231 |         });
232 | 		$("#addBtn").click(function(){
233 | 	    	var friend = $("#friendInput").val();
234 | 	    	if(friend.length == 0){
235 | 	    		return;
236 | 	    	}
237 | 	    	$.post("/addfriend",
238 | 	    			{"friend" : friend},
239 | 	    			function(data){
240 | 	    				alert(data);
241 | 	    				window.location.reload();
242 | 	    			}
243 | 	    	);
244 | 	    });
245 |     });
246 | 	</script>
247 | </body>
248 | </html>


--------------------------------------------------------------------------------
/src/main/resources/templates/login.html:
--------------------------------------------------------------------------------
 1 | <!DOCTYPE html>
 2 | <html>
 3 | <head>
 4 | <meta charset="UTF-8"/>
 5 | <title>Login</title>
 6 | </head>
 7 | <body>
 8 | 	<div>
 9 | 		<form action="/login" method="post">
10 | 			<input type="text" class="form-control" name="username" id="username" placeholder="用户名"/>
11 | 			<input type="password" class="form-control" name="password" id="password" placeholder="密码"/>
12 | 			<button type="submit" class="btn btn-primary btn-block">登录</button>
13 | 			<a href="/register">创建账号</a>
14 | 		</form>
15 | 	</div>
16 | </body>
17 | </html>


--------------------------------------------------------------------------------
/src/main/resources/templates/register.html:
--------------------------------------------------------------------------------
 1 | <!DOCTYPE html>
 2 | <html>
 3 | <head>
 4 | <meta charset="UTF-8"/>
 5 | <title>register</title>
 6 | <script src="http://ajax.aspnetcdn.com/ajax/jQuery/jquery-1.8.0.js"></script>
 7 | </head>
 8 | <body>
 9 | 	<form onsubmit="return false">
10 |     	<input type="text" class="form-control" name="username" id="username" placeholder="用户名"/>
11 |         <input type="password" class="form-control" name="password" id="password" placeholder="密码"/>
12 |      	<input type="text" class="form-control" name="nickname" id="nickname" placeholder="昵称"/>
13 |         <div class="form-group">
14 |         	<div class="input-group col-md-6 col-md-offset-3">
15 |             	<button class="btn btn-success btn-block" id="Complete">完成</button>
16 |        		</div>
17 |       	</div>
18 |       	<div class="form-group">
19 |         	<div class="input-group col-md-6 col-md-offset-3" style="text-align: center">
20 |           		<a href="/login"><span class="glyphicon glyphicon-triangle-left"></span>返回登录页面</a>
21 |           	</div>
22 |      	</div>
23 | 	</form>
24 | 	
25 | 	<script>
26 |     /* 注册点击事件 */
27 |     $("#Complete").click(function () {
28 |         $.post("/register",
29 |                {
30 |                    "username": $("#username").val(),
31 |                    "password": $("#password").val(),
32 |                    "nickname": $("#nickname").val(),
33 |                },
34 |                 function (data) {
35 |                    if (data){
36 |                        alert("注册成功");
37 |                    }
38 |                }
39 |         );
40 | 
41 |     });
42 | 
43 | </script>
44 | </body>
45 | </html>


--------------------------------------------------------------------------------