chanched handlers

storage/sqlite/queries.go

@@ -12,7 +12,7 @@ select
     from package;
 `
 	queryAddComment = `
-insert into comment(text) values ($1) ;
+insert into comment(text) values ($1) returning id;
 update package
 set comments = json_insert(comments, '$[#]', ( select last_insert_rowid() as id from comment ))
 where id = $2 ;

storage/sqlite/storage.go

@@ -123,10 +123,10 @@ func (s *Sqlite) ListPackage(ctx context.Context) (models.ListPackage, error) {
 	return result, err
 }
 
-func (s *Sqlite) AddComment(ctx context.Context, req *pb.AddCommentReq) error {
+func (s *Sqlite) AddComment(ctx context.Context, req *pb.AddCommentReq) (id uint64, err error) {
 	tx, err := s.db.BeginTx(ctx, nil)
 	if err != nil {
-		return err
+		return 0, err
 	}
 
 	defer func() {
@@ -139,18 +139,11 @@ func (s *Sqlite) AddComment(ctx context.Context, req *pb.AddCommentReq) error {
 		}
 	}()
 
-	res, err := tx.ExecContext(ctx, queryAddComment, req.Text, req.IdPackage.Value)
-	if err != nil {
-		return err
+	if err = tx.QueryRowContext(ctx, queryAddComment, req.Text, req.IdPackage).Scan(&id); err != nil {
+		return id, err
 	}
-
-	if aff, affErr := res.RowsAffected(); err != nil {
-		err = affErr
-	} else if aff == 0 {
-		err = errors.New("rows affected is 0")
-	}
-
-	return err
+	
+	return id, err
 }
 
 func (s *Sqlite) AddPackage(ctx context.Context, req *pb.AddPackageReq) error {
@@ -169,7 +162,7 @@ func (s *Sqlite) AddPackage(ctx context.Context, req *pb.AddPackageReq) error {
 		}
 	}()
 
-	res, err := tx.ExecContext(ctx, queryAddPackage, req.Name.Value, req.Url.Value, pq.Array(req.Modules))
+	res, err := tx.ExecContext(ctx, queryAddPackage, req.Name, req.Url, pq.Array(req.Modules))
 	if err != nil {
 		return err
 	}