35 lines
465 B
Go
35 lines
465 B
Go
|
package plancontext
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
type tempDirContext struct {
|
||
|
|
l sync.RWMutex
|
||
|
|
store map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *tempDirContext) Add(dir, key string) {
|
||
|
|
c.l.Lock()
|
||
|
|
defer c.l.Unlock()
|
||
|
|
|
||
|
|
c.store[key] = dir
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *tempDirContext) Get(key string) string {
|
||
|
|
c.l.RLock()
|
||
|
|
defer c.l.RUnlock()
|
||
|
|
|
||
|
|
return c.store[key]
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *tempDirContext) Clean() {
|
||
|
|
c.l.RLock()
|
||
|
|
defer c.l.RUnlock()
|
||
|
|
|
||
|
|
for _, s := range c.store {
|
||
|
|
defer os.RemoveAll(s)
|
||
|
|
}
|
||
|
|
}
|