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 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} 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) aesKey, err := a.key.GetAesKey(ctx, keyConversationsID)
if err != nil { 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 return aesKey, nil
} }
func (a *aesKeyDatabase) AcquireAesKeys(ctx context.Context, userId string) (key []*relation.AesKeyModel, err error) { func (a *aesKeyDatabase) AcquireAesKeys(ctx context.Context, userId string) (key []*relation.AesKeyModel, err error) {
//TODO implement me return a.key.GetAllAesKey(ctx, userId)
panic("implement me")
} }
func (a *aesKeyDatabase) generateKeyConversationsID(args ...string) string { func (a *aesKeyDatabase) generateKeyConversationsID(args ...string) string {
sort.Strings(args) sort.Strings(args)
combinedStr := strings.Join(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) md5Str := fmt.Sprintf("%x", md5Value)
return md5Str[:16] return md5Str[:16]
} }