unistack-org/pkgdash
4
0
Fork 1
Code Issues 2 Pull Requests 1 Packages Projects Releases Wiki Activity

add logic to GetModule handler

Browse Source
This commit is contained in:
Денис Евстигнеев
2023-08-11 17:07:37 +03:00
parent 7de883b8c0
commit 71f438e108
8 changed files with 169 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
storage/sqlite/storage.go
View File

@@ -225,15 +225,15 @@ func (s *Sqlite) InsertButchModules(ctx context.Context, req []models.Module) ([
return result, err
}
func (s *Sqlite) GetModule(ctx context.Context, req *pb.GetModuleReq) (result []models.Module, err error) {
func (s *Sqlite) GetModule(ctx context.Context, req *pb.GetModuleReq) (result models.ListModule, err error) {
query := ""
if len(req.ModulesId) < 1 {
if len(req.Id) < 1 {
query = fmt.Sprintf(queryGetModule, "() or 1=1")
} else {
query = fmt.Sprintf(queryGetModule, generateArrayIneq(len(req.GetModulesId())))
query = fmt.Sprintf(queryGetModule, generateArrayIneq(len(req.Id)))
}
rows, err := s.db.QueryContext(ctx, query, convertSliceUInt(req.ModulesId...)...)
rows, err := s.db.QueryContext(ctx, query, convertSliceUInt(req.Id...)...)
if err != nil {
return nil, err
}
@@ -245,7 +245,7 @@ func (s *Sqlite) GetModule(ctx context.Context, req *pb.GetModuleReq) (result []
}()
for rows.Next() {
tmp := models.Module{}
tmp := &models.Module{}
if err = rows.Scan(
&tmp.ID,
&tmp.Name,