feat: go chat 1.0

Author: kone-net

.gitignore

@@ -0,0 +1,24 @@
+bin/
+release/
+test/tmp/
+test/logs/
+logs
+doc/blueprint/
+*.iml
+*.swp
+*.log
+coverage.*
+y.output
+
+.DS_Store
+.vscode/
+.idea
+_tools/
+
+TestMarkdown2Html.html
+
+#test log
+test/logs
+
+#upload file
+static/img/*

Makefile

@@ -0,0 +1,61 @@
+BINARY="chat"
+
+# colors compatible setting
+CRED:=$(shell tput setaf 1 2>/dev/null)
+CGREEN:=$(shell tput setaf 2 2>/dev/null)
+CYELLOW:=$(shell tput setaf 3 2>/dev/null)
+CEND:=$(shell tput sgr0 2>/dev/null)
+
+.PHONY: all
+all: | fmt build
+
+# Code format
+.PHONY: fmt
+fmt: 
+	@echo "$(CGREEN)Run gofmt ...$(CEND)"
+	@echo "gofmt -l -s -w ..."
+	@ret=0 && for d in $$(go list -f '{{.Dir}}' ./... | grep -v /vendor/); do \
+		gofmt -l -s -w $$d/*.go || ret=$$? ; \
+	done ; exit $$ret
+
+# build
+.PHONY: build-darwin
+build-darwin: fmt
+	@echo "$(CGREEN)Building for darwin ...$(CEND)"
+	@mkdir -p bin
+	CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build -o bin/${BINARY} cmd/main.go
+	@echo "$(CGREEN)Build Success!$(CEND)"
+
+# build
+.PHONY: build
+build: fmt
+	@echo "$(CGREEN)Building for linux ...$(CEND)"
+	@mkdir -p bin
+	CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o bin/${BINARY} cmd/main.go
+	@echo "$(CGREEN)Build Success!$(CEND)"
+
+# install
+install: build
+	@echo "$(CGREEN)Install ...$(CEND)"
+	go install ./...
+	@echo "$(CGREEN)Install Success!$(CEND)"
+
+# clean
+.PHONY: clean
+clean:
+	@echo "$(CGREEN)Cleanup ...$(CEND)"
+	go clean
+	@rm -f bin/${BINARY}
+	@echo "rm -f bin/${BINARY}"
+	@for GOOS in darwin linux windows; do \
+	    for GOARCH in 386 amd64; do \
+			rm -f bin/${BINARY}.$${GOOS}-$${GOARCH} ;\
+		done ;\
+	done
+	rm -f ${BINARY} coverage.* test/tmp/*
+	find . -name "*.log" -delete
+
+# protoc build
+.PHONY: protoc
+protoc:
+	protoc --gogo_out=. protocol/*.proto

README.md

@@ -0,0 +1,5 @@
+使用go开发的聊天程序。
+前端代码仓库：
+https://github.com/kone-net/go-chat-web
+
+已完成发送文字，截图，照片，文件，语音，视频等。

api/v1/file_router.go

@@ -0,0 +1,17 @@
+package v1
+
+import (
+	"io/ioutil"
+
+	"chat-room/global/log"
+
+	"github.com/gin-gonic/gin"
+)
+
+// 前端通过文件名称获取文件流，显示文件
+func GetFile(c *gin.Context) {
+	fileName := c.Param("fileName")
+	log.Info(fileName)
+	data, _ := ioutil.ReadFile("static/img/" + fileName)
+	c.Writer.Write(data)
+}

api/v1/group_router.go

@@ -0,0 +1,50 @@
+package v1
+
+import (
+	"chat-room/model"
+	"chat-room/response"
+	"chat-room/service"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+// 获取分组列表
+func GetGroup(c *gin.Context) {
+	uuid := c.Param("uuid")
+	groups, err := service.GroupService.GetGroups(uuid)
+	if err != nil {
+		c.JSON(http.StatusOK, response.FailMsg(err.Error()))
+		return
+	}
+
+	c.JSON(http.StatusOK, response.SuccessMsg(groups))
+}
+
+// 保存分组列表
+func SaveGroup(c *gin.Context) {
+	uuid := c.Param("uuid")
+	var group model.Group
+	c.ShouldBindJSON(&group)
+
+	service.GroupService.SaveGroup(uuid, group)
+	c.JSON(http.StatusOK, response.SuccessMsg(nil))
+}
+
+// 加入组别
+func JoinGroup(c *gin.Context) {
+	userUuid := c.Param("userUuid")
+	groupUuid := c.Param("groupUuid")
+	err := service.GroupService.JoinGroup(groupUuid, userUuid)
+	if err != nil {
+		c.JSON(http.StatusOK, response.FailMsg(err.Error()))
+		return
+	}
+	c.JSON(http.StatusOK, response.SuccessMsg(nil))
+}
+
+// 获取组内成员信息
+func GetGroupUsers(c *gin.Context) {
+	groupUuid := c.Param("uuid")
+	users := service.GroupService.GetUserIdByGroupUuid(groupUuid)
+	c.JSON(http.StatusOK, response.SuccessMsg(users))
+}

api/v1/message_router.go

@@ -0,0 +1,31 @@
+package v1
+
+import (
+	"net/http"
+
+	"chat-room/global/log"
+	"chat-room/model/request"
+	"chat-room/response"
+	"chat-room/service"
+
+	"github.com/gin-gonic/gin"
+)
+
+// 获取消息列表
+func GetMessage(c *gin.Context) {
+	log.Info(c.Query("uuid"))
+	var messageRequest request.MessageRequest
+	err := c.BindQuery(&messageRequest)
+	if nil != err {
+		log.Error("bindQueryError", log.Any("bindQueryError", err))
+	}
+	log.Info("messageRequest params: ", log.Any("messageRequest", messageRequest))
+
+	messages, err := service.MessageService.GetMessages(messageRequest)
+	if err != nil {
+		c.JSON(http.StatusOK, response.FailMsg(err.Error()))
+		return
+	}
+
+	c.JSON(http.StatusOK, response.SuccessMsg(messages))
+}

api/v1/user_router.go

@@ -0,0 +1,76 @@
+package v1
+
+import (
+	"net/http"
+
+	"chat-room/global/log"
+	"chat-room/model"
+	"chat-room/model/request"
+	"chat-room/response"
+	"chat-room/service"
+
+	"github.com/gin-gonic/gin"
+)
+
+func Login(c *gin.Context) {
+	var user model.User
+	// c.BindJSON(&user)
+	c.ShouldBindJSON(&user)
+	log.Debug("user", log.Any("user", user))
+
+	if service.UserService.Login(&user) {
+		c.JSON(http.StatusOK, response.SuccessMsg(user))
+		return
+	}
+
+	c.JSON(http.StatusOK, response.FailMsg("Login failed"))
+}
+
+func Register(c *gin.Context) {
+	var user model.User
+	c.ShouldBindJSON(&user)
+	err := service.UserService.Register(&user)
+	if err != nil {
+		c.JSON(http.StatusOK, response.FailMsg(err.Error()))
+		return
+	}
+
+	c.JSON(http.StatusOK, response.SuccessMsg(user))
+}
+
+func ModifyUserInfo(c *gin.Context) {
+	var user model.User
+	c.ShouldBindJSON(&user)
+	log.Debug("user", log.Any("user", user))
+	if err := service.UserService.ModifyUserInfo(&user); err != nil {
+		c.JSON(http.StatusOK, response.FailMsg(err.Error()))
+		return
+	}
+
+	c.JSON(http.StatusOK, response.SuccessMsg(nil))
+}
+
+
+func GetUserDetails(c *gin.Context) {
+	uuid := c.Param("uuid")
+
+	c.JSON(http.StatusOK, response.SuccessMsg(service.UserService.GetUserDetails(uuid)))
+}
+
+func GetUserList(c *gin.Context) {
+	uuid := c.Query("uuid")
+	c.JSON(http.StatusOK, response.SuccessMsg(service.UserService.GetUserList(uuid)))
+}
+
+func AddFriend(c *gin.Context) {
+	var userFriendRequest request.FriendRequest
+	c.ShouldBindJSON(&userFriendRequest)
+
+	err := service.UserService.AddFriend(&userFriendRequest)
+	if nil != err {
+		c.JSON(http.StatusOK, response.FailMsg(err.Error()))
+		return
+	}
+
+	c.JSON(http.StatusOK, response.SuccessMsg(nil))
+}

chat.sql

@@ -0,0 +1,82 @@
+DROP TABLE IF EXISTS `users`;
+CREATE TABLE IF NOT EXISTS `users` (
+    `id` int NOT NULL AUTO_INCREMENT COMMENT 'id',
+    `uuid` varchar(150) NOT NULL COMMENT 'uuid',
+    `username` varchar(191) NOT NULL COMMENT '''用户名''',
+    `nickname` varchar(255) DEFAULT NULL COMMENT '昵称',
+    `email` varchar(80) DEFAULT NULL COMMENT '邮箱',
+    `password` varchar(150) NOT NULL COMMENT '密码',
+    `avatar` varchar(250) NOT NULL COMMENT '头像',
+    `create_at` datetime(3) DEFAULT NULL,
+    `update_at` datetime(3) DEFAULT NULL,
+    `delete_at` bigint DEFAULT NULL,
+    PRIMARY KEY (`id`),
+    UNIQUE KEY `username` (`username`),
+    UNIQUE KEY `idx_uuid` (`uuid`),
+    UNIQUE KEY `username_2` (`username`)
+) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 COLLATE = utf8mb4_0900_ai_ci COMMENT '用户表';
+
+
+DROP TABLE IF EXISTS `user_friends`;
+CREATE TABLE IF NOT EXISTS `user_friends` (
+    `id` int NOT NULL AUTO_INCREMENT COMMENT 'id',
+    `created_at` datetime(3) DEFAULT NULL COMMENT '创建时间',
+    `updated_at` datetime(3) DEFAULT NULL COMMENT '更新时间',
+    `deleted_at` bigint unsigned DEFAULT NULL COMMENT '删除时间戳',
+    `user_id` int DEFAULT NULL COMMENT '用户ID',
+    `friend_id` int DEFAULT NULL COMMENT '好友ID',
+    PRIMARY KEY (`id`),
+    KEY `idx_user_friends_user_id` (`user_id`),
+    KEY `idx_user_friends_friend_id` (`friend_id`)
+) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 COLLATE = utf8mb4_0900_ai_ci COMMENT '好友信息表';
+
+
+DROP TABLE IF EXISTS `messages`;
+CREATE TABLE IF NOT EXISTS `messages` (
+  `id` bigint unsigned NOT NULL AUTO_INCREMENT COMMENT 'id',
+  `created_at` datetime(3) DEFAULT NULL COMMENT '创建时间',
+  `updated_at` datetime(3) DEFAULT NULL COMMENT '更新时间',
+  `deleted_at` bigint unsigned DEFAULT NULL COMMENT '删除时间戳',
+  `from_user_id` int DEFAULT NULL COMMENT '发送人ID',
+  `to_user_id` int DEFAULT NULL COMMENT '发送对象ID',
+  `content` varchar(2500) DEFAULT NULL COMMENT '消息内容',
+  `url` varchar(350) DEFAULT NULL COMMENT '''文件或者图片地址''',
+  `pic` text COMMENT '缩略图',
+  `message_type` smallint DEFAULT NULL COMMENT '''消息类型：1单聊，2群聊''',
+  `content_type` smallint DEFAULT NULL COMMENT '''消息内容类型：1文字，2语音，3视频''',
+  PRIMARY KEY (`id`),
+  KEY `idx_messages_deleted_at` (`deleted_at`),
+  KEY `idx_messages_from_user_id` (`from_user_id`),
+  KEY `idx_messages_to_user_id` (`to_user_id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT '消息表';
+
+
+DROP TABLE IF EXISTS `groups`;
+CREATE TABLE IF NOT EXISTS  `groups` (
+  `id` int NOT NULL AUTO_INCREMENT,
+  `created_at` datetime(3) DEFAULT NULL,
+  `updated_at` datetime(3) DEFAULT NULL,
+  `deleted_at` bigint unsigned DEFAULT NULL,
+  `user_id` int DEFAULT NULL COMMENT '''群主ID''',
+  `name` varchar(150) DEFAULT NULL COMMENT '''群名称',
+  `notice` varchar(350) DEFAULT NULL COMMENT '''群公告',
+  `uuid` varchar(150) NOT NULL COMMENT '''uuid''',
+  PRIMARY KEY (`id`),
+  KEY `idx_groups_user_id` (`user_id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT '群组表';
+
+
+DROP TABLE IF EXISTS `group_members`;
+CREATE TABLE  IF NOT EXISTS `group_members` (
+  `id` int NOT NULL AUTO_INCREMENT,
+  `created_at` datetime(3) DEFAULT NULL,
+  `updated_at` datetime(3) DEFAULT NULL,
+  `deleted_at` bigint unsigned DEFAULT NULL,
+  `user_id` int DEFAULT NULL COMMENT '''用户ID''',
+  `group_id` int DEFAULT NULL COMMENT '''群组ID''',
+  `nickname` varchar(350) DEFAULT NULL COMMENT '''昵称',
+  `mute` smallint DEFAULT NULL COMMENT '''是否禁言''',
+  PRIMARY KEY (`id`),
+  KEY `idx_group_members_user_id` (`user_id`),
+  KEY `idx_group_members_group_id` (`group_id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT '群组成员表';

cmd/main.go

@@ -0,0 +1,34 @@
+package main
+
+import (
+	"chat-room/global/log"
+	"chat-room/config"
+	"chat-room/router"
+	"chat-room/server"
+	"go.uber.org/zap"
+	"net/http"
+	"time"
+)
+
+func main() {
+	log.InitLogger(config.GetConfig().Log.Path, config.GetConfig().Log.Level)
+	log.Info("config", zap.Any("config", config.GetConfig()))
+
+	log.Info("start server", zap.String("start", "start web sever..."))
+
+	newRouter := router.NewRouter()
+
+	go server.MyServer.Start()
+
+	s := &http.Server{
+		Addr:           "127.0.0.1:8888",
+		Handler:        newRouter,
+		ReadTimeout:    10 * time.Second,
+		WriteTimeout:   10 * time.Second,
+		MaxHeaderBytes: 1 << 20,
+	}
+	err := s.ListenAndServe()
+	if nil != err {
+		log.Error("server error", zap.Any("serverError", err))
+	}
+}