35 lines
947 B
Go
35 lines
947 B
Go
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)
|
|
}
|