ocean-subgraph/src/mappings/utils/dispenserUtils.ts

18 lines
471 B
TypeScript

import { Dispenser } from '../../@types/schema'
import { getToken } from './tokenUtils'
export function createDispenser(address: string): Dispenser {
const dispenser = new Dispenser(address)
dispenser.token = getToken(address).id
dispenser.save()
return dispenser
}
export function getDispenser(address: string): Dispenser {
let dispenser = Dispenser.load(address)
if (dispenser === null) {
dispenser = createDispenser(address)
}
return dispenser
}