blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-08 21:47:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

implementation AesKeyDatabase

Browse Source
This commit is contained in:
YuanJay 2023-11-15 10:05:15 +08:00
parent e7904a09aa
commit edf40893b9
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/common/db/controller/aes_key.go
View File

@ -22,7 +22,7 @@ type aesKeyDatabase struct {
tx tx.Tx
}
func newAesKeyDatabase(key relation.AesKeyModelInterface, tx tx.Tx) *aesKeyDatabase {
func NewAesKeyDatabase(key relation.AesKeyModelInterface, tx tx.Tx) *aesKeyDatabase {
return &aesKeyDatabase{key: key, tx: tx}
}
@ -44,20 +44,36 @@ func (a *aesKeyDatabase) AcquireAesKey(ctx context.Context, conversationType int
}
aesKey, err := a.key.GetAesKey(ctx, keyConversationsID)
if err != nil {
//生成key，并插入
generateAesKey := a.generateAesKey(keyConversationsID)
keyModel := relation.AesKeyModel{
KeyConversationsID: keyConversationsID,
Key: generateAesKey,
ConversationType: conversationType,
OwnerUserID: userId,
FriendUserID: friendId,
GroupID: groupId,
}
var keys []*relation.AesKeyModel
keys = append(keys, &keyModel)
err := a.key.Installs(ctx, keys)
if err != nil {
return &keyModel, err
}
}
return aesKey, nil
}
func (a *aesKeyDatabase) AcquireAesKeys(ctx context.Context, userId string) (key []*relation.AesKeyModel, err error) {
//TODO implement me
panic("implement me")
return a.key.GetAllAesKey(ctx, userId)
}
func (a *aesKeyDatabase) generateKeyConversationsID(args ...string) string {
sort.Strings(args)
combinedStr := strings.Join(args, "")
md5Value := md5.Sum([]byte(combinedStr))
return combinedStr
}
func (a *aesKeyDatabase) generateAesKey(keyConversationsID string) string {
md5Value := md5.Sum([]byte(keyConversationsID))
md5Str := fmt.Sprintf("%x", md5Value)
return md5Str[:16]
}