AcquireAesKey

pkg/common/db/controller/aes_key.go

@@ -3,7 +3,9 @@ package controller
 import (
 	"context"
 	"crypto/md5"
+	"errors"
 	"fmt"
+	"github.com/OpenIMSDK/protocol/constant"
 	"github.com/OpenIMSDK/tools/tx"
 	"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
 	"sort"
@@ -25,8 +27,26 @@ func newAesKeyDatabase(key relation.AesKeyModelInterface, tx tx.Tx) *aesKeyDatab
 }
 
 func (a *aesKeyDatabase) AcquireAesKey(ctx context.Context, conversationType int32, userId, friendId, groupId string) (key *relation.AesKeyModel, err error) {
-	//TODO implement me
-	panic("implement me")
+	var keyConversationsID string
+	switch conversationType {
+	case constant.SingleChatType:
+		if userId == "" || friendId == "" {
+			return nil, errors.New("userId or friendId is null")
+		}
+		keyConversationsID = a.generateKeyConversationsID(userId, friendId)
+	case constant.GroupChatType:
+		if userId == "" || groupId == "" {
+			return nil, errors.New("userId or groupId is null")
+		}
+		keyConversationsID = a.generateKeyConversationsID(groupId)
+	default:
+		return nil, errors.New("conversationType err")
+	}
+	aesKey, err := a.key.GetAesKey(ctx, keyConversationsID)
+	if err != nil {
+		//生成key，并插入
+	}
+	return aesKey, nil
 }
 
 func (a *aesKeyDatabase) AcquireAesKeys(ctx context.Context, userId string) (key []*relation.AesKeyModel, err error) {