diff --git a/internal/bd/filmsrepo.go b/internal/bd/filmsrepo.go index 787eeee..b045c3b 100644 --- a/internal/bd/filmsrepo.go +++ b/internal/bd/filmsrepo.go @@ -51,6 +51,19 @@ func (f *Filmsrepo) FindByAll() (*[]model.Films, error) { return &buffs, nil } +func (f *Filmsrepo) GetCountFilms() (*int, error) { + rows, err := f.db.db.Query(context.Background(), "SELECT count(*) FROM films") + var count int + if err != nil { + return nil, err + } + + for rows.Next() { + rows.Scan(&count) + } + return &count, nil +} + func (f *Filmsrepo) FindById(id string) (*model.Films, error) { query := fmt.Sprintf("SELECT * FROM films WHERE Id = %s;", id) var buff model.Films diff --git a/internal/restserver/filmhandle.go b/internal/restserver/filmhandle.go index 33f48f8..e1fcd8a 100644 --- a/internal/restserver/filmhandle.go +++ b/internal/restserver/filmhandle.go @@ -5,7 +5,9 @@ import ( "fmt" "io" "net/http" + "strconv" + resultstruct "git.ukamnya.ru/stulyaganov/RestApi/internal/restserver/resultStruct" "github.com/gorilla/mux" ) @@ -20,17 +22,30 @@ func (r *RestServer) HandleHello() http.HandlerFunc { return func(w http.ResponseWriter, res *http.Request) { id := res.URL.Query().Get("id") fmt.Println(id) - io.WriteString(w, id) + cout, err := r.db.Films().GetCountFilms() + if err != nil { + r.logger.Errorln(err) + } + io.WriteString(w, strconv.Itoa(*cout)) } } func (r *RestServer) HendleFindAll() http.HandlerFunc { - film, err := r.db.Films().FindByAll() + var filmData resultstruct.Films + data, err := r.db.Films().FindByAll() if err != nil { r.logger.Errorln(err) } - - jsonData, err := json.Marshal(film) + cout, err := r.db.Films().GetCountFilms() + if err != nil { + r.logger.Errorln(err) + filmData.Result = false + } else { + filmData.Result = true + filmData.Data = *data + filmData.LastPage = *cout + } + jsonData, err := json.Marshal(filmData) if err != nil { r.logger.Errorln(err) } diff --git a/internal/restserver/resultStruct/films.go b/internal/restserver/resultStruct/films.go new file mode 100644 index 0000000..f67a3a4 --- /dev/null +++ b/internal/restserver/resultStruct/films.go @@ -0,0 +1,10 @@ +package resultstruct + +import "git.ukamnya.ru/stulyaganov/RestApi/internal/bd/model" + +type Films struct { + Result bool + Data []model.Films + Page int + LastPage int +}