Handling row not found error from spanner and yo package

erros.go

package errors

import (
	"regexp"

	"cloud.google.com/go/spanner"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

const (
	yoNotFoundMessage      = "no more items in iterator"
	spannerNotFoundMessage = "row not found"
)

type YoError interface {
	Error() string
	Unwrap() error
	DBTableName() string
	GRPCStatus() *status.Status
	Timeout() bool
	Temporary() bool
	NotFound() bool
}

func IsSpannerRecordNotFound(err error) bool {
	yoErr, ok := err.(YoError)
	if !ok {
		// check spanner
		if codes.NotFound != spanner.ErrCode(err) {
			return false
		}

		matched, _ := regexp.MatchString(spannerNotFoundMessage, err.Error())
		return matched
	}

	if !yoErr.NotFound() {
		return false
	}

	matched, _ := regexp.MatchString(yoNotFoundMessage, yoErr.Error())
	return matched
}