happyDomain/happyDomain
happyDomain
/
happyDomain
1
0
Fork 0
Code Issues 1 Pull Requests 4 Releases Activity

Add log during leveldb migration process

This commit is contained in:
nemunaire 2021-06-08 22:36:27 +02:00
parent e3fb589cd7
commit 1f7cae7403
2 changed files with 40 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
storage/leveldb/updates-from-0.go
View File

@ -33,6 +33,7 @@ package database
import (
"bytes"
"log"
)
type sourceMeta struct {
@ -48,9 +49,17 @@ func migrateFrom0(s *LevelDBStorage) (err error) {
for iter.Next() {
src := iter.Value()
for src[0] == '"' {
err = decodeData(src, &src)
if err != nil {
return
}
}
src = bytes.Replace(src, []byte("\"Source\":"), []byte("\"Provider\":"), 1)
var srcMeta sourceMeta
err = decodeData(iter.Value(), &srcMeta)
err = decodeData(src, &srcMeta)
if err != nil {
return
}
@ -60,21 +69,39 @@ func migrateFrom0(s *LevelDBStorage) (err error) {
switch srcMeta.Type {
case "ddns.DDNSServer":
newType = "DDNSServer"
case "DDNSServer":
newType = "DDNSServer"
case "gandi.GandiAPI":
newType = "GandiAPI"
case "GandiAPI":
newType = "GandiAPI"
case "ovh.OVHAPI":
newType = "OVHAPI"
case "OVHAPI":
newType = "OVHAPI"
default:
// Keep other source type to update in future version
log.Printf("Migrating v0 -> v1: skip %s (%s)...", iter.Key(), srcMeta.Type)
continue
}
if newType != "" {
bytes.Replace(src, []byte(srcMeta.Type), []byte(newType), 1)
src = bytes.Replace(src, []byte(srcMeta.Type), []byte(newType), 1)
if newType == "DDNSServer" {
src = bytes.Replace(src, []byte("\"hmac-md5.sig-alg.reg.int.\""), []byte("\"hmac-md5\""), 1)
src = bytes.Replace(src, []byte("\"hmac-sha1.\""), []byte("\"hmac-sha1\""), 1)
src = bytes.Replace(src, []byte("\"hmac-sha256.\""), []byte("\"hmac-sha256\""), 1)
src = bytes.Replace(src, []byte("\"hmac-sha512.\""), []byte("\"hmac-sha512\""), 1)
src = bytes.Replace(src, []byte(".\""), []byte("\""), 1)
}
}
newKey := bytes.Replace(iter.Key(), []byte("source-"), []byte("provider-"), 1)
err = s.put(string(newKey), src)
log.Printf("Migrating v0 -> v1: %s to %s (%s)...", iter.Key(), newKey, newType)
err = s.db.Put(newKey, src, nil)
if err != nil {
return
}

11
storage/leveldb/updates.go
View File

@ -31,7 +31,10 @@
package database
import ()
import (
"fmt"
"log"
)
type LevelDBMigrationFunc func(s *LevelDBStorage) error
@ -62,7 +65,12 @@ func (s *LevelDBStorage) DoMigration() (err error) {
return
}
if version > len(migrations) {
return fmt.Errorf("Your database has revision %d, which is newer than the revision this happyDNS version can handle (max DB revision %d). Please update happyDNS", version, len(migrations))
}
for v, migration := range migrations[version:] {
log.Printf("Doing migration from %d to %d", v, v+1)
// Do the migration
if err = migration(s); err != nil {
return
@ -72,6 +80,7 @@ func (s *LevelDBStorage) DoMigration() (err error) {
if err = s.put("version", v+1); err != nil {
return
}
log.Printf("Migration from %d to %d DONE!", v, v+1)
}
return nil