57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
|
package e621
|
||
|
|
|
||
|
|
import (
|
||
|
|
"e621_to_neo4j/utils"
|
||
|
|
"encoding/json"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
type e621APITask[T utils.DataType] struct {
|
||
|
|
uri string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e e621APITask[T]) UriPath() string {
|
||
|
|
return e.uri
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e e621APITask[T]) HandleStatusCode(statusCode int) utils.DataResponse[T] {
|
||
|
|
var err error
|
||
|
|
switch statusCode {
|
||
|
|
case 421:
|
||
|
|
err = utils.RateLimitReachedError{}
|
||
|
|
case 424:
|
||
|
|
err = utils.InvalidParametersError{}
|
||
|
|
case 520:
|
||
|
|
err = utils.OriginConnectionTimeOutError{}
|
||
|
|
case 522:
|
||
|
|
err = utils.OriginConnectionTimeOutError{}
|
||
|
|
case 524:
|
||
|
|
err = utils.OriginConnectionTimeOutError{}
|
||
|
|
case 525:
|
||
|
|
err = utils.SSLHandshakeFailedError{}
|
||
|
|
default:
|
||
|
|
err = utils.StatusCodesToError(statusCode)
|
||
|
|
}
|
||
|
|
return utils.DataResponse[T]{Error: err}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e e621APITask[T]) HandleResponse(responseData *http.Response) utils.DataResponse[T] {
|
||
|
|
var data T
|
||
|
|
err := json.NewDecoder(responseData.Body).Decode(&data)
|
||
|
|
defer responseData.Body.Close()
|
||
|
|
if err != nil {
|
||
|
|
return utils.DataResponse[T]{Error: err}
|
||
|
|
}
|
||
|
|
return utils.DataResponse[T]{Data: data}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e e621APITask[T]) HandleError(error error) utils.DataResponse[T] {
|
||
|
|
return utils.DataResponse[T]{Error: error}
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewE621ApiTask[T utils.DataType](uri string) utils.Task[T] {
|
||
|
|
return &e621APITask[T]{
|
||
|
|
uri: uri,
|
||
|
|
}
|
||
|
|
}
|