Aviato updates

Author: caffix

engine/plugins/api/aviato/company_search.go

@@ -90,7 +90,7 @@ func (cs *companySearch) query(e *et.Event, orgent *dbt.Entity, apikey []string)
 		headers["Authorization"] = []string{"Bearer " + key}
 
 		_ = cs.plugin.rlimit.Wait(context.TODO())
-		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+		ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
 		defer cancel()
 
 		filters := []map[string]*dslEvalObj{
@@ -134,10 +134,10 @@ func (cs *companySearch) query(e *et.Event, orgent *dbt.Entity, apikey []string)
 		return nil
 	}
 
-	return cs.store(e, orgent, result.Items[0].ID, 90)
+	return cs.store(e, orgent, result.Items[0].ID)
 }
 
-func (cs *companySearch) store(e *et.Event, orgent *dbt.Entity, companyID string, conf int) *dbt.Entity {
+func (cs *companySearch) store(e *et.Event, orgent *dbt.Entity, companyID string) *dbt.Entity {
 	oamid := &general.Identifier{
 		UniqueID: fmt.Sprintf("%s:%s", AviatoCompanyID, companyID),
 		ID:       companyID,
@@ -151,10 +151,11 @@ func (cs *companySearch) store(e *et.Event, orgent *dbt.Entity, companyID string
 
 	_, _ = e.Session.Cache().CreateEntityProperty(ident, &general.SourceProperty{
 		Source:     cs.plugin.source.Name,
-		Confidence: conf,
+		Confidence: cs.plugin.source.Confidence,
 	})
 
-	if err := cs.plugin.createRelation(e.Session, orgent, general.SimpleRelation{Name: "id"}, ident, conf); err != nil {
+	if err := cs.plugin.createRelation(e.Session, orgent,
+		general.SimpleRelation{Name: "id"}, ident, cs.plugin.source.Confidence); err != nil {
 		return nil
 	}
 	return ident

engine/plugins/api/aviato/employees.go

@@ -118,7 +118,7 @@ loop:
 			headers["Authorization"] = []string{"Bearer " + key}
 
 			_ = ae.plugin.rlimit.Wait(context.TODO())
-			ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+			ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
 			defer cancel()
 
 			u := fmt.Sprintf("https://data.api.aviato.co/company/%s/employees?perPage=%d&page=%d", oamid.ID, perPage, page)
@@ -183,10 +183,11 @@ func (ae *employees) store(e *et.Event, ident *dbt.Entity, employlist []*employe
 
 		_, _ = e.Session.Cache().CreateEntityProperty(ident, &general.SourceProperty{
 			Source:     ae.plugin.source.Name,
-			Confidence: 90,
+			Confidence: ae.plugin.source.Confidence,
 		})
 
-		if err := ae.plugin.createRelation(e.Session, orgent, general.SimpleRelation{Name: "member"}, personent, 90); err == nil {
+		if err := ae.plugin.createRelation(e.Session, orgent,
+			general.SimpleRelation{Name: "member"}, personent, ae.plugin.source.Confidence); err == nil {
 			employents = append(employents, personent)
 		}
 	}

engine/plugins/api/aviato/plugin.go

@@ -16,7 +16,7 @@ import (
 )
 
 func NewAviato() et.Plugin {
-	limit := rate.Every(time.Second)
+	limit := rate.Every(2 * time.Second)
 
 	return &aviato{
 		name:   "Aviato",

engine/plugins/api/aviato/types.go

@@ -12,6 +12,7 @@ import (
 )
 
 const (
+	AviatoPersonID  = "aviato_person_id"
 	AviatoCompanyID = "aviato_company_id"
 )