Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cfd1a177dd | ||
|
|
db1838b3ab | ||
|
|
ec030676ea | ||
|
|
3b4dd1ba70 |
@@ -87,7 +87,7 @@ func (fe FastExact) _login(username, password string, db storage.DB) error {
|
||||
return err
|
||||
}
|
||||
b, _ := ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return fmt.Errorf("bad status logging into fast exact: %d: %s", resp.StatusCode, b)
|
||||
@@ -152,6 +152,7 @@ func (fe FastExact) searchOneZip(zip string) ([]Job, error) {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
return fe.parse(resp)
|
||||
}
|
||||
|
||||
@@ -165,6 +166,7 @@ func (fe FastExact) searchOneState(state config.State) ([]Job, error) {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
return fe.parse(resp)
|
||||
}
|
||||
|
||||
|
||||
@@ -374,6 +374,9 @@ func (ntg NTGVision) _searchStates(states []config.State) (io.ReadCloser, error)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
b, _ := ioutil.ReadAll(resp.Body)
|
||||
resp.Body = io.NopCloser(bytes.NewReader(b))
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
b, _ := ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
|
||||
7
main.go
7
main.go
@@ -526,15 +526,12 @@ func updateDeadJobs(jobs []broker.Job) error {
|
||||
return err
|
||||
}
|
||||
*/
|
||||
if err := message.NewMatrix().Remove(recorded.MatrixID); err != nil {
|
||||
logtr.Debugf("failed to remove matrix: %v: %v", recorded.MatrixID, err)
|
||||
}
|
||||
if err := db.Set(listEntry, nil); err != nil {
|
||||
logtr.Debugf("failed to remove db: %v: %v", listEntry, err)
|
||||
}
|
||||
for _, imageid := range recorded.MatrixImageIDs {
|
||||
if err := message.NewMatrix().Remove(imageid); err != nil {
|
||||
logtr.Debugf("failed to remove matrix image: %v: %v", imageid, err)
|
||||
if err := message.NewMatrix().Update(imageid, "<job no longer available>"); err != nil {
|
||||
logtr.Debugf("failed to update matrix image: %v: %v", imageid, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -64,7 +64,7 @@ func uploadImage(b []byte) (string, error) {
|
||||
request.Header.Set("Authorization", "Bearer "+images.AccessToken)
|
||||
request.Header.Set("Content-Type", writer.FormDataContentType())
|
||||
|
||||
c := &http.Client{Timeout: time.Minute}
|
||||
c := &http.Client{Timeout: time.Minute * 5}
|
||||
response, err := c.Do(request)
|
||||
if err != nil {
|
||||
return "", err
|
||||
|
||||
@@ -45,13 +45,20 @@ func newMatrix(conf config.Matrix, cont string) Matrix {
|
||||
}
|
||||
}
|
||||
|
||||
func (m Matrix) closeclient(client *gomatrix.Client) {
|
||||
go func() {
|
||||
client.StopSync()
|
||||
client.Logout()
|
||||
}()
|
||||
}
|
||||
|
||||
func (m Matrix) getclient() (*gomatrix.Client, error) {
|
||||
client, err := gomatrix.NewClient(m.homeserver, m.username, m.token)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
client.Client = &http.Client{
|
||||
Timeout: time.Minute,
|
||||
Timeout: time.Minute * 5,
|
||||
Transport: &http.Transport{
|
||||
TLSClientConfig: &tls.Config{
|
||||
InsecureSkipVerify: true,
|
||||
@@ -92,6 +99,7 @@ func (m *Matrix) Receive() ([]Message, error) {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer m.closeclient(c)
|
||||
messages := make([]Message, 0)
|
||||
result, err := c.Messages(m.room, "999999999999999999", m.Continuation(), 'b', 50)
|
||||
if err != nil {
|
||||
@@ -147,6 +155,7 @@ func (m Matrix) Remove(id string) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer m.closeclient(c)
|
||||
_, err = c.RedactEvent(m.room, id, &gomatrix.ReqRedact{Reason: "stale"})
|
||||
return err
|
||||
}
|
||||
@@ -160,6 +169,7 @@ func (m Matrix) Update(id, text string) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer m.closeclient(c)
|
||||
type MRelatesTo struct {
|
||||
EventID string `json:"event_id"`
|
||||
RelType string `json:"rel_type"`
|
||||
@@ -203,6 +213,7 @@ func (m Matrix) SendTracked(text string) (string, error) {
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer m.closeclient(c)
|
||||
resp, err := c.SendText(m.room, text)
|
||||
if err != nil {
|
||||
return "", err
|
||||
@@ -238,6 +249,7 @@ func (m Matrix) SendImageTracked(uri string) (string, error) {
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer m.closeclient(c)
|
||||
mediaUpload, err := c.UploadToContentRepo(bytes.NewReader(b), "image/jpeg", int64(len(b)))
|
||||
if err != nil {
|
||||
return "", err
|
||||
|
||||
Reference in New Issue
Block a user