SqlLinkMetadataStore migrates to plain errors (#14674)

Automatic Merge
Этот коммит содержится в:
Rodrigo Villablanca
2020-06-03 16:32:16 -04:00
коммит произвёл GitHub
родитель 2ba119d7f6
Коммит 3649d04dfa
7 изменённых файлов: 29 добавлений и 39 удалений

Просмотреть файл

@@ -6578,14 +6578,6 @@
"id": "store.sql_license.save.app_error", "id": "store.sql_license.save.app_error",
"translation": "We encountered an error saving the license." "translation": "We encountered an error saving the license."
}, },
{
"id": "store.sql_link_metadata.get.app_error",
"translation": "Unable to get the link metadata."
},
{
"id": "store.sql_link_metadata.save.app_error",
"translation": "Unable to save the link metadata."
},
{ {
"id": "store.sql_oauth.delete.commit_transaction.app_error", "id": "store.sql_oauth.delete.commit_transaction.app_error",
"translation": "Unable to commit transaction." "translation": "Unable to commit transaction."

Просмотреть файл

@@ -3900,7 +3900,7 @@ func (s *OpenTracingLayerLicenseStore) Save(license *model.LicenseRecord) (*mode
return resultVar0, resultVar1 return resultVar0, resultVar1
} }
func (s *OpenTracingLayerLinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, *model.AppError) { func (s *OpenTracingLayerLinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, error) {
origCtx := s.Root.Store.Context() origCtx := s.Root.Store.Context()
span, newCtx := tracing.StartSpanWithParentByContext(s.Root.Store.Context(), "LinkMetadataStore.Get") span, newCtx := tracing.StartSpanWithParentByContext(s.Root.Store.Context(), "LinkMetadataStore.Get")
s.Root.Store.SetContext(newCtx) s.Root.Store.SetContext(newCtx)
@@ -3918,7 +3918,7 @@ func (s *OpenTracingLayerLinkMetadataStore) Get(url string, timestamp int64) (*m
return resultVar0, resultVar1 return resultVar0, resultVar1
} }
func (s *OpenTracingLayerLinkMetadataStore) Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, *model.AppError) { func (s *OpenTracingLayerLinkMetadataStore) Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, error) {
origCtx := s.Root.Store.Context() origCtx := s.Root.Store.Context()
span, newCtx := tracing.StartSpanWithParentByContext(s.Root.Store.Context(), "LinkMetadataStore.Save") span, newCtx := tracing.StartSpanWithParentByContext(s.Root.Store.Context(), "LinkMetadataStore.Save")
s.Root.Store.SetContext(newCtx) s.Root.Store.SetContext(newCtx)

Просмотреть файл

@@ -5,12 +5,12 @@ package sqlstore
import ( import (
"database/sql" "database/sql"
"net/http"
sq "github.com/Masterminds/squirrel"
"github.com/mattermost/mattermost-server/v5/model" "github.com/mattermost/mattermost-server/v5/model"
"github.com/mattermost/mattermost-server/v5/store" "github.com/mattermost/mattermost-server/v5/store"
sq "github.com/Masterminds/squirrel"
"github.com/pkg/errors"
) )
type SqlLinkMetadataStore struct { type SqlLinkMetadataStore struct {
@@ -38,7 +38,7 @@ func (s SqlLinkMetadataStore) createIndexesIfNotExists() {
} }
} }
func (s SqlLinkMetadataStore) Save(metadata *model.LinkMetadata) (*model.LinkMetadata, *model.AppError) { func (s SqlLinkMetadataStore) Save(metadata *model.LinkMetadata) (*model.LinkMetadata, error) {
if err := metadata.IsValid(); err != nil { if err := metadata.IsValid(); err != nil {
return nil, err return nil, err
} }
@@ -47,13 +47,13 @@ func (s SqlLinkMetadataStore) Save(metadata *model.LinkMetadata) (*model.LinkMet
err := s.GetMaster().Insert(metadata) err := s.GetMaster().Insert(metadata)
if err != nil && !IsUniqueConstraintError(err, []string{"PRIMARY", "linkmetadata_pkey"}) { if err != nil && !IsUniqueConstraintError(err, []string{"PRIMARY", "linkmetadata_pkey"}) {
return nil, model.NewAppError("SqlLinkMetadataStore.Save", "store.sql_link_metadata.save.app_error", nil, "url="+metadata.URL+", "+err.Error(), http.StatusInternalServerError) return nil, errors.Wrap(err, "could not save link metadata")
} }
return metadata, nil return metadata, nil
} }
func (s SqlLinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, *model.AppError) { func (s SqlLinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, error) {
var metadata *model.LinkMetadata var metadata *model.LinkMetadata
query, args, err := s.getQueryBuilder(). query, args, err := s.getQueryBuilder().
Select("*"). Select("*").
@@ -61,19 +61,19 @@ func (s SqlLinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetad
Where(sq.Eq{"URL": url, "Timestamp": timestamp}). Where(sq.Eq{"URL": url, "Timestamp": timestamp}).
ToSql() ToSql()
if err != nil { if err != nil {
return nil, model.NewAppError("SqlLinkMetadataStore.Get", "store.sql.build_query.app_error", nil, err.Error(), http.StatusInternalServerError) return nil, errors.Wrap(err, "could not create query with querybuilder")
} }
err = s.GetReplica().SelectOne(&metadata, query, args...) err = s.GetReplica().SelectOne(&metadata, query, args...)
if err != nil { if err != nil {
if err == sql.ErrNoRows { if err == sql.ErrNoRows {
return nil, model.NewAppError("SqlLinkMetadataStore.Get", "store.sql_link_metadata.get.app_error", nil, "url="+url+", "+err.Error(), http.StatusNotFound) return nil, store.NewErrNotFound("LinkMetadata", "url="+url)
} }
return nil, model.NewAppError("SqlLinkMetadataStore.Get", "store.sql_link_metadata.get.app_error", nil, "url="+url+", "+err.Error(), http.StatusInternalServerError) return nil, errors.Wrapf(err, "could not get metadata with selectone: url=%s", url)
} }
err = metadata.DeserializeDataToConcreteType() err = metadata.DeserializeDataToConcreteType()
if err != nil { if err != nil {
return nil, model.NewAppError("SqlLinkMetadataStore.Get", "store.sql_link_metadata.get.app_error", nil, "url="+url+", "+err.Error(), http.StatusInternalServerError) return nil, errors.Wrapf(err, "could not deserialize metadata to concrete type for url=%s", url)
} }
return metadata, nil return metadata, nil

Просмотреть файл

@@ -713,8 +713,8 @@ type GroupStore interface {
} }
type LinkMetadataStore interface { type LinkMetadataStore interface {
Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, *model.AppError) Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, error)
Get(url string, timestamp int64) (*model.LinkMetadata, *model.AppError) Get(url string, timestamp int64) (*model.LinkMetadata, error)
} }
// ChannelSearchOpts contains options for searching channels. // ChannelSearchOpts contains options for searching channels.

Просмотреть файл

@@ -4,7 +4,7 @@
package storetest package storetest
import ( import (
"net/http" "errors"
"testing" "testing"
"time" "time"
@@ -153,7 +153,8 @@ func testLinkMetadataStoreGet(t *testing.T, ss store.Store) {
_, err = ss.LinkMetadata().Get("http://example.com/another_page", metadata.Timestamp) _, err = ss.LinkMetadata().Get("http://example.com/another_page", metadata.Timestamp)
require.NotNil(t, err) require.NotNil(t, err)
assert.Equal(t, http.StatusNotFound, err.StatusCode) var nfErr *store.ErrNotFound
assert.True(t, errors.As(err, &nfErr))
}) })
t.Run("should return not found with incorrect timestamp", func(t *testing.T) { t.Run("should return not found with incorrect timestamp", func(t *testing.T) {
@@ -170,7 +171,8 @@ func testLinkMetadataStoreGet(t *testing.T, ss store.Store) {
_, err = ss.LinkMetadata().Get(metadata.URL, getNextLinkMetadataTimestamp()) _, err = ss.LinkMetadata().Get(metadata.URL, getNextLinkMetadataTimestamp())
require.NotNil(t, err) require.NotNil(t, err)
assert.Equal(t, http.StatusNotFound, err.StatusCode) var nfErr *store.ErrNotFound
assert.True(t, errors.As(err, &nfErr))
}) })
} }

Просмотреть файл

@@ -15,7 +15,7 @@ type LinkMetadataStore struct {
} }
// Get provides a mock function with given fields: url, timestamp // Get provides a mock function with given fields: url, timestamp
func (_m *LinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, *model.AppError) { func (_m *LinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, error) {
ret := _m.Called(url, timestamp) ret := _m.Called(url, timestamp)
var r0 *model.LinkMetadata var r0 *model.LinkMetadata
@@ -27,20 +27,18 @@ func (_m *LinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetada
} }
} }
var r1 *model.AppError var r1 error
if rf, ok := ret.Get(1).(func(string, int64) *model.AppError); ok { if rf, ok := ret.Get(1).(func(string, int64) error); ok {
r1 = rf(url, timestamp) r1 = rf(url, timestamp)
} else { } else {
if ret.Get(1) != nil { r1 = ret.Error(1)
r1 = ret.Get(1).(*model.AppError)
}
} }
return r0, r1 return r0, r1
} }
// Save provides a mock function with given fields: linkMetadata // Save provides a mock function with given fields: linkMetadata
func (_m *LinkMetadataStore) Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, *model.AppError) { func (_m *LinkMetadataStore) Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, error) {
ret := _m.Called(linkMetadata) ret := _m.Called(linkMetadata)
var r0 *model.LinkMetadata var r0 *model.LinkMetadata
@@ -52,13 +50,11 @@ func (_m *LinkMetadataStore) Save(linkMetadata *model.LinkMetadata) (*model.Link
} }
} }
var r1 *model.AppError var r1 error
if rf, ok := ret.Get(1).(func(*model.LinkMetadata) *model.AppError); ok { if rf, ok := ret.Get(1).(func(*model.LinkMetadata) error); ok {
r1 = rf(linkMetadata) r1 = rf(linkMetadata)
} else { } else {
if ret.Get(1) != nil { r1 = ret.Error(1)
r1 = ret.Get(1).(*model.AppError)
}
} }
return r0, r1 return r0, r1

Просмотреть файл

@@ -3550,7 +3550,7 @@ func (s *TimerLayerLicenseStore) Save(license *model.LicenseRecord) (*model.Lice
return resultVar0, resultVar1 return resultVar0, resultVar1
} }
func (s *TimerLayerLinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, *model.AppError) { func (s *TimerLayerLinkMetadataStore) Get(url string, timestamp int64) (*model.LinkMetadata, error) {
start := timemodule.Now() start := timemodule.Now()
resultVar0, resultVar1 := s.LinkMetadataStore.Get(url, timestamp) resultVar0, resultVar1 := s.LinkMetadataStore.Get(url, timestamp)
@@ -3566,7 +3566,7 @@ func (s *TimerLayerLinkMetadataStore) Get(url string, timestamp int64) (*model.L
return resultVar0, resultVar1 return resultVar0, resultVar1
} }
func (s *TimerLayerLinkMetadataStore) Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, *model.AppError) { func (s *TimerLayerLinkMetadataStore) Save(linkMetadata *model.LinkMetadata) (*model.LinkMetadata, error) {
start := timemodule.Now() start := timemodule.Now()
resultVar0, resultVar1 := s.LinkMetadataStore.Save(linkMetadata) resultVar0, resultVar1 := s.LinkMetadataStore.Save(linkMetadata)