2023-07-31 10:48:32 +08:00

28 lines
775 B
Go

package controller
import (
"context"
relationTb "github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/relation"
)
type KeyDatabase interface {
DBGetKey(ctx context.Context, cID string) (key relationTb.KeyModel, err error)
DBInstallKey(ctx context.Context, key relationTb.KeyModel) (err error)
}
type keyDatabase struct {
keyDB relationTb.KeyModelInterface
//cache cache.KeyCache
}
func NewKeyDatabase(key relationTb.KeyModelInterface) KeyDatabase {
return &keyDatabase{
keyDB: key,
}
}
func (k *keyDatabase) DBGetKey(ctx context.Context, cID string) (key relationTb.KeyModel, err error) {
return k.keyDB.GetKey(ctx, cID)
}
func (k *keyDatabase) DBInstallKey(ctx context.Context, key relationTb.KeyModel) (err error) {
return k.keyDB.InstallKey(ctx, key)
}