From c3cb74a7050c3e89eb64d6e565a127df61a2c141 Mon Sep 17 00:00:00 2001
From: shoopea <shoopea@siteop.biz>
Date: Sun, 5 May 2019 19:29:28 +0800
Subject: [PATCH] test

---
 main.go    |  8 +++--
 sql.go     | 94 ++++++++++++++++++++++++++++++++++++++++++------------
 workers.go | 29 +++++++++++++++++
 3 files changed, 108 insertions(+), 23 deletions(-)
 create mode 100644 workers.go

diff --git a/main.go b/main.go
index 6d817ec..85eae4b 100644
--- a/main.go
+++ b/main.go
@@ -51,8 +51,9 @@ var (
 
 	db *sql.DB
 
-	cfg          Config
-	MQCWMsgQueue chan ChatWarsMessage
+	cfg           Config
+	MQCWMsgQueue  chan ChatWarsMessage
+	SQLMsgIdQueue chan int64
 )
 
 func PrintText(m *tb.Message) {
@@ -103,6 +104,9 @@ func main() {
 	for w := 1; w <= 6; w++ {
 		go SQLCWMsgWorker(w, MQCWMsgQueue)
 	}
+	for w := 1; w <= 6; w++ {
+		go SQLIdentifyMsgWorker(w, SQLMsgIdQueue)
+	}
 
 	fmt.Println("Started !")
 
diff --git a/sql.go b/sql.go
index d0f067d..7053dac 100644
--- a/sql.go
+++ b/sql.go
@@ -62,6 +62,18 @@ func initDB() {
 		log.Fatal(err)
 	}
 
+	_, err = db.Exec(`CREATE TABLE code_obj_sub_type (
+							 id 			SMALLINT(5) UNSIGNED NOT NULL
+							,intl_id 	VARCHAR(32) NOT NULL
+							,name 		VARCHAR(80) NOT NULL
+							,obj_type_id SMALLINT UNSIGNED NOT NULL
+							,PRIMARY KEY (id)
+							,FOREIGN KEY (obj_type_id) REFERENCES code_obj_type(id) ON DELETE CASCADE
+						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
+	if err != nil {
+		log.Fatal(err)
+	}
+
 	_, err = db.Exec(`CREATE TABLE obj (
 							 id				BIGINT UNSIGNED NOT NULL AUTO_INCREMENT
 							,obj_type_id	SMALLINT UNSIGNED NOT NULL
@@ -112,6 +124,23 @@ func initDB() {
 		log.Fatal(err)
 	}
 
+	_, err = db.Exec(`CREATE TABLE obj_msg_report (
+							 obj_id         BIGINT UNSIGNED NOT NULL
+							,war_date       TIMESTAMP NOT NULL
+							,attack         SMALLINT UNSIGNED NOT NULL
+							,defense        SMALLINT UNSIGNED NOT NULL
+							,gold           SMALLINT UNSIGNED NOT NULL
+							,stock			SMALLINT UNSIGNED NOT NULL
+							,exp			SMALLINT UNSIGNED NOT NULL
+							,stamina		BOOLEAN NOT NULL
+							,crit			BOOLEAN NOT NULL
+							,FOREIGN KEY (obj_id) REFERENCES obj(id) ON DELETE CASCADE
+							,UNIQUE KEY (msg_id, chat_id, sender_user_id)
+						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
+	if err != nil {
+		log.Fatal(err)
+	}
+
 	_, err = db.Exec(`INSERT INTO code_obj_type (id, intl_id, name)
 						  VALUES (1, "user", "User")
 						        ,(2, "guild", "Guild")
@@ -120,40 +149,63 @@ func initDB() {
 		log.Fatal(err)
 	}
 
-	log.Println("Database set up")
-}
-
-func SQLCWMsgWorker(id int, msgs <-chan ChatWarsMessage) {
-	for m := range msgs {
-		err := putMsg(m)
-		if err != nil {
-			logOnError(err, "SQLCWMsgWorker["+strconv.Itoa(id)+"] : Inserting message.")
-		}
-	}
-
-	log.Printf("SQLCWMsgWorker[" + strconv.Itoa(id) + "] : Closing.")
-
-}
-
-func putMsg(m ChatWarsMessage) error {
-	res, err := db.Exec(`INSERT INTO obj (obj_type_id)
-						      VALUES (3);`)
+	_, err = db.Exec(`INSERT INTO code_obj_sub_type (id, intl_id, name, obj_type_id)
+						  VALUES (1, "unprocessed", "Unprocessed", 3)
+						        ,(2, "unknown", "Unknown", 3)
+						        ,(3, "report", "War report", 3)
+						        ,(4, "g_report", "Guild War report", 3)
+						        ,(5, "quest", "Quest result", 3)
+						        ,(6, "fight", "Fight result", 3)
+						        ,(7, "hero", "Hero summary", 3)
+						        ,(8, "me", "Hero short summary", 3)
+						        ,(9, "inv", "Inventory", 3);`)
 	if err != nil {
 		log.Fatal(err)
 	}
+
+	log.Println("Database set up")
+}
+
+func putUnprocessedMsg(m ChatWarsMessage) (int64, error) {
+	res, err := db.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
+						          VALUES (3          , 1);`)
+	if err != nil {
+		return nil, err
+	}
+
 	objId, err := res.LastInsertId()
+	if err != nil {
+		return nil, err
+	}
 
 	stmt, err := db.Prepare(`INSERT INTO obj_msg	(obj_id, msg_id, chat_id, sender_user_id, date            , text)
 							 VALUES 				(?,      ?,      ?,       ?,              FROM_UNIXTIME(?), ?);`)
 	if err != nil {
-		return err
+		return nil, err
 	}
 	defer stmt.Close()
 
 	_, err = stmt.Exec(objId, m.ID64, m.ChatID64, m.SenderUserID64, m.Date, m.Text)
 	if err != nil {
-		return err
+		return nil, err
 	}
 
-	return nil
+	return objId, nil
+}
+
+func getMsg(objId int64) (ChatWarsMessage, error) {
+	var m ChatWarsMessage
+
+	stmt, err := db.Prepare(`SELECT om.msg_id, om.chat_id, om.sender_user_id, UNIX_TIMESTAMP(om.date), om.text FROM obj_msg om WHERE om.id = ?`)
+	if err != nil {
+		return nil, err
+	}
+	defer stmt.Close()
+
+	err = stmtOut.QueryRow(objId).Scan(&m.ID64, &m.ChatID64, &m.SenderUserID64, &m.Date, &m.Text)
+	if err != nil {
+		return nil, err
+	}
+
+	return m, nil
 }
diff --git a/workers.go b/workers.go
new file mode 100644
index 0000000..1630fe8
--- /dev/null
+++ b/workers.go
@@ -0,0 +1,29 @@
+package main
+
+import (
+	"log"
+	"strconv"
+)
+
+func SQLCWMsgWorker(id int, msgs <-chan ChatWarsMessage) {
+	log.Printf("SQLCWMsgWorker[" + strconv.Itoa(id) + "] : Starting.")
+	for m := range msgs {
+		objId, err := putUnprocessedMsg(m)
+		if err != nil {
+			logOnError(err, "SQLCWMsgWorker["+strconv.Itoa(id)+"] : Inserting message.")
+		} else {
+
+		}
+	}
+
+	log.Printf("SQLCWMsgWorker[" + strconv.Itoa(id) + "] : Closing.")
+
+}
+
+func SQLIdentifyMsgWorker(id int, objIds <-chan int64) {
+	log.Printf("SQLIdentifyMsgWorker[" + strconv.Itoa(id) + "] : Starting.")
+	for objId := range objIds {
+
+	}
+	log.Printf("SQLIdentifyMsgWorker[" + strconv.Itoa(id) + "] : Closing.")
+}