Skip to content

Commit c3a6fb0

Browse files
authored
fix(gpud run): create metadata table if not exists (#666)
regresson from #641 Signed-off-by: Gyuho Lee <[email protected]> Signed-off-by: Gyuho Lee <[email protected]>
1 parent 3417085 commit c3a6fb0

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

cmd/gpud/command/run.go

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ func cmdRun(cliContext *cli.Context) error {
8282

8383
rootCtx, rootCancel := context.WithCancel(context.Background())
8484
defer rootCancel()
85+
8586
start := time.Now()
8687

8788
signals := make(chan os.Signal, 2048)
@@ -103,11 +104,22 @@ func cmdRun(cliContext *cli.Context) error {
103104
if err != nil {
104105
return fmt.Errorf("failed to get state file: %w", err)
105106
}
107+
108+
dbRW, err := sqlite.Open(stateFile)
109+
if err != nil {
110+
return fmt.Errorf("failed to open state file: %w", err)
111+
}
112+
defer dbRW.Close()
113+
106114
dbRO, err := sqlite.Open(stateFile, sqlite.WithReadOnly(true))
107115
if err != nil {
108116
return fmt.Errorf("failed to open state file: %w", err)
109117
}
110118
defer dbRO.Close()
119+
120+
if err := gpudstate.CreateTableMachineMetadata(rootCtx, dbRW); err != nil {
121+
return fmt.Errorf("failed to create table: %w", err)
122+
}
111123
uid, err := gpudstate.ReadMachineID(rootCtx, dbRO)
112124
if err != nil {
113125
return fmt.Errorf("failed to read machine ID: %w", err)

0 commit comments

Comments
 (0)