Bernd-René Predota
5 years ago
9 changed files with 188 additions and 60 deletions
-
1.gitignore
-
6go.mod
-
11go.sum
-
107lib/cache/cache.go
-
54lib/common/common.go
-
51lib/common/guid.go
-
1main.go
-
5models/token.go
-
12repositories/repsitory.go
@ -0,0 +1,107 @@ |
|||||
|
package cache |
||||
|
|
||||
|
import ( |
||||
|
"fmt" |
||||
|
"time" |
||||
|
|
||||
|
badger "github.com/dgraph-io/badger/v2" |
||||
|
// log "github.com/sirupsen/logrus"
|
||||
|
) |
||||
|
|
||||
|
type Reader interface { |
||||
|
Get(key string) (*string, error) |
||||
|
} |
||||
|
|
||||
|
type Writer interface { |
||||
|
Set(key string, value string) error |
||||
|
SetWithTTL(key string, value string, ttl int) error |
||||
|
} |
||||
|
|
||||
|
type Cache interface { |
||||
|
Reader |
||||
|
Writer |
||||
|
Close() |
||||
|
} |
||||
|
|
||||
|
type bCache struct { |
||||
|
db *badger.DB |
||||
|
// logger *log.Entry
|
||||
|
} |
||||
|
|
||||
|
func NewCache(db *badger.DB) Cache { |
||||
|
return &bCache{ |
||||
|
db: db, |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func (cache bCache) close() { |
||||
|
cache.db.Close() |
||||
|
} |
||||
|
|
||||
|
// Get gets a cache entry by key
|
||||
|
func (cache bCache) Get(key string) (*string, error) { |
||||
|
var result []byte |
||||
|
if err := cache.db.View(func(txn *badger.Txn) error { |
||||
|
item, err := txn.Get([]byte(key)) |
||||
|
if err != nil { |
||||
|
if err.Error() != "Key not found" { |
||||
|
return nil |
||||
|
} |
||||
|
} |
||||
|
if err == nil || (item != nil && !item.IsDeletedOrExpired()) { |
||||
|
if err := item.Value(func(val []byte) error { |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
result = append([]byte{}, val...) |
||||
|
return nil |
||||
|
}); err != nil { |
||||
|
return err |
||||
|
} |
||||
|
} |
||||
|
return nil |
||||
|
}); err != nil { |
||||
|
return nil, err |
||||
|
} |
||||
|
r := string(result) |
||||
|
return &r, nil |
||||
|
} |
||||
|
|
||||
|
// SetWithTTL sets a cache entry
|
||||
|
func (cache bCache) Set(key string, value string) error { |
||||
|
if err := cache.db.Update(func(txn *badger.Txn) error { |
||||
|
err := txn.Set([]byte(key), []byte(value)) |
||||
|
return err |
||||
|
}); err != nil { |
||||
|
// cache.logger.WithFields(log.Fields{"module": "cache"}).Errorln(err.Error())
|
||||
|
return err |
||||
|
} |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
// SetWithTTL sets a cache entry with defined ttl
|
||||
|
func (cache bCache) SetWithTTL(key string, value string, ttl int) error { |
||||
|
if err := cache.db.Update(func(txn *badger.Txn) error { |
||||
|
e := badger.NewEntry([]byte(key), []byte(value)).WithTTL(time.Duration(ttl) * time.Second) |
||||
|
err := txn.SetEntry(e) |
||||
|
return err |
||||
|
}); err != nil { |
||||
|
// cache.logger.WithFields(log.Fields{"module": "cache"}).Errorln(err.Error())
|
||||
|
return err |
||||
|
} |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func (cache bCache) Close() { |
||||
|
cache.close() |
||||
|
} |
||||
|
|
||||
|
func Bootstrap() (Cache, error) { |
||||
|
db, err := badger.Open(badger.DefaultOptions("").WithInMemory(true)) |
||||
|
if err != nil { |
||||
|
return nil, fmt.Errorf("Unable to open badger db: %s", err.Error()) |
||||
|
} |
||||
|
|
||||
|
cache := NewCache(db) |
||||
|
return cache, err |
||||
|
} |
||||
@ -0,0 +1,51 @@ |
|||||
|
package common |
||||
|
|
||||
|
import ( |
||||
|
"database/sql/driver" |
||||
|
|
||||
|
"github.com/oklog/ulid/v2" |
||||
|
) |
||||
|
|
||||
|
// GUID our new ULID based datatype
|
||||
|
type GUID ulid.ULID |
||||
|
|
||||
|
// StringToGUID -> parse string to ULID GUID
|
||||
|
func StringToGUID(s string) (GUID, error) { |
||||
|
id, err := ulid.Parse(s) // maybe ParseStrict
|
||||
|
return GUID(id), err |
||||
|
} |
||||
|
|
||||
|
// String Representation of Binary16
|
||||
|
func (guid GUID) String() string { |
||||
|
return ulid.ULID(guid).String() |
||||
|
} |
||||
|
|
||||
|
// GormDataType sets type to binary(16)
|
||||
|
func (guid GUID) GormDataType() string { |
||||
|
return "binary(16)" |
||||
|
} |
||||
|
|
||||
|
func (guid GUID) MarshalJSON() ([]byte, error) { |
||||
|
str := "\"" + ulid.ULID(guid).String() + "\"" |
||||
|
return []byte(str), nil |
||||
|
} |
||||
|
|
||||
|
func (guid *GUID) UnmarshalJSON(data []byte) error { |
||||
|
id, err := ulid.Parse(string(data)) // maybe ParseStrict
|
||||
|
*guid = GUID(id) |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
// Scan tells GORM how to receive from the database
|
||||
|
func (guid *GUID) Scan(value interface{}) error { |
||||
|
converted := ulid.ULID(*guid) |
||||
|
c := &converted |
||||
|
err := c.Scan(value) |
||||
|
*guid = GUID(*c) |
||||
|
return err |
||||
|
} |
||||
|
|
||||
|
// Value tells GORM how to save into the database
|
||||
|
func (guid GUID) Value() (driver.Value, error) { |
||||
|
return ulid.ULID(guid).Value() |
||||
|
} |
||||
@ -0,0 +1,12 @@ |
|||||
|
package repositories |
||||
|
|
||||
|
import ( |
||||
|
"git.devices.local/mawas/golang-api-skeleton/lib/common" |
||||
|
"gorm.io/gorm" |
||||
|
) |
||||
|
|
||||
|
type repo struct { |
||||
|
db *gorm.DB |
||||
|
userID common.GUID |
||||
|
username string |
||||
|
} |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue