diff --git a/pkg/plug/server.go b/pkg/plug/server.go index d7f8890..bd3295d 100644 --- a/pkg/plug/server.go +++ b/pkg/plug/server.go @@ -3,6 +3,7 @@ package plug import ( "context" "fmt" + "github.com/pkg/errors" "log" "net" @@ -51,9 +52,10 @@ func (p *Plug) Listen() error { source, err = database.GetSourceByDomain(p.ctx, p.source.Domain) if err != nil { - if err.Error() == otterError.NoDataFound { + if errors.Is(err, &otterError.Database{Reason: otterError.NoDataFound}) { log.Printf("Initalizing source!") - if source, err = database.CreateSource(p.ctx, p.source); err != nil { + source, err = database.CreateSource(p.ctx, p.source) + if err != nil { panic(err) } } else {