连接数据库

entity/message.go

@@ -0,0 +1,16 @@
+package entity
+
+import "github.com/go-xorm/xorm"
+
+//定义orm引擎
+var Engine *xorm.Engine
+
+type Message struct {
+	Id            int `xorm:"id" form:"id" json:"id"`                                     // 发送方用户id
+	ReceiveUserid int `xorm:"receive_userid" form:"receive_userid" json:"receive_userid"` // 接收方用户id
+	SendUserid    int `xorm:"send_userid" form:"send_userid" json:"send_userid"`          // 消息id
+	CreateTime    int `xorm:"createtime" form:"createtime" json:"createtime"`             // 创建时间
+	Content       int `xorm:"content" form:"content" json:"content"`                      // '内容
+	IsRead        int `xorm:"isRead" form:"isRead" json:"isRead"`                         //  '0：未读  1：已读'
+	Isdel         int `xorm:"isdel" form:"isdel" json:"isdel"`                            // 1：未删除  -1：已删除
+}

rpc/etc/message.yaml

@@ -4,7 +4,7 @@ Etcd:
   Hosts:
   - 127.0.0.1:2379
   Key: message.rpc
-DataSource: root:Topnet123@tcp(192.168.3.11:3366)/jyactivities?charset=utf8mb4&parseTime=true&loc=Local
+DataSource: root:Topnet123@tcp(127.0.0.1:3366)/messageCenter?charset=utf8mb4&parseTime=true&loc=Local
 FileSystemConf:
   Etcd:
     Hosts:

rpc/internal/config/config.go

@@ -7,3 +7,4 @@ type Config struct {
 	DataSource     string // 手动代码
 
 }
+var  ConfigJson Config

rpc/internal/logic/changereadstatuslogic.go

@@ -1,7 +1,9 @@
 package logic
 
 import (
+	"app.yhyue.com/moapp/MessageCenter/service"
 	"context"
+	"log"
 
 	"app.yhyue.com/moapp/MessageCenter/rpc/internal/svc"
 	"app.yhyue.com/moapp/MessageCenter/rpc/message"
@@ -25,6 +27,8 @@ func NewChangeReadStatusLogic(ctx context.Context, svcCtx *svc.ServiceContext) *
 
 func (l *ChangeReadStatusLogic) ChangeReadStatus(in *message.ChangeReadStatusRequest) (*message.Response, error) {
 	// todo: add your logic here and delete this line
-
+	m:=&service.MessageService{}
+	log.Println("12222")
+	m.ChangeReadStatus()
 	return &message.Response{}, nil
 }

rpc/message.go

@@ -4,13 +4,17 @@
 package main
 
 import (
+	"app.yhyue.com/moapp/MessageCenter/entity"
 	"flag"
 	"fmt"
+	"github.com/go-xorm/xorm"
+	"log"
 
 	"app.yhyue.com/moapp/MessageCenter/rpc/internal/config"
 	"app.yhyue.com/moapp/MessageCenter/rpc/internal/server"
 	"app.yhyue.com/moapp/MessageCenter/rpc/internal/svc"
 	"app.yhyue.com/moapp/MessageCenter/rpc/message"
+	_ "github.com/go-sql-driver/mysql"
 
 	"github.com/tal-tech/go-zero/core/conf"
 	"github.com/tal-tech/go-zero/zrpc"
@@ -35,3 +39,16 @@ func main() {
 	fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
 	s.Start()
 }
+//创建orm引擎
+func init() {
+	conf.MustLoad(*configFile, &config.ConfigJson)
+	var err error
+	log.Println(config.ConfigJson.DataSource)
+	entity.Engine, err = xorm.NewEngine("mysql", config.ConfigJson.DataSource)
+	log.Println(err)
+	entity.Engine.ShowSQL(true)
+	if err != nil {
+		log.Fatal("数据库连接失败:", err)
+	}
+	fmt.Println(config.ConfigJson.DataSource+"链接成功")
+}

rpc/test/message_test.go

@@ -0,0 +1,25 @@
+package test
+
+import (
+	"app.yhyue.com/moapp/MessageCenter/rpc/messageclient"
+	"context"
+	"github.com/tal-tech/go-zero/core/discov"
+	"github.com/tal-tech/go-zero/zrpc"
+	"log"
+	"testing"
+	"time"
+)
+
+func  Test_ChangeReadStatus(t *testing.T)  {
+	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
+	std := messageclient.NewMessage(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "message.rpc"}}))
+	//std := stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{Etcd: discov.EtcdConf{Hosts: []string{"127.0.0.1:2379"}, Key: "jydocs.stdlib.rpc"}}))
+	req := &messageclient.ChangeReadStatusRequest{Id: 1}
+	res, err := std.ChangeReadStatus(ctx, req)
+	log.Println("err ", err,res)
+	//if res.State == true {
+	//	log.Println("req true")
+	//} else {
+	//	log.Println("req false", res)
+	//}
+}

service/messageService.go

@@ -0,0 +1,22 @@
+package service
+
+import (
+	"app.yhyue.com/moapp/MessageCenter/entity"
+	"log"
+)
+
+type MessageService struct {
+}
+	func (service *MessageService) ChangeReadStatus() (int64, string) {
+	orm := entity.Engine.NewSession()
+	log.Println("123")
+	defer orm.Close()
+	m:=entity.Message{}
+	_, err := orm.Table("message").Count(m)
+	log.Println("err",err)
+	if err != nil {
+		return 0, ""
+	}
+log.Println(m)
+	return 0,""
+}