update

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

internal/storage/sqlite/storage.go

@@ -6,27 +6,26 @@ import (
 	"fmt"
 	"time"
 
-	"git.unistack.org/unistack-org/pkgdash/internal/models"
-	"git.unistack.org/unistack-org/pkgdash/internal/storage"
-	pb "git.unistack.org/unistack-org/pkgdash/proto"
 	"github.com/jmoiron/sqlx"
-	"go.unistack.org/micro/v4/logger"
+	"go.unistack.org/micro/v3/logger"
+	"go.unistack.org/pkgdash/internal/models"
+	"go.unistack.org/pkgdash/internal/storage"
+	pb "go.unistack.org/pkgdash/proto"
 )
 
 func init() {
-	storage.RegisterStorage("sqlite", NewStorage())
+	storage.RegisterStorage("sqlite", NewStorage)
 }
 
 var _ storage.Storage = (*Sqlite)(nil)
 
 type Sqlite struct {
-	db *sqlx.DB
+	logger logger.Logger
+	db     *sqlx.DB
 }
 
-func NewStorage() func(*sqlx.DB) interface{} {
-	return func(db *sqlx.DB) interface{} {
-		return &Sqlite{db: db}
-	}
+func NewStorage(log logger.Logger, db *sqlx.DB) interface{} {
+	return &Sqlite{db: db, logger: log}
 }
 
 func (s *Sqlite) PackageModulesCreate(ctx context.Context, pkg *models.Package, modules []*models.Module) error {
@@ -126,7 +125,7 @@ func (s *Sqlite) CommentCreate(ctx context.Context, req *pb.CommentCreateReq) (*
 	defer func() {
 		if err != nil {
 			if rollbackErr := tx.Rollback(); rollbackErr != nil {
-				logger.Error(ctx, "AddComment: unable to rollback: %v", rollbackErr)
+				s.logger.Error(ctx, "AddComment: unable to rollback: %v", rollbackErr)
 			}
 		} else {
 			err = tx.Commit()