migrate from github
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/XingfenD/yoresee_doc/internal/constant"
|
||||
"github.com/XingfenD/yoresee_doc/internal/utils"
|
||||
"github.com/XingfenD/yoresee_doc/pkg/storage"
|
||||
)
|
||||
|
||||
func initializeConfigInConsul(ctx context.Context) error {
|
||||
registerModeKey := utils.GenConfigKey(
|
||||
constant.ConfigKey_First_System,
|
||||
constant.ConfigKey_Second_Security,
|
||||
constant.ConfigKey_Third_RegisterMode,
|
||||
)
|
||||
|
||||
if _, ok, err := storage.Consul.Get(ctx, registerModeKey); err != nil {
|
||||
return err
|
||||
} else if !ok {
|
||||
if err := storage.Consul.Set(ctx, registerModeKey, constant.RegisterMode_Invite); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func markDatabaseInitializedInConsul(ctx context.Context) error {
|
||||
initializedKey := utils.GenConfigKey(
|
||||
constant.ConfigKey_First_System,
|
||||
constant.ConfigKey_Second_Database,
|
||||
constant.ConfigKey_Third_Initialized,
|
||||
)
|
||||
return storage.Consul.Set(ctx, initializedKey, constant.Database_Initialized_True)
|
||||
}
|
||||
Reference in New Issue
Block a user