James Tombleson
2b959e140c
* updated db calls to new endpoints. Not all are finished yet * updated models to use the new dto objects to hide orm values * updated reddit to build articles based on what type of post it is * getting reddit ready to post the articles to the db * Now able to post to the db under the new DTO object * moved the reddit model to articles model to the struct and out of main
100 lines
1.9 KiB
Go
100 lines
1.9 KiB
Go
package database
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/jtom38/newsbot/collector/domain/model"
|
|
)
|
|
|
|
// Generate this struct fr
|
|
type ArticlesClient struct {
|
|
rootUri string
|
|
}
|
|
|
|
func (ac *ArticlesClient) List() ([]model.Articles, error) {
|
|
var items []model.Articles
|
|
url := fmt.Sprintf("%v/api/v1/articles", ac.rootUri)
|
|
resp, err := getContent(url)
|
|
if err != nil {
|
|
return items, err
|
|
}
|
|
|
|
err = json.Unmarshal(resp, &items)
|
|
if err != nil {
|
|
return []model.Articles{}, err
|
|
}
|
|
|
|
return items, nil
|
|
}
|
|
|
|
func (ac *ArticlesClient) FindByID(ID uint) (model.Articles, error) {
|
|
var items model.Articles
|
|
url := fmt.Sprintf("%v/api/v1/articles/%v", ac.rootUri, ID)
|
|
resp, err := getContent(url)
|
|
if err != nil {
|
|
return items, err
|
|
}
|
|
|
|
err = json.Unmarshal(resp, &items)
|
|
if err != nil {
|
|
return items, err
|
|
}
|
|
|
|
return items, nil
|
|
}
|
|
|
|
func (ac *ArticlesClient) FindByUrl(url string) (model.Articles, error) {
|
|
var item model.Articles
|
|
get := fmt.Sprintf("%v/api/v1/articles/url/%v", ac.rootUri, url)
|
|
resp, err := getContent(get)
|
|
if err != nil {
|
|
return item, err
|
|
}
|
|
|
|
err = json.Unmarshal(resp, &item)
|
|
if err != nil {
|
|
return item, err
|
|
}
|
|
|
|
return item, nil
|
|
}
|
|
|
|
func (ac *ArticlesClient) Delete(id int32) error {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
func (ac *ArticlesClient) Add(item model.Articles) error {
|
|
//return errors.New("not implemented")
|
|
url := fmt.Sprintf("%v/api/v1/articles/", ac.rootUri)
|
|
|
|
bItem, err := json.Marshal(item)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client := &http.Client{}
|
|
req, err := http.NewRequest("POST", url, bytes.NewBuffer(bItem))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req.Header.Set("Content-Type", "application/json")
|
|
resp, err := client.Do(req)
|
|
defer resp.Body.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if resp.StatusCode != 200 {
|
|
return errors.New("failed to post to the DB")
|
|
}
|
|
|
|
return nil
|
|
//body, err := ioutil.ReadAll(resp.Body)
|
|
//if err != nil { return err }
|
|
|
|
}
|