From 95093616a09387748ada2b22287fbbfb2cdb4532 Mon Sep 17 00:00:00 2001
From: "Timofey.Kovalev" <timofey.kovalev@acronis.com>
Date: Wed, 23 Jun 2021 00:49:08 +0300
Subject: [PATCH] small fixes

---
 dataBase.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/dataBase.go b/dataBase.go
index a54664b..e843160 100644
--- a/dataBase.go
+++ b/dataBase.go
@@ -134,6 +134,10 @@ func (db *dbLayer) getPlayerByID(ctx context.Context, id string) (*Player, error
 
 	row := db.db.QueryRowContext(ctx, query, id)
 	if err := row.Err(); err != nil {
+		if err == sql.ErrNoRows {
+			return nil, nil
+		}
+
 		return nil, errors.Wrapf(err, "Failed to run db query [%s]", query)
 	}
 
@@ -152,6 +156,10 @@ func (db *dbLayer) getPlayerByName(ctx context.Context, name string) (*Player, e
 
 	row := db.db.QueryRowContext(ctx, query, name)
 	if err := row.Err(); err != nil {
+		if err == sql.ErrNoRows {
+			return nil, nil
+		}
+
 		return nil, errors.Wrapf(err, "Failed to run db query [%s]", query)
 	}
 
@@ -170,6 +178,10 @@ func (db *dbLayer) getKillsByPlayerID(ctx context.Context, playerID string) (int
 
 	row := db.db.QueryRowContext(ctx, query, playerID)
 	if err := row.Err(); err != nil {
+		if err == sql.ErrNoRows {
+			return 0, nil
+		}
+
 		return 0, errors.Wrapf(err, "Failed to run db query [%s]", query)
 	}