55 "encoding/json"
66 "flag"
77 "fmt"
8+ "github.com/LF-Engineering/insights-datasource-shared/aws"
89 "io"
910 "math"
1011 "net/url"
@@ -618,9 +619,15 @@ func (j *DSGit) AddLogger(ctx *shared.Ctx) {
618619}
619620
620621// WriteLog - writes to log
621- func (j * DSGit ) WriteLog (ctx * shared.Ctx , timestamp time.Time , status , message string ) {
622- _ = j .Logger .Write (& logger.Log {
622+ func (j * DSGit ) WriteLog (ctx * shared.Ctx , timestamp time.Time , status , message string ) error {
623+ arn , err := aws .GetContainerARN ()
624+ if err != nil {
625+ j .log .WithFields (logrus.Fields {"operation" : "WriteLog" }).Errorf ("getContainerMetadata Error : %+v" , err )
626+ return err
627+ }
628+ err = j .Logger .Write (& logger.Log {
623629 Connector : GitDataSource ,
630+ TaskARN : arn ,
624631 Configuration : []map [string ]string {
625632 {
626633 "REPO_URL" : j .URL ,
@@ -630,6 +637,7 @@ func (j *DSGit) WriteLog(ctx *shared.Ctx, timestamp time.Time, status, message s
630637 CreatedAt : timestamp ,
631638 Message : message ,
632639 })
640+ return err
633641}
634642
635643// AddFlags - add git specific flags
@@ -2647,14 +2655,21 @@ func main() {
26472655 shared .SetLogLoggerError (false )
26482656 shared .AddLogger (& git .Logger , GitDataSource , logger .Internal , []map [string ]string {{"REPO_URL" : git .URL , "ProjectSlug" : ctx .Project }})
26492657 git .AddCacheProvider ()
2650- git .WriteLog (& ctx , timestamp , logger .InProgress , "" )
2658+ err = git .WriteLog (& ctx , timestamp , logger .InProgress , "" )
2659+ if err != nil {
2660+ git .log .WithFields (logrus.Fields {"operation" : "main" }).Errorf ("WriteLog Error : %+v" , err )
2661+ return
2662+ }
26512663 err = git .Sync (& ctx )
26522664 if err != nil {
26532665 git .log .WithFields (logrus.Fields {"operation" : "main" }).Errorf ("Error: %+v" , err )
2654- git .WriteLog (& ctx , timestamp , logger .Failed , err .Error ())
2666+ er := git .WriteLog (& ctx , timestamp , logger .Failed , err .Error ())
2667+ if er != nil {
2668+ err = er
2669+ }
26552670 return
26562671 }
2657- git .WriteLog (& ctx , timestamp , logger .Done , "" )
2672+ err = git .WriteLog (& ctx , timestamp , logger .Done , "" )
26582673}
26592674
26602675// createStructuredLogger...
0 commit comments