Anthrove/otter-space-sdk
5
0
Fork 0
Code Issues Pull Requests Actions Projects Releases 9 Activity

fix: added documentation & fixed logging

Browse Source
This commit is contained in:
SoXX 2024-08-13 14:01:48 +02:00
parent 71488dade9
commit 3fadaac69a
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pkg/database/userSource.go
View File

@ -16,14 +16,6 @@ func CreateUserSource(ctx context.Context, userSource models.UserSource) (models
ctx, span, localLogger := utils.SetupTracing(ctx, tracer, "CreateUserSource") ctx, span, localLogger := utils.SetupTracing(ctx, tracer, "CreateUserSource")
defer span.End() defer span.End()
localLogger = localLogger.WithFields(log.Fields{
"user_source_id": userSource.ID,
})
span.SetAttributes(
attribute.String("user_source_id", string(userSource.SourceID)),
)
utils.HandleEvent(span, localLogger, "Starting user source creation") utils.HandleEvent(span, localLogger, "Starting user source creation")
if client == nil { if client == nil {
@ -38,10 +30,25 @@ func CreateUserSource(ctx context.Context, userSource models.UserSource) (models
return models.UserSource{}, utils.HandleError(ctx, span, localLogger, result.Error) return models.UserSource{}, utils.HandleError(ctx, span, localLogger, result.Error)
} }
localLogger = localLogger.WithFields(log.Fields{
"user_source_id": userSource.ID,
})
span.SetAttributes(
attribute.String("user_source_id", string(userSource.SourceID)),
)
utils.HandleEvent(span, localLogger, "User source created successfully") utils.HandleEvent(span, localLogger, "User source created successfully")
return userSource, nil return userSource, nil
} }
// UpdateUserSource updates the user source information in the database.
// Only a few parameter can be updated:
// - AccountID
// - ScrapeTimeInterval
// - AccountUsername
// - LastScrapeTime
// - AccountValidate
func UpdateUserSource(ctx context.Context, userSource models.UserSource) error { func UpdateUserSource(ctx context.Context, userSource models.UserSource) error {
ctx, span, localLogger := utils.SetupTracing(ctx, tracer, "UpdateUserSource") ctx, span, localLogger := utils.SetupTracing(ctx, tracer, "UpdateUserSource")
defer span.End() defer span.End()