seaweedfs/weed/filer2/abstract_sql/abstract_sql_store.go

132 lines
3.4 KiB
Go
Raw Normal View History

2018-05-26 12:32:15 +00:00
package abstract_sql
import (
2019-03-15 22:55:34 +00:00
"context"
2018-05-26 12:32:15 +00:00
"database/sql"
2018-05-27 18:52:26 +00:00
"fmt"
2018-05-26 12:32:15 +00:00
"github.com/chrislusf/seaweedfs/weed/filer2"
"github.com/chrislusf/seaweedfs/weed/glog"
)
type AbstractSqlStore struct {
2018-05-27 05:02:49 +00:00
DB *sql.DB
SqlInsert string
SqlUpdate string
SqlFind string
SqlDelete string
SqlListExclusive string
SqlListInclusive string
2018-05-26 12:32:15 +00:00
}
2019-03-15 22:55:34 +00:00
func (store *AbstractSqlStore) InsertEntry(ctx context.Context, entry *filer2.Entry) (err error) {
2018-05-26 12:32:15 +00:00
dir, name := entry.FullPath.DirAndName()
meta, err := entry.EncodeAttributesAndChunks()
if err != nil {
2018-05-27 05:02:49 +00:00
return fmt.Errorf("encode %s: %s", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
2018-05-27 05:02:49 +00:00
res, err := store.DB.Exec(store.SqlInsert, hashToLong(dir), name, dir, meta)
2018-05-26 12:32:15 +00:00
if err != nil {
2018-05-27 05:02:49 +00:00
return fmt.Errorf("insert %s: %s", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
_, err = res.RowsAffected()
if err != nil {
2018-05-27 05:02:49 +00:00
return fmt.Errorf("insert %s but no rows affected: %s", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
return nil
}
2019-03-15 22:55:34 +00:00
func (store *AbstractSqlStore) UpdateEntry(ctx context.Context, entry *filer2.Entry) (err error) {
2018-05-26 12:32:15 +00:00
dir, name := entry.FullPath.DirAndName()
meta, err := entry.EncodeAttributesAndChunks()
if err != nil {
2018-05-27 05:02:49 +00:00
return fmt.Errorf("encode %s: %s", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
2018-05-27 05:02:49 +00:00
res, err := store.DB.Exec(store.SqlUpdate, meta, hashToLong(dir), name, dir)
2018-05-26 12:32:15 +00:00
if err != nil {
2018-05-27 05:02:49 +00:00
return fmt.Errorf("update %s: %s", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
_, err = res.RowsAffected()
if err != nil {
2018-05-27 05:02:49 +00:00
return fmt.Errorf("update %s but no rows affected: %s", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
return nil
}
2019-03-15 22:55:34 +00:00
func (store *AbstractSqlStore) FindEntry(ctx context.Context, fullpath filer2.FullPath) (*filer2.Entry, error) {
2018-05-26 12:32:15 +00:00
dir, name := fullpath.DirAndName()
2018-05-27 05:02:49 +00:00
row := store.DB.QueryRow(store.SqlFind, hashToLong(dir), name, dir)
2018-05-26 12:32:15 +00:00
var data []byte
if err := row.Scan(&data); err != nil {
return nil, filer2.ErrNotFound
2018-05-26 12:32:15 +00:00
}
entry := &filer2.Entry{
FullPath: fullpath,
}
if err := entry.DecodeAttributesAndChunks(data); err != nil {
2018-05-27 05:02:49 +00:00
return entry, fmt.Errorf("decode %s : %v", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
return entry, nil
}
2019-03-15 22:55:34 +00:00
func (store *AbstractSqlStore) DeleteEntry(ctx context.Context, fullpath filer2.FullPath) error {
2018-05-26 12:32:15 +00:00
dir, name := fullpath.DirAndName()
2018-05-27 05:02:49 +00:00
res, err := store.DB.Exec(store.SqlDelete, hashToLong(dir), name, dir)
2018-05-26 12:32:15 +00:00
if err != nil {
return fmt.Errorf("delete %s: %s", fullpath, err)
2018-05-26 12:32:15 +00:00
}
_, err = res.RowsAffected()
if err != nil {
return fmt.Errorf("delete %s but no rows affected: %s", fullpath, err)
2018-05-26 12:32:15 +00:00
}
return nil
2018-05-26 12:32:15 +00:00
}
2019-03-15 22:55:34 +00:00
func (store *AbstractSqlStore) ListDirectoryEntries(ctx context.Context, fullpath filer2.FullPath, startFileName string, inclusive bool, limit int) (entries []*filer2.Entry, err error) {
2018-05-26 12:32:15 +00:00
2018-05-27 05:02:49 +00:00
sqlText := store.SqlListExclusive
if inclusive {
2018-05-27 05:02:49 +00:00
sqlText = store.SqlListInclusive
}
2018-05-27 04:24:03 +00:00
rows, err := store.DB.Query(sqlText, hashToLong(string(fullpath)), startFileName, string(fullpath), limit)
2018-05-26 12:32:15 +00:00
if err != nil {
2018-05-27 05:02:49 +00:00
return nil, fmt.Errorf("list %s : %v", fullpath, err)
2018-05-26 12:32:15 +00:00
}
defer rows.Close()
for rows.Next() {
var name string
var data []byte
if err = rows.Scan(&name, &data); err != nil {
2018-05-27 05:02:49 +00:00
glog.V(0).Infof("scan %s : %v", fullpath, err)
return nil, fmt.Errorf("scan %s: %v", fullpath, err)
2018-05-26 12:32:15 +00:00
}
entry := &filer2.Entry{
FullPath: filer2.NewFullPath(string(fullpath), name),
2018-05-26 12:32:15 +00:00
}
if err = entry.DecodeAttributesAndChunks(data); err != nil {
2018-05-27 05:02:49 +00:00
glog.V(0).Infof("scan decode %s : %v", entry.FullPath, err)
return nil, fmt.Errorf("scan decode %s : %v", entry.FullPath, err)
2018-05-26 12:32:15 +00:00
}
entries = append(entries, entry)
}
return entries, nil
}