gg

src/main/java/kr/co/swh/lecture/opensource/discord/project/DicAnalysis.java

@@ -0,0 +1,67 @@
+package kr.co.swh.lecture.opensource.discord.project;
+
+import java.io.IOException;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
+
+import org.apache.http.HttpResponse;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.impl.client.HttpClients;
+import org.apache.http.util.EntityUtils;
+
+import com.google.gson.Gson;
+
+public class DicAnalysis {
+
+	public static void main(String[] args) {
+		// TODO Auto-generated method stub
+		Connection connection = null;
+		try {
+			Class.forName("com.mysql.jdbc.Driver");
+			connection = DriverManager.getConnection("jdbc:mysql://dev-swh.ga:3306/market","root","swhacademy!");
+		} catch (SQLException e1) {
+			// TODO Auto-generated catch block
+			e1.printStackTrace();
+		} catch (ClassNotFoundException e) {
+			// TODO Auto-generated catch block
+			e.printStackTrace();
+		}
+			
+		Gson gson = new Gson();
+		CloseableHttpClient client = HttpClients.createDefault();
+		HttpResponse response = null;
+		String responseString = "";
+		HttpGet requestGet = new HttpGet("https://raw.githubusercontent.com/park1200656/KnuSentiLex/master/data/SentiWord_info.json");
+		try {
+			response = client.execute(requestGet);
+			responseString = EntityUtils.toString(response.getEntity());
+			SentimentalDic[] value = gson.fromJson(responseString, SentimentalDic[].class);
+			for(SentimentalDic dic : value) {
+				String insertTableSQL = "INSERT INTO SENTIMENTAL_DIC"
+						+ "(word_root, polarity) VALUES"
+						+ "(?,?)";
+				PreparedStatement preparedStatementInsert = connection.prepareStatement(insertTableSQL);
+				preparedStatementInsert.setString(1, dic.getWord_root());
+				preparedStatementInsert.setInt(2, Integer.parseInt(dic.getPolarity()));
+				preparedStatementInsert.executeUpdate();
+			}
+		} catch (IOException e2) {
+			// TODO Auto-generated catch block
+			e2.printStackTrace();
+		} catch (SQLException e) {
+			// TODO Auto-generated catch block
+			e.printStackTrace();
+		} finally {
+			try {
+				connection.close();
+			} catch (SQLException e) {
+				// TODO Auto-generated catch block
+				e.printStackTrace();
+			}
+		}
+	}
+
+}

src/main/java/kr/co/swh/lecture/opensource/discord/project/SendJDA.java

@@ -1,6 +1,11 @@
 package kr.co.swh.lecture.opensource.discord.project;
 
 import java.io.IOException;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
@@ -25,7 +30,7 @@ public class SendJDA {
 	public static JDA jda;
 	public static void main(String[] args) {
 		// TODO Auto-generated method stub
-		List<SentimentalDic> dicList = initDatas();
+		List<SentimentalDic> dicList = initDatabase();
 		JDABuilder jb = new JDABuilder(AccountType.BOT);
 		jb.setAutoReconnect(true);
 		jb.setStatus(OnlineStatus.DO_NOT_DISTURB);
@@ -51,9 +56,6 @@ public class SendJDA {
 			response = client.execute(requestGet);
 			responseString = EntityUtils.toString(response.getEntity());
 			SentimentalDic[] value = gson.fromJson(responseString, SentimentalDic[].class);
-			for(SentimentalDic dic : value) {
-				System.out.println(dic.getWord());
-			}
 			dicList.addAll(Arrays.asList(value));
 		} catch (IOException e2) {
 			// TODO Auto-generated catch block
@@ -61,4 +63,42 @@ public class SendJDA {
 		}
 		return dicList;
 	}
+	
+	public static List<SentimentalDic> initDatabase(){
+		List<SentimentalDic> dicList = new ArrayList<SentimentalDic>();
+		Connection connection = null;
+		try {
+			Class.forName("com.mysql.jdbc.Driver");
+			connection = DriverManager.getConnection("jdbc:mysql://dev-swh.ga:3306/market","root","swhacademy!");
+		} catch (SQLException e1) {
+			// TODO Auto-generated catch block
+			e1.printStackTrace();
+		} catch (ClassNotFoundException e) {
+			// TODO Auto-generated catch block
+			e.printStackTrace();
+		}
+		Statement statement = null;
+		try {
+			statement = connection.createStatement();
+			ResultSet rs = statement.executeQuery("SELECT * FROM SENTIMENTAL_DIC;");
+			while(rs.next()){
+				SentimentalDic dic = new SentimentalDic();
+				dic.setWord_root(rs.getString("word_root"));
+				dic.setPolarity(String.valueOf(rs.getInt("polarity")));
+				dicList.add(dic);
+			}
+			rs.close();
+		} catch (SQLException e) {
+			// TODO Auto-generated catch block
+			e.printStackTrace();
+		} finally {
+			try {
+				connection.close();
+			} catch (SQLException e) {
+				// TODO Auto-generated catch block
+				e.printStackTrace();
+			}
+		}
+		return dicList;
+	}
 }

src/main/java/kr/co/swh/lecture/opensource/discord/project/TListener.java

@@ -2,13 +2,15 @@ package kr.co.swh.lecture.opensource.discord.project;
 
 import java.util.List;
 
-import kr.co.swh.lecture.opensource.discord.TwitterContents;
+import org.bitbucket.eunjeon.seunjeon.Analyzer;
+import org.bitbucket.eunjeon.seunjeon.LNode;
+import org.bitbucket.eunjeon.seunjeon.Morpheme;
+
 import net.dv8tion.jda.api.entities.Message;
 import net.dv8tion.jda.api.entities.TextChannel;
 import net.dv8tion.jda.api.entities.User;
 import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
 import net.dv8tion.jda.api.hooks.ListenerAdapter;
-import twitter4j.Status;
 
 public class TListener extends ListenerAdapter{
 	List<SentimentalDic> dicList;
@@ -23,16 +25,29 @@ public class TListener extends ListenerAdapter{
 		TextChannel tc = event.getTextChannel();
 		Message msg = event.getMessage();
 		if(user.isBot()) return;
-		if(msg.getContentRaw().charAt(0) == '!') {
-			String[] args = msg.getContentRaw().substring(1).split(" ");
-			if(args.length < 0) return;
-			if(args.length == 2 && args[0].equalsIgnoreCase("lol")) {
-				List<Status> status = new TwitterContents().getContents(args[1]);
-				for(Status s : status) {
-					tc.sendMessage(s.getText()).queue();
+//		// 형태소 분석
+//		for (Eojeol eojeol : Analyzer.parseEojeolJava(msg.getContentRaw())) {
+//			for (LNode node : eojeol.nodesJava()) {
+//				Morpheme m = node.morpheme();
+//				tc.sendMessage(m.surface() + " / 품사:" + m.feature().head()).queue();
+//			}
+//		}
+		int count = 0;
+		for (LNode node : Analyzer.parseJava(msg.getContentRaw())) {
+			Morpheme m = node.morpheme();
+			if(!m.feature().head().equalsIgnoreCase("jks")) {
+				for(SentimentalDic dic : dicList) {
+					if(dic.getWord_root().equalsIgnoreCase(m.surface())){
+						count += Integer.parseInt(dic.getPolarity());
+					}
 				}
 			}
-		}
+		}	
+		if(count > 0)
+			tc.sendMessage(String.format("%s가 해피하네요", user.getName())).queue();
+		else if(count < 0)
+			tc.sendMessage(String.format("%s가 우울하네요", user.getName())).queue();
+		
 	}
 
 }