Merge pull request #2 from Knifechik/feat/add-interfaces

feat: fix

Author: Knifechik

data_storage.go

@@ -35,12 +35,13 @@ func (d *dataStorage[K, V]) Set(userID int64, key K, value V) error {
 }
 
 // Get gets user's data from data storage
-func (d *dataStorage[K, V]) Get(userID int64, key K) (any, error) {
+func (d *dataStorage[K, V]) Get(userID int64, key K) (V, error) {
 	d.mu.Lock()
 	defer d.mu.Unlock()
 
 	if _, ok := d.Storage[userID]; !ok {
-		return nil, fmt.Errorf("%w, userID:%d, comparable:%v", errNoUserData, userID, key)
+		var empty V
+		return empty, fmt.Errorf("%w, userID:%d, comparable:%v", errNoUserData, userID, key)
 	}
 
 	return d.Storage[userID][key], nil

fsm.go

@@ -29,7 +29,7 @@ type UserStateStorage interface {
 // DataStorage is an interface for data storage
 type DataStorage[K comparable, V any] interface {
 	Set(userID int64, key K, value V) error
-	Get(userID int64, key K) (any, error)
+	Get(userID int64, key K) (V, error)
 	Delete(userID int64, key K) error
 }
 
@@ -122,10 +122,11 @@ func (f *FSM[K, V]) Set(userID int64, key K, value V) error {
 }
 
 // Get gets a value from data storage by userID and comparable
-func (f *FSM[K, V]) Get(userID int64, key K) (any, error) {
+func (f *FSM[K, V]) Get(userID int64, key K) (V, error) {
 	v, err := f.storage.Get(userID, key)
 	if err != nil {
-		return nil, fmt.Errorf("failed to get user data: %w", err)
+		var empty V
+		return empty, fmt.Errorf("failed to get user data: %w", err)
 	}
 
 	return v, nil