重构代码，去掉对omid的依赖