refactor: change driver interface

This commit is contained in:
Noah Hsu
2022-06-15 20:31:23 +08:00
parent d9eb188b7a
commit 09ef7c7106
12 changed files with 181 additions and 136 deletions

View File

@ -12,7 +12,7 @@ import (
"github.com/pkg/errors"
)
func containsByName(files []model.FileInfo, file model.FileInfo) bool {
func containsByName(files []model.Object, file model.Object) bool {
for _, f := range files {
if f.GetName() == file.GetName() {
return true
@ -23,7 +23,7 @@ func containsByName(files []model.FileInfo, file model.FileInfo) bool {
var httpClient = &http.Client{}
func getFileStreamFromLink(file model.FileInfo, link *model.Link) (model.FileStreamer, error) {
func getFileStreamFromLink(file model.Object, link *model.Link) (model.FileStreamer, error) {
var rc io.ReadCloser
mimetype := mime.TypeByExtension(stdpath.Ext(file.GetName()))
if link.Data != nil {
@ -57,7 +57,7 @@ func getFileStreamFromLink(file model.FileInfo, link *model.Link) (model.FileStr
mimetype = "application/octet-stream"
}
stream := model.FileStream{
FileInfo: file,
Object: file,
ReadCloser: rc,
Mimetype: mimetype,
}