feat(lstv2 move): moved lstv2 into this app to keep them combined and easier to maintain
This commit is contained in:
84
lstV2/server/services/ocp/controller/labeling/labelRatio.ts
Normal file
84
lstV2/server/services/ocp/controller/labeling/labelRatio.ts
Normal file
@@ -0,0 +1,84 @@
|
||||
import { eq, sql } from "drizzle-orm";
|
||||
import { db } from "../../../../../database/dbclient.js";
|
||||
import { labelRatio } from "../../../../../database/schema/ratios.js";
|
||||
import { tryCatch } from "../../../../globalUtils/tryCatch.js";
|
||||
import { createLog } from "../../../logger/logger.js";
|
||||
|
||||
export const autoLabelCreated = async () => {
|
||||
const { error } = await tryCatch(
|
||||
db
|
||||
.insert(labelRatio)
|
||||
.values({
|
||||
name: "label",
|
||||
autoLabel: 1,
|
||||
})
|
||||
.onConflictDoUpdate({
|
||||
target: labelRatio.name,
|
||||
set: { autoLabel: sql`${labelRatio.autoLabel} + 1` },
|
||||
})
|
||||
);
|
||||
|
||||
if (error) {
|
||||
createLog(
|
||||
"error",
|
||||
"labeling",
|
||||
"ocp",
|
||||
"There was an error updating auto label ratio"
|
||||
);
|
||||
}
|
||||
};
|
||||
|
||||
export const manualLabelCreated = async () => {
|
||||
const { error } = await tryCatch(
|
||||
db
|
||||
.insert(labelRatio)
|
||||
.values({
|
||||
name: "label",
|
||||
manualLabel: 1,
|
||||
})
|
||||
.onConflictDoUpdate({
|
||||
target: labelRatio.name,
|
||||
set: { manualLabel: sql`${labelRatio.manualLabel} + 1` },
|
||||
})
|
||||
);
|
||||
|
||||
if (error) {
|
||||
createLog(
|
||||
"error",
|
||||
"labeling",
|
||||
"ocp",
|
||||
"There was an error updating manual Label Ratio"
|
||||
);
|
||||
}
|
||||
};
|
||||
|
||||
export const resetLabelRatio = async () => {
|
||||
const { error } = await tryCatch(
|
||||
db
|
||||
.update(labelRatio)
|
||||
.set({
|
||||
name: sql`'label-' || (SELECT count(*) FROM ${labelRatio})`,
|
||||
lastReset: sql`NOW()`,
|
||||
})
|
||||
.where(eq(labelRatio.name, "label"))
|
||||
);
|
||||
|
||||
if (error) {
|
||||
console.log(error);
|
||||
createLog(
|
||||
"error",
|
||||
"labeling",
|
||||
"ocp",
|
||||
"There was an error resetting Label Ratio"
|
||||
);
|
||||
return {
|
||||
success: false,
|
||||
message: "There was an issue resetting the label data.",
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
success: true,
|
||||
message: "Label Ratio has been reset.",
|
||||
};
|
||||
};
|
||||
Reference in New Issue
Block a user